Merge pull request #730 from Labelbox/eg/fixing-tests

Fixing data row import tests

Author: tytalus

tests/integration/test_data_rows.py

@@ -352,7 +352,7 @@ def test_create_data_rows_with_invalid_metadata(dataset, image_url):
         DataRow.metadata_fields: fields
     }])
     task.wait_till_done()
-    assert task.status == "COMPLETE"
+    assert task.status == "FAILED"
     assert len(task.failed_data_rows) > 0
 
 
@@ -633,9 +633,10 @@ def test_data_row_bulk_creation_with_same_global_keys(dataset, sample_image):
     }])
 
     task.wait_till_done()
-    assert task.status == "COMPLETE"
+    assert task.status == "FAILED"
     assert len(task.failed_data_rows) > 0
     assert len(list(dataset.data_rows())) == 0
+    assert task.errors == "Import job failed"
 
     task = dataset.create_data_rows([{
         DataRow.row_data: sample_image,

tests/integration/test_task.py

@@ -1,6 +1,6 @@
 import pytest
 
-from labelbox import DataRow, Task
+from labelbox import DataRow
 from labelbox.schema.data_row_metadata import DataRowMetadataField
 
 EMBEDDING_SCHEMA_ID = "ckpyije740000yxdk81pbgjdc"
@@ -21,13 +21,12 @@ def test_task_errors(dataset, image_url):
             ]
         },
     ])
+
     assert task in client.get_user().created_tasks()
     task.wait_till_done()
-    assert task.status == "COMPLETE"
+    assert task.status == "FAILED"
     assert len(task.failed_data_rows) > 0
     assert task.errors is not None
-    assert 'message' in task.errors[0]
-    assert len(task.result) == 0
 
 
 def test_task_success_json(dataset, image_url):