update to tests and error handling

Author: jtsodapop

labelbox/client.py

@@ -1014,17 +1014,14 @@ def assign_global_keys_to_data_rows(
                 if res['invalidGlobalKeyAssignments']:
                     errors.append("Invalid Global Keys: " +
                                   str(res['invalidGlobalKeyAssignments']))
-                if res['unmodifiedAssignments']:
-                    errors.append("Unmodified Assignments: " +
-                                  str(res['unmodifiedAssignments']))
                 if res['accessDeniedAssignments']:
                     errors.append("Access Denied Assignments: " +
                                   str(res['accessDeniedAssignments']))
                 if len(errors) > 0:
                     raise Exception(
                         "Failed to assign global keys to data rows: " +
                         str(errors))
-                return res['sanitizedAssignments']
+                return res['sanitizedAssignments'] + res['unmodifiedAssignments']
             time.sleep(2)
             timeout -= 2
 

tests/integration/test_assign_global_key_to_data_row.py

@@ -10,7 +10,7 @@ def test_assign_global_keys_to_data_rows(client, dataset, image_url):
 
     dr_1 = dataset.create_data_row(row_data=image_url, external_id="hello")
     dr_2 = dataset.create_data_row(row_data=image_url, external_id="world")
-    row_ids = [dr_1.uid, dr_2.uid]
+    row_ids = set([dr_1.uid, dr_2.uid])
 
     gk_1 = str(uuid.uuid4())
     gk_2 = str(uuid.uuid4())
@@ -24,5 +24,5 @@ def test_assign_global_keys_to_data_rows(client, dataset, image_url):
     res = client.get_data_row_ids_for_global_keys([gk_1, gk_2])
 
     assert len(res) == 2
-    for row in res:
-        assert row['id'] in row_ids
+    successful_assignments = set(a['id'] for a in res)
+    assert successful_assignments == row_ids