Merge branch 'master' of https://github.com/superannotateai/superannotate-python-sdk

Author: agyurjin

.vscode/settings.json

@@ -1,5 +1,11 @@
 {
     "python.formatting.provider": "yapf",
     "python.linting.pylintPath": "~/miniconda3/bin/pylint",
-    "python.linting.pylintEnabled": true
+    "python.linting.pylintEnabled": true,
+    "python.testing.pytestArgs": [
+        "tests"
+    ],
+    "python.testing.unittestEnabled": false,
+    "python.testing.nosetestsEnabled": false,
+    "python.testing.pytestEnabled": true
 }

pytest.ini

@@ -0,0 +1,2 @@
+[pytest]
+addopts=-n8

tests/test_annotation_adding.py

@@ -131,7 +131,16 @@ def test_add_bbox_noinit(tmpdir):
     assert len(annotations_new["instances"]) == 2
     export = sa.prepare_export(project, include_fuse=True)
     sa.download_export(project, export, tmpdir)
-    assert len(list(Path(tmpdir).rglob("*.*"))) == 4
+
+    non_empty_annotations = 0
+    json_files = tmpdir.glob("*.json")
+    for json_file in json_files:
+        json_ann = json.load(open(json_file))
+        if "instances" in json_ann and len(json_ann["instances"]) > 0:
+            non_empty_annotations += 1
+            assert len(json_ann["instances"]) == 2
+
+    assert non_empty_annotations == 1
 
 
 def test_add_bbox_json(tmpdir):
@@ -169,5 +178,4 @@ def test_add_bbox_json(tmpdir):
 
     assert len(annotations_new["instances"]
               ) == len(annotations["instances"]) + 7
-    assert len(annotations_new["comments"]
-              ) == len(annotations["comments"]) + 1
+    assert len(annotations_new["comments"]) == len(annotations["comments"]) + 1

tests/test_recursive_folder.py

@@ -1,10 +1,9 @@
+import json
 from pathlib import Path
 import time
 
 import superannotate as sa
 
-sa.init(Path.home() / ".superannotate" / "config.json")
-
 TEMP_PROJECT_NAME = "test_recursive"
 
 
@@ -40,7 +39,14 @@ def test_nonrecursive_annotations_folder(tmpdir):
     time.sleep(1)
     sa.download_export(project, export, tmpdir)
 
-    assert len(list(tmpdir.glob("*.json"))) == 1
+    non_empty_annotations = 0
+    json_files = tmpdir.glob("*.json")
+    for json_file in json_files:
+        json_ann = json.load(open(json_file))
+        if "instances" in json_ann and len(json_ann["instances"]) > 0:
+            non_empty_annotations += 1
+
+    assert non_empty_annotations == 1
 
 
 def test_recursive_annotations_folder(tmpdir):
@@ -223,7 +229,14 @@ def test_annotations_nonrecursive_s3_folder(tmpdir):
     time.sleep(1)
     sa.download_export(project, export, tmpdir)
 
-    assert len(list(tmpdir.glob("*.json"))) == 1
+    non_empty_annotations = 0
+    json_files = tmpdir.glob("*.json")
+    for json_file in json_files:
+        json_ann = json.load(open(json_file))
+        if "instances" in json_ann and len(json_ann["instances"]) > 0:
+            non_empty_annotations += 1
+
+    assert non_empty_annotations == 1
 
 
 def test_preannotations_recursive_s3_folder(tmpdir):