def test_serialize(self):
""" test basic serialization with parser """
testanno1 = Annotation()
testanno2 = Annotation()
testanno2.class_label = 'person'
testanno2.object_id = 1
testanno2.difficult = True
obj = {
'img_1': [testanno1, testanno2],
'img_2': [testanno1, testanno1, testanno1],
}
string = self.parser.serialize(obj)
self.assertEqual(string, yaml_string)
def test_parser_serialize(self):
""" test basic serialization with parser """
testanno1 = Annotation()
testanno1.class_label = 'horse'
testanno1.x_top_left = 100
testanno1.y_top_left = 200
testanno1.width = 200
testanno1.height = 200
testanno1.difficult = True
testanno2 = Annotation()
testanno2.class_label = 'person'
testanno2.x_top_left = 1
testanno2.y_top_left = 2
testanno2.width = 2
testanno2.height = 2
testanno2.occluded = True
string = self.parser.serialize([testanno1, testanno2])
self.assertEqual(string, xml_string)