FIX: small test case fixes

Author: gdj0nes

labelbox/schema/data_row_metadata.py

@@ -39,30 +39,30 @@ class DataRowMetadataSchema(BaseModel):
 DataRowMetadataValue = Union[Embedding, DateTime, String, OptionId]
 
 
-class CamelCaseMixin(BaseModel):
+class _CamelCaseMixin(BaseModel):
 
     class Config:
         allow_population_by_field_name = True
         alias_generator = camel_case
 
 
 # Metadata base class
-class DataRowMetadataField(CamelCaseMixin):
+class DataRowMetadataField(_CamelCaseMixin):
     schema_id: SchemaId
     value: DataRowMetadataValue
 
 
-class DataRowMetadata(CamelCaseMixin):
+class DataRowMetadata(_CamelCaseMixin):
     data_row_id: str
     fields: List[DataRowMetadataField]
 
 
-class DeleteDataRowMetadata(CamelCaseMixin):
+class DeleteDataRowMetadata(_CamelCaseMixin):
     data_row_id: str
     fields: List[SchemaId]
 
 
-class DataRowMetadataBatchResponse(CamelCaseMixin):
+class DataRowMetadataBatchResponse(_CamelCaseMixin):
     data_row_id: str
     error: str
     fields: List[Union[DataRowMetadataField, SchemaId]]
@@ -73,18 +73,18 @@ class DataRowMetadataBatchResponse(CamelCaseMixin):
 
 
 # Bulk upsert values
-class _UpsertDataRowMetadataInput(CamelCaseMixin):
+class _UpsertDataRowMetadataInput(_CamelCaseMixin):
     schema_id: str
     value: Union[str, List, dict]
 
 
 # Batch of upsert values for a datarow
-class _UpsertBatchDataRowMetadata(CamelCaseMixin):
+class _UpsertBatchDataRowMetadata(_CamelCaseMixin):
     data_row_id: str
     fields: List[_UpsertDataRowMetadataInput]
 
 
-class _DeleteBatchDataRowMetadata(CamelCaseMixin):
+class _DeleteBatchDataRowMetadata(_CamelCaseMixin):
     data_row_id: str
     schema_ids: List[SchemaId]
 
@@ -238,7 +238,7 @@ def bulk_upsert(
             An empty list means the upload was successful.
         """
 
-        if not (len(metadata)):
+        if not len(metadata):
             raise ValueError("Empty list passed")
 
         def _batch_upsert(

tests/integration/test_data_row_metadata.py

@@ -102,6 +102,7 @@ def test_bulk_delete_datarow_metadata(datarow, mdo):
     assert not len(datarow.metadata["fields"])
 
     metadata = make_metadata(datarow.uid)
+    mdo.bulk_upsert([metadata])
 
     assert len(datarow.metadata["fields"])
 
@@ -228,25 +229,25 @@ def test_large_bulk_delete_non_existent_schema_id(big_dataset, mdo):
 
 def test_parse_raw_metadata(mdo):
     example = {
-        'dataRowId':
+        'data_row_id':
             'ckr6kkfx801ui0yrtg9fje8xh',
         'fields': [{
-            'schemaId': 'cko8s9r5v0001h2dk9elqdidh',
+            'schema_id': 'cko8s9r5v0001h2dk9elqdidh',
             'value': 'my-new-message'
         }, {
-            'schemaId': 'cko8sbczn0002h2dkdaxb5kal',
+            'schema_id': 'cko8sbczn0002h2dkdaxb5kal',
             'value': {}
         }, {
-            'schemaId': 'cko8sbscr0003h2dk04w86hof',
+            'schema_id': 'cko8sbscr0003h2dk04w86hof',
             'value': {}
         }, {
-            'schemaId': 'cko8sdzv70006h2dk8jg64zvb',
+            'schema_id': 'cko8sdzv70006h2dk8jg64zvb',
             'value': '2021-07-20T21:41:14.606710Z'
         }]
     }
 
     parsed = mdo.parse_metadata([example])
     assert len(parsed) == 1
     row = parsed[0]
-    assert row.data_row_id == example["dataRowId"]
+    assert row.data_row_id == example["data_row_id"]
     assert len(row.fields) == 3