ADD: test cases

Author: gdj0nes

labelbox/schema/batch.py

@@ -22,7 +22,6 @@ class Batch(DbObject):
     name = Field.String("name")
     created_at = Field.DateTime("created_at")
     updated_at = Field.DateTime("updated_at")
-    deleted = Field.Boolean()
     size = Field.Int("size")
 
     # Relationships

tests/integration/test_batch.py

@@ -8,12 +8,13 @@
 
 @pytest.fixture
 def big_dataset(dataset: Dataset):
-    task = dataset.create_data_rows([
-        {
-            "row_data": IMAGE_URL,
-            "external_id": "my-image"
-        },
-    ] * 250)
+    task = dataset.create_data_rows(
+        [
+            {
+                "row_data": IMAGE_URL,
+                "external_id": "my-image"
+            },
+        ] * 250)
     task.wait_till_done()
 
     yield dataset
@@ -24,4 +25,6 @@ def test_create_batch(configured_project: Project, big_dataset: Dataset):
     configured_project.update(queue_mode=QueueMode.Batch)
 
     data_rows = [dr.uid for dr in list(big_dataset.export_data_rows())]
-    queue_res = configured_project.create_batch("test-batch", data_rows, 3)
+    batch = configured_project.create_batch("test-batch", data_rows, 3)
+    assert batch.name == 'test-batch'
+    assert batch.size == len(data_rows)