def get_cluster_list(patterns): cl = ClusterList() for pattern in patterns: c = Cluster(pattern) c.files = [File(f) for f in glob(pattern)] cl.append(c) return cl
def test_add_no_changes(self): cluster = Cluster('Test') cluster.files = self.test_files update_metadata_images(cluster) added = add_metadata_images(cluster, [self.test_files[1]]) self.assertFalse(added) self.assertEqual(set(self.test_images), set(cluster.metadata.images))
def test_add_to_cluster(self): cluster = Cluster('Test') cluster.files = [self.test_files[0]] update_metadata_images(cluster) cluster.files += self.test_files[1:] add_metadata_images(cluster, self.test_files[1:]) self.assertEqual(set(self.test_images), set(cluster.metadata.images)) self.assertFalse(cluster.metadata.has_common_images)
def test_remove_from_cluster_with_common_images(self): cluster = Cluster('Test') cluster.files = list(self.test_files[1:]) update_metadata_images(cluster) cluster.files.remove(self.test_files[1]) remove_metadata_images(cluster, [self.test_files[1]]) self.assertEqual(set(self.test_images[1:]), set(cluster.metadata.images)) self.assertTrue(cluster.metadata.has_common_images)
def test_add_to_cluster(self): cluster = Cluster('Test') cluster.files = [self.test_files[0]] update_metadata_images(cluster) cluster.files += self.test_files[1:] add_metadata_images(cluster, self.test_files[1:]) self.assertEqual(set(self.test_images), set(cluster.metadata.images)) self.assertFalse(cluster.metadata.has_common_images)
def test_remove_from_cluster_with_common_images(self): cluster = Cluster('Test') cluster.files = list(self.test_files[1:]) update_metadata_images(cluster) cluster.files.remove(self.test_files[1]) remove_metadata_images(cluster, [self.test_files[1]]) self.assertEqual(set(self.test_images[1:]), set(cluster.metadata.images)) self.assertTrue(cluster.metadata.has_common_images)
def test_update_cluster_images(self): cluster = Cluster('Test') cluster.files = list(self.test_files) update_metadata_images(cluster) self.assertEqual(set(self.test_images), set(cluster.metadata.images)) self.assertFalse(cluster.metadata.has_common_images) cluster.files.remove(self.test_files[2]) update_metadata_images(cluster) self.assertEqual(set(self.test_images), set(cluster.metadata.images)) self.assertFalse(cluster.metadata.has_common_images) cluster.files.remove(self.test_files[0]) update_metadata_images(cluster) self.assertEqual(set(self.test_images[1:]), set(cluster.metadata.images)) self.assertTrue(cluster.metadata.has_common_images) cluster.files.append(self.test_files[2]) update_metadata_images(cluster) self.assertEqual(set(self.test_images[1:]), set(cluster.metadata.images)) self.assertTrue(cluster.metadata.has_common_images)
def test_update_cluster_images(self): cluster = Cluster('Test') cluster.files = list(self.test_files) update_metadata_images(cluster) self.assertEqual(set(self.test_images), set(cluster.metadata.images)) self.assertFalse(cluster.metadata.has_common_images) cluster.files.remove(self.test_files[2]) update_metadata_images(cluster) self.assertEqual(set(self.test_images), set(cluster.metadata.images)) self.assertFalse(cluster.metadata.has_common_images) cluster.files.remove(self.test_files[0]) update_metadata_images(cluster) self.assertEqual(set(self.test_images[1:]), set(cluster.metadata.images)) self.assertTrue(cluster.metadata.has_common_images) cluster.files.append(self.test_files[2]) update_metadata_images(cluster) self.assertEqual(set(self.test_images[1:]), set(cluster.metadata.images)) self.assertTrue(cluster.metadata.has_common_images)
def test_add_nothing(self): cluster = Cluster('Test') cluster.files = self.test_files update_metadata_images(cluster) added = add_metadata_images(cluster, []) self.assertFalse(added)