adjusting tests

Author: timmarkhuff

test/integration/test_groundlight.py

@@ -859,7 +859,7 @@ def test_binary_detector(gl: Groundlight, detector_name: Callable):
     name = detector_name()
     created_detector = gl.create_binary_detector(name, "Is there a dog", confidence_threshold=0.0)
     assert created_detector is not None
-    binary_iq = gl.submit_image_query(created_detector, "test/assets/dog.jpeg")
+    binary_iq = gl.submit_image_query(created_detector, "test/assets/dog.jpeg", wait=0)
     assert binary_iq.result.label is not None
 
 
@@ -870,7 +870,7 @@ def test_counting_detector(gl: Groundlight, detector_name: Callable):
     name = detector_name()
     created_detector = gl.create_counting_detector(name, "How many dogs", "dog", confidence_threshold=0.0)
     assert created_detector is not None
-    count_iq = gl.submit_image_query(created_detector, "test/assets/dog.jpeg")
+    count_iq = gl.submit_image_query(created_detector, "test/assets/dog.jpeg", wait=0)
     assert count_iq.result.count is not None
 
 
@@ -903,7 +903,7 @@ def test_multiclass_detector(gl: Groundlight, detector_name: Callable):
         name, "What kind of dog is this?", class_names=class_names, confidence_threshold=0.0
     )
     assert created_detector is not None
-    mc_iq = gl.submit_image_query(created_detector, "test/assets/dog.jpeg")
+    mc_iq = gl.submit_image_query(created_detector, "test/assets/dog.jpeg", wait=0)
     assert mc_iq.result.label is not None
     assert mc_iq.result.label in class_names
 
@@ -928,7 +928,7 @@ def test_delete_detector(gl: Groundlight, detector_name: Callable):
     # Create another detector to test deletion by ID string and that an attached image query is deleted
     name2 = detector_name("Test delete detector 2")
     detector2 = gl.create_detector(name=name2, query=query, pipeline_config=pipeline_config)
-    gl.submit_image_query(detector2, "test/assets/dog.jpeg")
+    gl.submit_image_query(detector2, "test/assets/dog.jpeg", wait=0)
 
     time.sleep(10)
 

test/unit/test_cli.py

@@ -81,6 +81,8 @@ def test_detector_and_image_queries(detector_name: Callable):
             "submit-image-query",
             det_id_on_create,
             "test/assets/cat.jpeg",
+            "--wait",
+            "0",
         ],
         stdout=subprocess.PIPE,
         stderr=subprocess.PIPE,

test/unit/test_experimental.py

@@ -99,7 +99,7 @@ def test_text_recognition_detector(gl_experimental: ExperimentalApi, detector_na
         name, "What is the date and time?", confidence_threshold=0.0
     )
     assert created_detector is not None
-    mc_iq = gl_experimental.submit_image_query(created_detector, "test/assets/dog.jpeg")
+    mc_iq = gl_experimental.submit_image_query(created_detector, "test/assets/dog.jpeg", wait=0)
     assert mc_iq.result.text is not None
 
 
@@ -112,7 +112,7 @@ def test_bounding_box_detector(gl_experimental: ExperimentalApi, detector_name:
         name, "Draw a bounding box around each dog in the image", "dog", confidence_threshold=0.0
     )
     assert created_detector is not None
-    bbox_iq = gl_experimental.submit_image_query(created_detector, "test/assets/dog.jpeg")
+    bbox_iq = gl_experimental.submit_image_query(created_detector, "test/assets/dog.jpeg", wait=0)
     assert bbox_iq.result.label is not None
     assert bbox_iq.rois is not None
 

test/unit/test_images.py

@@ -7,6 +7,6 @@
 def test_get_image(gl_experimental: ExperimentalApi):
     name = f"Test {datetime.utcnow()}"
     det = gl_experimental.get_or_create_detector(name, "test_query")
-    iq = gl_experimental.submit_image_query(det, image="test/assets/dog.jpeg", wait=10)
+    iq = gl_experimental.submit_image_query(det, image="test/assets/dog.jpeg", wait=0)
     gl_experimental.get_image(iq.id)
     assert isinstance(PIL.Image.open(gl_experimental.get_image(iq.id)), PIL.Image.Image)

test/unit/test_labels.py

@@ -7,7 +7,7 @@
 def test_binary_labels(gl_experimental: ExperimentalApi):
     name = f"Test binary labels{datetime.utcnow()}"
     det = gl_experimental.create_detector(name, "test_query")
-    iq1 = gl_experimental.submit_image_query(det, "test/assets/cat.jpeg")
+    iq1 = gl_experimental.submit_image_query(det, "test/assets/cat.jpeg", wait=0)
     gl_experimental.add_label(iq1, "YES")
     iq1 = gl_experimental.get_image_query(iq1.id)
     assert iq1.result.label == "YES"
@@ -24,7 +24,7 @@ def test_binary_labels(gl_experimental: ExperimentalApi):
 def test_counting_labels(gl_experimental: ExperimentalApi):
     name = f"Test binary labels{datetime.utcnow()}"
     det = gl_experimental.create_counting_detector(name, "test_query", "test_object_class")
-    iq1 = gl_experimental.submit_image_query(det, "test/assets/cat.jpeg")
+    iq1 = gl_experimental.submit_image_query(det, "test/assets/cat.jpeg", wait=0)
 
     gl_experimental.add_label(iq1, 0)
     iq1 = gl_experimental.get_image_query(iq1.id)
@@ -48,7 +48,7 @@ def test_counting_labels(gl_experimental: ExperimentalApi):
 def test_multiclass_labels(gl_experimental: ExperimentalApi):
     name = f"Test binary labels{datetime.utcnow()}"
     det = gl_experimental.create_multiclass_detector(name, "test_query", class_names=["apple", "banana", "cherry"])
-    iq1 = gl_experimental.submit_image_query(det, "test/assets/cat.jpeg")
+    iq1 = gl_experimental.submit_image_query(det, "test/assets/cat.jpeg", wait=0)
     gl_experimental.add_label(iq1, "apple")
     iq1 = gl_experimental.get_image_query(iq1.id)
     assert iq1.result.label == "apple"
@@ -69,7 +69,7 @@ def test_multiclass_labels(gl_experimental: ExperimentalApi):
 def test_bounding_box_labels(gl_experimental: ExperimentalApi):
     name = f"Test bounding box labels{datetime.utcnow()}"
     det = gl_experimental.create_bounding_box_detector(name, "test_query", "test_class")
-    iq1 = gl_experimental.submit_image_query(det, "test/assets/cat.jpeg")
+    iq1 = gl_experimental.submit_image_query(det, "test/assets/cat.jpeg", wait=0)
     gl_experimental.add_label(iq1, "NO_OBJECTS")
     iq1 = gl_experimental.get_image_query(iq1.id)
     assert iq1.result.label == "NO_OBJECTS"
@@ -87,7 +87,7 @@ def test_bounding_box_labels(gl_experimental: ExperimentalApi):
 def test_text_recognition_labels(gl_experimental: ExperimentalApi):
     name = f"Test text recognition labels{datetime.utcnow()}"
     det = gl_experimental.create_text_recognition_detector(name, "test_query")
-    iq1 = gl_experimental.submit_image_query(det, "test/assets/cat.jpeg")
+    iq1 = gl_experimental.submit_image_query(det, "test/assets/cat.jpeg", wait=0)
     gl_experimental.add_label(iq1, "apple text")
     iq1 = gl_experimental.get_image_query(iq1.id)
     assert iq1.result.text == "apple text"