Merge pull request #108 from Labelbox/gs/tests-fixes

Fixes in tests

Author: msokoloff1

.github/workflows/python-package.yml

@@ -73,4 +73,4 @@ jobs:
         # [email protected]
         LABELBOX_TEST_API_KEY_STAGING: ${{ secrets.STAGING_LABELBOX_API_KEY }}
       run: |
-        tox -e py -- -svv
+        tox -e py -- -svvx

tests/integration/test_client_errors.py

@@ -43,9 +43,17 @@ def test_semantic_error(client):
     assert excinfo.value.message.startswith("Cannot query field \"bbb\"")
 
 
-def test_timeout_error(client):
+def test_timeout_error(client, project):
+    time.sleep(60)  #Fails to connect if we don't wait
     with pytest.raises(labelbox.exceptions.TimeoutError) as excinfo:
-        client.execute("{projects {id}}", check_naming=False, timeout=0.001)
+        query_str = """query getOntology { 
+        project (where: {id: $%s}) { 
+            ontology { 
+                normalized 
+                } 
+            }
+        } """ % (project.uid)
+        client.execute(query_str, check_naming=False, timeout=0.01)
 
 
 def test_query_complexity_error(client):

tests/integration/test_ontology.py

@@ -1,5 +1,6 @@
 import unittest
 from typing import Any, Dict, List, Union
+from labelbox import LabelingFrontend
 
 
 def sample_ontology() -> Dict[str, Any]:
@@ -33,7 +34,9 @@ def sample_ontology() -> Dict[str, Any]:
 
 def test_create_ontology(client, project) -> None:
     """ Tests that the ontology that a project was set up with can be grabbed."""
-    frontend = list(client.get_labeling_frontends())[0]
+    frontend = list(
+        client.get_labeling_frontends(
+            where=LabelingFrontend.name == "Editor"))[0]
     project.setup(frontend, sample_ontology())
     normalized_ontology = project.ontology().normalized