update unit test

erdogant · Dec 29, 2024 · c9c383e · c9c383e
1 parent 08cb8ce
commit c9c383e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/clustimage/clustimage.py b/clustimage/clustimage.py
@@ -38,8 +38,8 @@
 import imagehash
 from PIL import Image
 
-# import clustimage.exif as exif
-import exif as exif
+import clustimage.exif as exif
+# import exif as exif
 import webbrowser
 
 # Support for Apple HEIC images

diff --git a/clustimage/tests/test_clustimage.py b/clustimage/tests/test_clustimage.py
@@ -42,7 +42,7 @@ def test_extract_feat(self):
         X = cl.import_example(data='flowers')
         X = cl.import_data(X)
         _ = cl.extract_feat(X)
-        assert cl.results['feat'].shape==(X['img'].shape[0], 153)
+        assert cl.results['feat'].shape==(X['img'].shape[0], 152)
 
         # Init with settings such as PCA
         cl = Clustimage(method='hog', verbose=50)
@@ -61,7 +61,7 @@ def test_embedding(self):
         X = cl.import_data(X)
         Xfeat = cl.extract_feat(X)
         out = cl.embedding(Xfeat)
-        assert out.shape==(214,2)
+        assert out.shape==(219,2)
         assert cl.results['xycoord'].shape==(X['img'].shape[0], 2)
 
     def test_embedding(self):
@@ -71,7 +71,7 @@ def test_embedding(self):
         X = cl.import_data(X)
         Xfeat = cl.extract_feat(X)
         xycoord = cl.embedding(Xfeat)
-        assert xycoord.shape ==(214, 2)
+        assert xycoord.shape ==(219, 2)
         labels = cl.cluster()
         assert len(cl.results['labels'])==X['img'].shape[0]