yapf

Author: Kevin Kim

tests/integration/test_global_keys.py

@@ -75,7 +75,7 @@ def test_assign_same_global_keys_to_data_rows(client, dataset, image_url):
     assert len(res['errors']) == 1
     assert res['errors'][0]['data_row_id'] == dr_2.uid
     assert res['errors'][0]['global_key'] == gk_1
-    assert res['errors'][0]['error'] == "Invalid global key"
+    assert res['errors'][0]['error'] == "Invalid Data Row or invalid global key"
 
 
 def test_global_key_sanitization(dataset, image_url):
@@ -111,7 +111,7 @@ def test_long_global_key_validation(client, dataset, image_url):
     assert res['results'][0]['global_key'] == gk_1
     assert res['errors'][0]['data_row_id'] == dr_2.uid
     assert res['errors'][0]['global_key'] == gk_2
-    assert res['errors'][0]['error'] == 'Invalid global key'
+    assert res['errors'][0]['error'] == 'Invalid Data Row or invalid global key'
 
 
 def test_global_key_with_whitespaces_validation(client, dataset, image_url):
@@ -143,8 +143,11 @@ def test_global_key_with_whitespaces_validation(client, dataset, image_url):
     assign_errors_msgs = set([e['error'] for e in res['errors']])
     assert assign_errors_ids == set([dr_1.uid, dr_2.uid, dr_3.uid])
     assert assign_errors_gks == set([gk_1, gk_2, gk_3])
-    assert assign_errors_msgs == set(
-        ['Invalid global key', 'Invalid global key', 'Invalid global key'])
+    assert assign_errors_msgs == set([
+        'Invalid Data Row or invalid global key',
+        'Invalid Data Row or invalid global key',
+        'Invalid Data Row or invalid global key'
+    ])
 
 
 @pytest.mark.skip(reason='get_data_rows_for_global_keys not included in '