diff --git a/situr/image/situ_image.py b/situr/image/situ_image.py
index 75c41d5..aa9f5df 100644
--- a/situr/image/situ_image.py
+++ b/situr/image/situ_image.py
@@ -1,3 +1,9 @@
+import numpy as np
+from PIL import Image, ImageDraw
+from skimage import img_as_float
+from skimage.feature import blob_dog
+
+
 def extend_dim(array):
     ones = np.ones((array.shape[0], 1))
     return np.append(array, ones, axis=1)
diff --git a/tests/test_situ_image.py b/tests/test_situ_image.py
new file mode 100644
index 0000000..f9649d6
--- /dev/null
+++ b/tests/test_situ_image.py
@@ -0,0 +1,22 @@
+from situr.image import extend_dim, remove_dim
+
+import numpy as np
+import unittest
+
+
+class TestDimMethods(unittest.TestCase):
+    def test_identical_after_expand_remove_(self):
+        arr = np.random.rand(5, 2)
+        self.assertTrue(
+            np.array_equal(
+                arr, remove_dim(extend_dim(np.copy(arr)))
+            )
+        )
+
+    def test_extend_dim_only_extends(self):
+        arr = np.random.rand(5, 2)
+        self.assertTrue(
+            np.array_equal(
+                arr, extend_dim(np.copy(arr))[:, :-1]
+            )
+        )