feat: Update PYTGPT_TEST_AUDIO to PYTGPT_TEST_MEDIA in test cases

fix: Remove deleted test files (test_gemini.py, test_imager.py)

Author: Simatwa

tests/test_api.py

@@ -72,8 +72,8 @@ def test_text_stream(self):
         self.assertTrue(resp.is_success)
 
     @unittest.skipUnless(
-        os.getenv("PYTGPT_TEST_AUDIO", "") == "true",
-        "PYTGPT_TEST_AUDIO environment variable is not set to 'true' ",
+        os.getenv("PYTGPT_TEST_MEDIA", "") == "true",
+        "PYTGPT_TEST_MEDIA environment variable is not set to 'true' ",
     )
     def test_prompt_to_image_post(self):
         resp = self.client.post(
@@ -90,8 +90,8 @@ def test_prompt_to_image_post(self):
         self.assertEqual(len(resp_dict["urls"]), 2)
 
     @unittest.skipUnless(
-        os.getenv("PYTGPT_TEST_AUDIO", "") == "true",
-        "PYTGPT_TEST_AUDIO environment variable is not set to 'true' ",
+        os.getenv("PYTGPT_TEST_MEDIA", "") == "true",
+        "PYTGPT_TEST_MEDIA environment variable is not set to 'true' ",
     )
     def test_prompt_to_image_bytes_post(self):
         resp = self.client.post(
@@ -100,8 +100,8 @@ def test_prompt_to_image_bytes_post(self):
         self.assertIsNotNone(resp.headers.get("Content-Disposition"))
 
     @unittest.skipUnless(
-        os.getenv("PYTGPT_TEST_AUDIO", "") == "true",
-        "PYTGPT_TEST_AUDIO environment variable is not set to 'true' ",
+        os.getenv("PYTGPT_TEST_MEDIA", "") == "true",
+        "PYTGPT_TEST_MEDIA environment variable is not set to 'true' ",
     )
     def test_prompt_to_image_bytes_get(self):
         resp = self.client.get(
@@ -110,8 +110,8 @@ def test_prompt_to_image_bytes_get(self):
         self.assertIsNotNone(resp.headers.get("Content-Disposition"))
 
     @unittest.skipUnless(
-        os.getenv("PYTGPT_TEST_AUDIO", "") == "true",
-        "PYTGPT_TEST_AUDIO environment variable is not set to 'true' ",
+        os.getenv("PYTGPT_TEST_MEDIA", "") == "true",
+        "PYTGPT_TEST_MEDIA environment variable is not set to 'true' ",
     )
     def test_prompt_to_image_bytes_get_redirect(self):
         resp = self.client.get(

tests/test_gemini.py

@@ -2,6 +2,7 @@
 from typing import List
 from typing import Generator
 import unittest
+import os
 
 
 class TestImager(unittest.TestCase):
@@ -10,12 +11,20 @@ def setUp(self):
         self.imager = imager.Imager()
         self.prompt: str = "hello world"
 
+    @unittest.skipUnless(
+        os.getenv("PYTGPT_TEST_MEDIA", "") == "true",
+        "PYTGPT_TEST_MEDIA environment variable is not set to 'true' ",
+    )
     def test_non_stream(self):
         """Test one photo generation"""
         img = self.imager.generate(self.prompt)
         self.assertIsInstance(img, List), "Image not generated"
         self.assertIsInstance(img[0], bytes), "Image not generated"
 
+    @unittest.skipUnless(
+        os.getenv("PYTGPT_TEST_MEDIA", "") == "true",
+        "PYTGPT_TEST_MEDIA environment variable is not set to 'true' ",
+    )
     def test_stream(self):
         """Test multiple photo generation"""
         generator = self.imager.generate(self.prompt, amount=2, stream=True)

tests/test_imager.py tests/test_imager_tgpt.pytests/test_imager.py renamed to tests/test_imager_tgpt.py

@@ -131,8 +131,8 @@ def setUp(self):
         self.text = "This is a speech synthesis test"
 
     @unittest.skipUnless(
-        os.getenv("PYTGPT_TEST_AUDIO", "") == "true",
-        "PYTGPT_TEST_AUDIO environment variable is not set to 'true' ",
+        os.getenv("PYTGPT_TEST_MEDIA", "") == "true",
+        "PYTGPT_TEST_MEDIA environment variable is not set to 'true' ",
     )
     def test_text_to_audio(self):
         """Speech synthesis"""
@@ -142,8 +142,8 @@ def test_text_to_audio(self):
         self.assertIs(type(voice_bytes), bytes)
 
     @unittest.skipUnless(
-        os.getenv("PYTGPT_TEST_AUDIO", "") == "true",
-        "PYTGPT_TEST_AUDIO environment variable is not set to 'true' ",
+        os.getenv("PYTGPT_TEST_MEDIA", "") == "true",
+        "PYTGPT_TEST_MEDIA environment variable is not set to 'true' ",
     )
     def test_text_to_audio_save_to(self):
         """Save speech to a file"""