def test_collate_annotations_with_list_and_scalar(self):
truth = [1, 2, 3]
one_and_two = AnnotatedImage(INFO_F0, truth[:2])
three = AnnotatedImage(INFO_F0, truth[2])
collated = one_and_two.collate_annotations_with(three)
self.assertEqual(collated.annotation, truth)
def test_collate_annotations_with_when_images_with_scalar_data(self):
truth = [1, 2]
one = AnnotatedImage(INFO_F0, truth[0])
two = AnnotatedImage(INFO_F0, truth[1])
collated = one.collate_annotations_with(two)
self.assertEqual(collated.annotation, truth)
def test_collate_annotations_with_list_and_tuple(self):
truth = [1, 2, 3, 4]
one_and_two = AnnotatedImage(INFO_F0, truth[:2])
three_and_four = AnnotatedImage(INFO_F0, tuple(truth[2:]))
collated = one_and_two.collate_annotations_with(three_and_four)
self.assertEqual(collated.annotation, truth)
def test_collate_annotations_with_scalar_and_list(self):
truth = [1, 2, 3]
one = AnnotatedImage(INFO_F0, truth[0])
two_and_three = AnnotatedImage(INFO_F0, truth[1:])
collated = one.collate_annotations_with(two_and_three)
self.assertEqual(collated.annotation, truth)