Improved test where

libs/labelbox/tests/integration/test_filtering.py

@@ -2,19 +2,38 @@
 
 from labelbox import Project
 from labelbox.exceptions import InvalidQueryError
-from labelbox.schema.queue_mode import QueueMode
+from labelbox import MediaType
+import time
+import logging
 
 
 @pytest.fixture
 def project_to_test_where(client, rand_gen):
+    num_retries = 5
+
     p_a_name = f"a-{rand_gen(str)}"
     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, 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)
+
+    while (num_retries > 0):
+        projects = client.get_projects()
+        try:
+            logging.log(level=2, msg=f"{len(list(projects))}")
+            if len(list(projects)) >= 3:
+                break
+            else:
+                raise TypeError()
+        except:
+            num_retries -= 1
+            time.sleep(5)
 
-    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)
-
+    if num_retries == 0:
+        raise TimeoutError("Hit max number of retries getting projects")
+
     yield p_a, p_b, p_c
 
     p_a.delete()