diff --git a/imagecluster/tests/test_ic.py b/imagecluster/tests/test_ic.py index a23def9..aadf7a0 100644 --- a/imagecluster/tests/test_ic.py +++ b/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