improve test stability

Author: Matt Sokoloff

Makefile

@@ -6,7 +6,7 @@ test-staging: build
 	docker run -it -v ${PWD}:/usr/src -w /usr/src \
 		-e LABELBOX_TEST_ENVIRON="staging" \
 		-e LABELBOX_TEST_API_KEY_STAGING=${LABELBOX_TEST_API_KEY_STAGING} \
-		local/labelbox-python:test pytest $(PATH_TO_TEST) -svv
+		local/labelbox-python:test pytest $(PATH_TO_TEST) -svvx
 
 test-prod: build
 	docker run -it -v ${PWD}:/usr/src -w /usr/src \

tests/integration/bulk_import/conftest.py

@@ -110,9 +110,11 @@ def configured_project(client, ontology, rand_gen):
         client.get_labeling_frontends(
             where=LabelingFrontend.name == "editor"))[0]
     project.setup(editor, ontology)
+    data_row_ids = []
     for _ in range(len(ontology['tools']) + len(ontology['classifications'])):
-        dataset.create_data_row(row_data=IMG_URL)
+        data_row_ids.append(dataset.create_data_row(row_data=IMG_URL).uid)
     project.datasets.connect(dataset)
+    project.data_row_ids = data_row_ids
     yield project
     project.delete()
     dataset.delete()
@@ -122,7 +124,6 @@ def configured_project(client, ontology, rand_gen):
 def prediction_id_mapping(configured_project):
     #Maps tool types to feature schema ids
     ontology = configured_project.ontology().normalized
-    datarows = [d for d in list(configured_project.datasets())[0].data_rows()]
     result = {}
 
     for idx, tool in enumerate(ontology['tools'] + ontology['classifications']):
@@ -134,7 +135,7 @@ def prediction_id_mapping(configured_project):
             "uuid": str(uuid.uuid4()),
             "schemaId": tool['featureSchemaId'],
             "dataRow": {
-                "id": datarows[idx].uid,
+                "id": configured_project.data_row_ids[idx],
             },
             'tool': tool
         }