1+ import os
12import unittest
23from fastapi .testclient import TestClient
34from fastapi import status
@@ -70,6 +71,10 @@ def test_text_stream(self):
7071 )
7172 self .assertTrue (resp .is_success )
7273
74+ @unittest .skipUnless (
75+ os .getenv ('PYTGPT_TEST_AUDIO' , '' ) == "true" ,
76+ "PYTGPT_TEST_AUDIO environment variable is not set to 'true' "
77+ )
7378 def test_prompt_to_image_post (self ):
7479 resp = self .client .post (
7580 "/v1/image" ,
@@ -84,18 +89,30 @@ def test_prompt_to_image_post(self):
8489 self .assertIsNotNone (resp_dict .get ("urls" ))
8590 self .assertEqual (len (resp_dict ["urls" ]), 2 )
8691
92+ @unittest .skipUnless (
93+ os .getenv ('PYTGPT_TEST_AUDIO' , '' ) == "true" ,
94+ "PYTGPT_TEST_AUDIO environment variable is not set to 'true' "
95+ )
8796 def test_prompt_to_image_bytes_post (self ):
8897 resp = self .client .post (
8998 "/v1/image/bytes" , json = {"prompt" : "Jay Z performing" , "timeout" : 30 }
9099 )
91100 self .assertIsNotNone (resp .headers .get ("Content-Disposition" ))
92101
102+ @unittest .skipUnless (
103+ os .getenv ('PYTGPT_TEST_AUDIO' , '' ) == "true" ,
104+ "PYTGPT_TEST_AUDIO environment variable is not set to 'true' "
105+ )
93106 def test_prompt_to_image_bytes_get (self ):
94107 resp = self .client .get (
95108 "/v1/image/bytes" , params = {"prompt" : "Jay Z performing" , "timeout" : 30 }
96109 )
97110 self .assertIsNotNone (resp .headers .get ("Content-Disposition" ))
98111
112+ @unittest .skipUnless (
113+ os .getenv ('PYTGPT_TEST_AUDIO' , '' ) == "true" ,
114+ "PYTGPT_TEST_AUDIO environment variable is not set to 'true' "
115+ )
99116 def test_prompt_to_image_bytes_get_redirect (self ):
100117 resp = self .client .get (
101118 "/v1/image/bytes" ,
0 commit comments