示例#1
0
    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
示例#2
0
    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
示例#3
0
    def test_bool(self):
        catalog = Catalog()
        assert catalog.__bool__() == False

        catalog.box2d = Box2DSubcatalog()
        assert catalog.__bool__() == True