Fix tests

cvat-ai · Oct 29, 2021 · 2c6d67a · 2c6d67a
1 parent 4f5053a
commit 2c6d67a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/cvat/apps/dataset_manager/tests/test_formats.py b/cvat/apps/dataset_manager/tests/test_formats.py
@@ -323,7 +323,7 @@ def test_import_formats_query(self):
             'ICDAR Segmentation 1.0',
             'Kitti Raw Format 1.0',
             'Sly Point Cloud Format 1.0',
-            'Cityscapes 1.0'
+            'Cityscapes 1.0',
             'Datumaro 1.0',
             'Datumaro 3D 1.0'
         })

diff --git a/cvat/apps/dataset_manager/tests/test_rest_api_formats.py b/cvat/apps/dataset_manager/tests/test_rest_api_formats.py
@@ -1011,7 +1011,7 @@ def test_api_v1_tasks_annotations_dump_and_upload_with_datumaro(self):
                         "ICDAR Segmentation 1.0", # changed points values
                         'Kitti Raw Format 1.0',
                         'Sly Point Cloud Format 1.0',
-                        'Cityscapes 1.0' # changed points value
+                        'Cityscapes 1.0', # changed points value
                         'Datumaro 3D 1.0'
                     ]:
                         self.skipTest("Format is fail")

diff --git a/cvat/apps/engine/tests/test_rest_api.py b/cvat/apps/engine/tests/test_rest_api.py
@@ -4811,6 +4811,10 @@ def _get_initial_annotation(annotation_format):
                 annotations["shapes"] = points_wo_attrs \
                                       + rectangle_shapes_wo_attrs
 
+            elif annotation_format == "Cityscapes 1.0":
+                annotations["shapes"] = points_wo_attrs \
+                                      + rectangle_shapes_wo_attrs
+
             elif annotation_format == "Market-1501 1.0":
                 tags_with_attrs = [{
                     "frame": 1,