TST: Adapt test to recent changes

imagecluster/tests/test_ic.py

@@ -10,7 +10,6 @@
 logging.getLogger("tensorflow").setLevel(logging.ERROR)
 
 from imagecluster import main
-from imagecluster import imagecluster as ic
 pj = os.path.join
 
 
@@ -20,8 +19,8 @@ def test():
         imagedir = tempfile.mkdtemp(prefix='imagecluster_')
         dbfn = pj(imagedir, main.ic_base_dir, 'fingerprints.pk')
         arr = misc.face()
-        images = [arr, 
-                  ni.gaussian_filter(arr, 10), 
+        images = [arr,
+                  ni.gaussian_filter(arr, 10),
                   ni.gaussian_filter(arr, 20),
                   arr[...,0], # fake gray-scale image
                   ]
@@ -32,8 +31,8 @@ def test():
             image_fns.append(fn)
         # run 1: create fingerprints database, run clustering
         main.main(imagedir)
-        # run 2: only run clustering, should be much faster
-        main.main(imagedir)
+        # run 2: only run clustering, should be much faster, this time use PCA
+        main.main(imagedir, pca=True)
         with open(dbfn, 'rb') as fd:
             fps = pickle.load(fd)
         assert len(fps.keys()) == 4