def test_intersection(self):
ann_1 = Annotations(self.ann_path_1)
ann_2 = Annotations(self.ann_path_2)
ann_1.add_entity(*ann_2.annotations[0])
ann_1.add_entity(*ann_2.annotations[1])
expected = {ann_2.annotations[0], ann_2.annotations[1]}
actual = ann_1.intersection(ann_2)
self.assertSetEqual(actual, expected)
def test_confusion_matrix(self):
ann_1 = Annotations(self.ann_path_1)
ann_2 = Annotations(self.ann_path_2)
ann_1.add_entity(*ann_2.annotations[0])
self.assertEqual(
len(ann_1.compute_confusion_matrix(ann_2, self.entities)[0]),
len(self.entities))
self.assertEqual(
len(ann_1.compute_confusion_matrix(ann_2, self.entities)),
len(self.entities))
