add basic tests for Dim expansion and reduction and added imports

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)

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]
+            )
+        )