small bulk job

Author: Matt Sokoloff

tests/integration/test_data_row_metadata.py

@@ -42,7 +42,7 @@ def big_dataset(dataset: Dataset, image_url):
             "row_data": image_url,
             "external_id": "my-image"
         },
-    ] * 250)
+    ] * 5)
     task.wait_till_done()
 
     yield dataset
@@ -92,10 +92,7 @@ def test_bulk_upsert_datarow_metadata(datarow, mdo: DataRowMetadataOntology):
     mdo.bulk_upsert([metadata])
     exported = mdo.bulk_export([datarow.uid])
     assert len(exported)
-    assert len([
-        field for field in exported[0].fields
-        if field.schema_id != PRE_COMPUTED_EMBEDDINGS_ID
-    ]) == 4
+    assert len([field for field in exported[0].fields if field.schema_id != PRE_COMPUTED_EMBEDDINGS_ID]) == 4
 
 
 @pytest.mark.slow
@@ -107,15 +104,13 @@ def test_large_bulk_upsert_datarow_metadata(big_dataset, mdo):
     errors = mdo.bulk_upsert(metadata)
     assert len(errors) == 0
 
+
     metadata_lookup = {
         metadata.data_row_id: metadata
         for metadata in mdo.bulk_export(data_row_ids)
     }
     for data_row_id in data_row_ids:
-        assert len([
-            f for f in metadata_lookup.get(data_row_id).fields
-            if f.schema_id != PRE_COMPUTED_EMBEDDINGS_ID
-        ]), metadata_lookup.get(data_row_id).fields
+        assert len([f for f in metadata_lookup.get(data_row_id).fields if f.schema_id != PRE_COMPUTED_EMBEDDINGS_ID]), metadata_lookup.get(data_row_id).fields
 
 
 def test_bulk_delete_datarow_metadata(datarow, mdo):
@@ -175,10 +170,7 @@ def test_large_bulk_delete_datarow_metadata(big_dataset, mdo):
     errors = mdo.bulk_delete(deletes)
     assert len(errors) == 0
     for data_row_id in data_row_ids:
-        fields = [
-            f for f in mdo.bulk_export([data_row_id])[0].fields
-            if f.schema_id != PRE_COMPUTED_EMBEDDINGS_ID
-        ]
+        fields = [f for f in mdo.bulk_export([data_row_id])[0].fields if f.schema_id != PRE_COMPUTED_EMBEDDINGS_ID]
         assert len(fields) == 1, fields
         assert EMBEDDING_SCHEMA_ID not in [field.schema_id for field in fields]
 
@@ -200,8 +192,7 @@ def test_bulk_delete_datarow_enum_metadata(datarow: DataRow, mdo):
         DeleteDataRowMetadata(data_row_id=datarow.uid, fields=[SPLIT_SCHEMA_ID])
     ])
     exported = mdo.bulk_export([datarow.uid])[0].fields
-    assert len(
-        [f for f in exported if f.schema_id != PRE_COMPUTED_EMBEDDINGS_ID]) == 0
+    assert len([f for f in exported if f.schema_id != PRE_COMPUTED_EMBEDDINGS_ID]) == 0
 
 
 def test_raise_enum_upsert_schema_error(datarow, mdo):