update unit test

erdogant · Jun 17, 2023 · 03f33e4 · 03f33e4
1 parent 62ff3ec
commit 03f33e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/clustimage/tests/test_clustimage.py b/clustimage/tests/test_clustimage.py
@@ -21,7 +21,7 @@ def test_import_data(self):
         cl = Clustimage(dim=(128, 128), grayscale=False)
         _ =  cl.import_data(X)
         assert np.all(np.isin([*cl.results.keys()], ['img', 'feat', 'xycoord', 'pathnames', 'labels', 'filenames', 'url']))
-        assert cl.results['img'].shape==(len(X), 49152)
+        assert cl.results['img'].shape==(len(X), 65536)
         # Check grayscale parameter with imports
         cl = Clustimage(dim=(128, 128), grayscale=True)
         _ = cl.import_data(X)
@@ -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)