Silence tests in staging (#419)

cohere-ai · Mar 20, 2024 · 6064382 · 6064382
1 parent 889f389
commit 6064382
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/tests/test_async_client.py b/tests/test_async_client.py
@@ -60,12 +60,14 @@ async def test_moved_fn(self) -> None:
         with self.assertRaises(ValueError):
             await self.co.list_connectors("dummy", dummy="dummy")  # type: ignore
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     async def test_generate(self) -> None:
         response = await self.co.generate(
             prompt='Please explain to me how LLMs work',
         )
         print(response)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     async def test_embed(self) -> None:
         response = await self.co.embed(
             texts=['hello', 'goodbye'],
@@ -74,6 +76,7 @@ async def test_embed(self) -> None:
         )
         print(response)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     async def test_embed_job_crud(self) -> None:
         dataset = await self.co.datasets.create(
             name="test",
@@ -122,6 +125,7 @@ async def test_rerank(self) -> None:
 
         print(response)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     async def test_classify(self) -> None:
         examples = [
             ClassifyExample(text="Dermatologists don't like her!", label="Spam"),
@@ -149,6 +153,7 @@ async def test_classify(self) -> None:
         )
         print(response)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     async def test_datasets_crud(self) -> None:
         my_dataset = await self.co.datasets.create(
             name="test",
@@ -212,6 +217,7 @@ async def test_detokenize(self) -> None:
         )
         print(response)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     async def test_connectors_crud(self) -> None:
         created_connector = await self.co.connectors.create(
             name="Example connector",

diff --git a/tests/test_client.py b/tests/test_client.py
@@ -59,6 +59,7 @@ def test_moved_fn(self) -> None:
         with self.assertRaises(ValueError):
             co.list_connectors("dummy", dummy="dummy") # type: ignore
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     def test_generate(self) -> None:
         response = co.generate(
             prompt='Please explain to me how LLMs work',
@@ -73,6 +74,7 @@ def test_embed(self) -> None:
         )
         print(response)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     def test_embed_job_crud(self) -> None:
         dataset = co.datasets.create(
             name="test",
@@ -121,6 +123,7 @@ def test_rerank(self) -> None:
 
         print(response)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     def test_classify(self) -> None:
         examples = [
             ClassifyExample(text="Dermatologists don't like her!", label="Spam"),
@@ -148,6 +151,7 @@ def test_classify(self) -> None:
         )
         print(response)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     def test_datasets_crud(self) -> None:
         my_dataset = co.datasets.create(
             name="test",
@@ -211,6 +215,7 @@ def test_detokenize(self) -> None:
         )
         print(response)
 
+    @unittest.skipIf(os.getenv("CO_API_URL") is not None, "Doesn't work in staging.")
     def test_connectors_crud(self) -> None:
         created_connector = co.connectors.create(
             name="Example connector",