def test_eq(self): catalog1 = Catalog() catalog1.box2d = Box2DSubcatalog() catalog2 = Catalog() catalog2.box2d = Box2DSubcatalog() catalog3 = Catalog() catalog3.box3d = Box3DSubcatalog() assert catalog1 == catalog2 assert catalog1 != catalog3
def test_dumps(self, categories, keypoints, catalog_contents): catalog = Catalog() catalog.box2d = Box2DSubcatalog() catalog.box3d = Box3DSubcatalog() catalog.polygon = PolygonSubcatalog() catalog.polyline2d = Polyline2DSubcatalog() catalog.sentence = SentenceSubcatalog() classificationsubcatalog = ClassificationSubcatalog() classificationsubcatalog.categories = categories catalog.classification = classificationsubcatalog keypoints2dsubcatalog = Keypoints2DSubcatalog() keypoints2dsubcatalog._keypoints = [keypoints] catalog.keypoints2d = keypoints2dsubcatalog assert catalog.dumps() == catalog_contents