format

Author: Grant Eaton

test.py

@@ -7,7 +7,7 @@
 
 from typing import Type
 from labelbox.schema.labeling_frontend import LabelingFrontend
-from labelbox.schema.annotation_import import MALPredictionImport, AnnotationImportState 
+from labelbox.schema.annotation_import import MALPredictionImport, AnnotationImportState
 
 
 @pytest.fixture
@@ -346,13 +346,14 @@ def model_run_annotation_groups(client, configured_project,
 
 
 class AnnotationImportTestHelpers:
+
     @staticmethod
     def assert_file_content(url: str, predictions):
         response = requests.get(url)
         assert response.text == ndjson.dumps(predictions)
 
     @staticmethod
-    def check_running_state(req, name, url=None): 
+    def check_running_state(req, name, url=None):
         assert req.name == name
         if url is not None:
             assert req.input_file_url == url
@@ -364,4 +365,3 @@ def check_running_state(req, name, url=None):
 @pytest.fixture
 def annotation_import_test_helpers() -> Type[AnnotationImportTestHelpers]:
     return AnnotationImportTestHelpers()
-

tests/integration/annotation_import/conftest.py

@@ -39,7 +39,8 @@ def test_validate_file(client, configured_project):
         #Schema ids shouldn't match
 
 
-def test_create_from_objects(configured_project, predictions, annotation_import_test_helpers):
+def test_create_from_objects(configured_project, predictions,
+                             annotation_import_test_helpers):
     name = str(uuid.uuid4())
 
     bulk_import_request = configured_project.upload_annotations(
@@ -50,7 +51,8 @@ def test_create_from_objects(configured_project, predictions, annotation_import_
     assert bulk_import_request.error_file_url is None
     assert bulk_import_request.status_file_url is None
     assert bulk_import_request.state == BulkImportRequestState.RUNNING
-    annotation_import_test_helpers.assert_file_content(bulk_import_request.input_file_url, predictions)
+    annotation_import_test_helpers.assert_file_content(
+        bulk_import_request.input_file_url, predictions)
 
 
 def test_create_from_local_file(tmp_path, predictions, configured_project):
@@ -68,7 +70,8 @@ def test_create_from_local_file(tmp_path, predictions, configured_project):
     assert bulk_import_request.error_file_url is None
     assert bulk_import_request.status_file_url is None
     assert bulk_import_request.state == BulkImportRequestState.RUNNING
-    annotation_import_test_helpers.assert_file_content(bulk_import_request.input_file_url, predictions)
+    annotation_import_test_helpers.assert_file_content(
+        bulk_import_request.input_file_url, predictions)
 
 
 def test_get(client, configured_project):

tests/integration/annotation_import/test_bulk_import_request.py

@@ -14,18 +14,26 @@
 def test_create_from_url(client, project, annotation_import_test_helpers):
     name = str(uuid.uuid4())
     url = "https://storage.googleapis.com/labelbox-public-bucket/predictions_test_v2.ndjson"
-    label_import = LabelImport.create_from_url(client=client, project_id=project.uid, name=name, url=url)
-    assert label_import.parent_id == project.uid 
+    label_import = LabelImport.create_from_url(client=client,
+                                               project_id=project.uid,
+                                               name=name,
+                                               url=url)
+    assert label_import.parent_id == project.uid
     annotation_import_test_helpers.check_running_state(label_import, name, url)
 
 
-def test_create_from_objects(client, project, object_predictions, annotation_import_test_helpers):
+def test_create_from_objects(client, project, object_predictions,
+                             annotation_import_test_helpers):
     name = str(uuid.uuid4())
 
-    label_import = LabelImport.create_from_objects(client=client, project_id=project.uid, name=name, labels=object_predictions)
-    assert label_import.parent_id == project.uid 
+    label_import = LabelImport.create_from_objects(client=client,
+                                                   project_id=project.uid,
+                                                   name=name,
+                                                   labels=object_predictions)
+    assert label_import.parent_id == project.uid
     annotation_import_test_helpers.check_running_state(label_import, name)
-    annotation_import_test_helpers.assert_file_content(label_import.input_file_url, object_predictions)
+    annotation_import_test_helpers.assert_file_content(
+        label_import.input_file_url, object_predictions)
 
 
 #   TODO: add me when we add this ability
@@ -48,7 +56,10 @@ def test_get(client, project, annotation_import_test_helpers):
     name = str(uuid.uuid4())
     url = "https://storage.googleapis.com/labelbox-public-bucket/predictions_test_v2.ndjson"
 
-    label_import = LabelImport.create_from_url(client=client, project_id=project.uid, name=name, url=url)
+    label_import = LabelImport.create_from_url(client=client,
+                                               project_id=project.uid,
+                                               name=name,
+                                               url=url)
 
     assert label_import.parent_id == project.uid
     annotation_import_test_helpers.check_running_state(label_import, name, url)
@@ -57,7 +68,10 @@ def test_get(client, project, annotation_import_test_helpers):
 @pytest.mark.slow
 def test_wait_till_done(client, project, predictions):
     name = str(uuid.uuid4())
-    label_import = LabelImport.create_from_objects(client=client, project_id=project.uid, name=name, labels=predictions)
+    label_import = LabelImport.create_from_objects(client=client,
+                                                   project_id=project.uid,
+                                                   name=name,
+                                                   labels=predictions)
 
     assert len(label_import.inputs) == len(predictions)
     label_import.wait_until_done()
@@ -68,9 +82,7 @@ def test_wait_till_done(client, project, predictions):
     # # Check that the status files are being returned as expected
     # assert len(label_import.errors) == 0
     assert len(label_import.inputs) == len(predictions)
-    input_uuids = [
-        input_annot['uuid'] for input_annot in label_import.inputs
-    ]
+    input_uuids = [input_annot['uuid'] for input_annot in label_import.inputs]
     inference_uuids = [pred['uuid'] for pred in predictions]
     assert set(input_uuids) == set(inference_uuids)
     assert len(label_import.statuses) == len(predictions)
@@ -80,4 +92,3 @@ def test_wait_till_done(client, project, predictions):
         input_annot['uuid'] for input_annot in label_import.statuses
     ]
     assert set(input_uuids) == set(status_uuids)
-

tests/integration/annotation_import/test_label_import.py

@@ -10,28 +10,33 @@
 """
 
 
-def test_create_from_url(model_run_annotation_groups, annotation_import_test_helpers):
+def test_create_from_url(model_run_annotation_groups,
+                         annotation_import_test_helpers):
     name = str(uuid.uuid4())
     url = "https://storage.googleapis.com/labelbox-public-bucket/predictions_test_v2.ndjson"
     annotation_import = model_run_annotation_groups.add_predictions(
         name=name, predictions=url)
     assert annotation_import.model_run_id == model_run_annotation_groups.uid
-    annotation_import_test_helpers.check_running_state(annotation_import, name, url)
+    annotation_import_test_helpers.check_running_state(annotation_import, name,
+                                                       url)
 
 
-def test_create_from_objects(model_run_annotation_groups, object_predictions, annotation_import_test_helpers):
+def test_create_from_objects(model_run_annotation_groups, object_predictions,
+                             annotation_import_test_helpers):
     name = str(uuid.uuid4())
 
     annotation_import = model_run_annotation_groups.add_predictions(
         name=name, predictions=object_predictions)
 
     assert annotation_import.model_run_id == model_run_annotation_groups.uid
     annotation_import_test_helpers.check_running_state(annotation_import, name)
-    annotation_import_test_helpers.assert_file_content(annotation_import.input_file_url, object_predictions)
+    annotation_import_test_helpers.assert_file_content(
+        annotation_import.input_file_url, object_predictions)
 
 
 def test_create_from_local_file(tmp_path, model_run_annotation_groups,
-                                object_predictions, annotation_import_test_helpers):
+                                object_predictions,
+                                annotation_import_test_helpers):
     name = str(uuid.uuid4())
     file_name = f"{name}.ndjson"
     file_path = tmp_path / file_name
@@ -43,10 +48,12 @@ def test_create_from_local_file(tmp_path, model_run_annotation_groups,
 
     assert annotation_import.model_run_id == model_run_annotation_groups.uid
     annotation_import_test_helpers.check_running_state(annotation_import, name)
-    annotation_import_test_helpers.assert_file_content(annotation_import.input_file_url, object_predictions)
+    annotation_import_test_helpers.assert_file_content(
+        annotation_import.input_file_url, object_predictions)
 
 
-def test_get(client, model_run_annotation_groups, annotation_import_test_helpers):
+def test_get(client, model_run_annotation_groups,
+             annotation_import_test_helpers):
     name = str(uuid.uuid4())
     url = "https://storage.googleapis.com/labelbox-public-bucket/predictions_test_v2.ndjson"
     model_run_annotation_groups.add_predictions(name=name, predictions=url)
@@ -55,7 +62,8 @@ def test_get(client, model_run_annotation_groups, annotation_import_test_helpers
         client, model_run_id=model_run_annotation_groups.uid, name=name)
 
     assert annotation_import.model_run_id == model_run_annotation_groups.uid
-    annotation_import_test_helpers.check_running_state(annotation_import, name, url)
+    annotation_import_test_helpers.check_running_state(annotation_import, name,
+                                                       url)
 
 
 @pytest.mark.slow
@@ -82,4 +90,3 @@ def test_wait_till_done(model_run_predictions, model_run_annotation_groups):
         input_annot['uuid'] for input_annot in annotation_import.statuses
     ]
     assert set(input_uuids) == set(status_uuids)
-