def test_annotation_collection_can_remove_valid_annotation(self): collection = AnnotationCollection(self.collection) annotation = Annotation(copy.copy(examples["vincent"])) collection.add_annotation(annotation.id) self.assertEqual(collection.has_annotation(annotation.id), True) collection.remove_annotation(annotation.id) self.assertEqual(collection.has_annotation(annotation.id), False)
def test_annotation_collection_can_retrieve_annotation(self): collection = AnnotationCollection(self.collection) annotation = Annotation(copy.copy(examples["vincent"])) collection.add_annotation(annotation.id) self.assertTrue(collection.has_annotation(annotation.id))