Added media_type to test projects

Author: Val Brodsky

libs/labelbox/tests/data/export/conftest.py

@@ -1,9 +1,12 @@
-import uuid
 import time
+import uuid
+
 import pytest
-from labelbox.schema.queue_mode import QueueMode
+
+from labelbox.schema.annotation_import import AnnotationImportState, LabelImport
 from labelbox.schema.labeling_frontend import LabelingFrontend
-from labelbox.schema.annotation_import import LabelImport, AnnotationImportState
+from labelbox.schema.media_type import MediaType
+from labelbox.schema.queue_mode import QueueMode
 
 
 @pytest.fixture
@@ -247,6 +250,7 @@ def configured_project_with_ontology(
     project = client.create_project(
         name=rand_gen(str),
         queue_mode=QueueMode.Batch,
+        media_type=MediaType.Image,
     )
     editor = list(
         client.get_labeling_frontends(where=LabelingFrontend.name == "editor")
@@ -274,6 +278,7 @@ def configured_project_without_data_rows(
         name=rand_gen(str),
         description=rand_gen(str),
         queue_mode=QueueMode.Batch,
+        media_type=MediaType.Image,
     )
     editor = list(
         client.get_labeling_frontends(where=LabelingFrontend.name == "editor")

libs/labelbox/tests/integration/test_client_errors.py

@@ -8,6 +8,7 @@
 
 import labelbox.client
 from labelbox import Project, User
+from labelbox.schema.media_type import MediaType
 
 
 def test_missing_api_key():
@@ -29,7 +30,7 @@ def test_bad_key(rand_gen):
     client = labelbox.client.Client(api_key=bad_key)
 
     with pytest.raises(lbox.exceptions.AuthenticationError) as excinfo:
-        client.create_project(name=rand_gen(str))
+        client.create_project(name=rand_gen(str), media_type=MediaType.Image)
 
 
 def test_syntax_error(client):
@@ -77,7 +78,7 @@ def test_network_error(client):
     )
 
     with pytest.raises(lbox.exceptions.NetworkError) as excinfo:
-        client.create_project(name="Project name")
+        client.create_project(name="Project name", media_type=MediaType.Image)
 
 
 def test_invalid_attribute_error(
@@ -86,12 +87,12 @@ def test_invalid_attribute_error(
 ):
     # Creation
     with pytest.raises(lbox.exceptions.InvalidAttributeError) as excinfo:
-        client.create_project(name="Name", invalid_field="Whatever")
+        client.create_project(name="Name", invalid_field="Whatever", media_type=MediaType.Image)
     assert excinfo.value.db_object_type == Project
     assert excinfo.value.field == "invalid_field"
 
     # Update
-    project = client.create_project(name=rand_gen(str))
+    project = client.create_project(name=rand_gen(str), media_type=MediaType.Image)
     with pytest.raises(lbox.exceptions.InvalidAttributeError) as excinfo:
         project.update(invalid_field="Whatever")
     assert excinfo.value.db_object_type == Project

libs/labelbox/tests/integration/test_filtering.py

@@ -1,7 +1,9 @@
 import pytest
 from lbox.exceptions import InvalidQueryError
+from libs.labelbox.src.labelbox.schema.media_type import MediaType
 
 from labelbox import Project
+from labelbox.schema import media_type
 from labelbox.schema.queue_mode import QueueMode
 
 
@@ -11,9 +13,9 @@ def project_to_test_where(client, rand_gen):
     p_b_name = f"b-{rand_gen(str)}"
     p_c_name = f"c-{rand_gen(str)}"
 
-    p_a = client.create_project(name=p_a_name, queue_mode=QueueMode.Batch)
-    p_b = client.create_project(name=p_b_name, queue_mode=QueueMode.Batch)
-    p_c = client.create_project(name=p_c_name, queue_mode=QueueMode.Batch)
+    p_a = client.create_project(name=p_a_name, media_type=MediaType.Image)
+    p_b = client.create_project(name=p_b_name, media_type=MediaType.Image)
+    p_c = client.create_project(name=p_c_name, media_type=MediaType.Image)
 
     yield p_a, p_b, p_c
 

libs/labelbox/tests/integration/test_legacy_project.py

@@ -1,5 +1,7 @@
 import pytest
 
+from labelbox.schema import media_type
+from labelbox.schema.media_type import MediaType
 from labelbox.schema.queue_mode import QueueMode
 
 
@@ -11,6 +13,7 @@ def test_project_dataset(client, rand_gen):
         client.create_project(
             name=rand_gen(str),
             queue_mode=QueueMode.Dataset,
+            media_type=MediaType.Image,
         )
 
 
@@ -19,13 +22,13 @@ def test_project_auto_audit_parameters(client, rand_gen):
         ValueError,
         match="quality_modes must be set instead of auto_audit_percentage or auto_audit_number_of_labels.",
     ):
-        client.create_project(name=rand_gen(str), auto_audit_percentage=0.5)
+        client.create_project(name=rand_gen(str), media_type=MediaType.Image,  auto_audit_percentage=0.5)
 
     with pytest.raises(
         ValueError,
         match="quality_modes must be set instead of auto_audit_percentage or auto_audit_number_of_labels.",
     ):
-        client.create_project(name=rand_gen(str), auto_audit_number_of_labels=2)
+        client.create_project(name=rand_gen(str), media_type=MediaType.Image, auto_audit_number_of_labels=2)
 
 
 def test_project_name_parameter(client, rand_gen):