@@ -11,47 +11,48 @@ def test_audio_transcribe_success():
1111
1212 assert len (response ["text" ]) > 0
1313
14- def test_audio_transcribe_timestamps_success ():
15- test_client = PredictionGuard ()
16-
17- response = test_client .audio .transcriptions .create (
18- model = "base" ,
19- file = "fixtures/test_audio.wav" ,
20- timestamp_granularities = ["word" , "segment" ],
21- response_format = "verbose_json"
22- )
23-
24- assert len (response ["text" ]) > 0
25- assert len (response ["segments" ]) > 0
26- assert len (response ["segments" ][0 ]["text" ]) > 0
27- assert len (response ["words" ]) > 0
28- assert len (response ["words" ][0 ]["text" ]) > 0
29-
30- def test_audio_transcribe_diarization_success ():
31- test_client = PredictionGuard ()
32-
33- response = test_client .audio .transcriptions .create (
34- model = "base" ,
35- file = "fixtures/test_audio.wav" ,
36- diarization = True ,
37- response_format = "verbose_json"
38- )
39-
40- assert len (response ["text" ]) > 0
41- assert len (response ["segments" ]) > 0
42- assert len (response ["segments" ][0 ]["text" ]) > 0
43-
44- def test_audio_transcribe_diarization_timestamps_success ():
45- test_client = PredictionGuard ()
46-
47- response = test_client .audio .transcriptions .create (
48- model = "base" ,
49- file = "fixtures/test_audio.wav" ,
50- diarization = True ,
51- timestamp_granularities = ["word" ],
52- response_format = "verbose_json"
53- )
54-
55- assert len (response ["text" ]) > 0
56- assert len (response ["words" ]) > 0
57- assert len (response ["words" ][0 ]["text" ]) > 0
14+ # Uncomment when in production
15+ # def test_audio_transcribe_timestamps_success():
16+ # test_client = PredictionGuard()
17+ #
18+ # response = test_client.audio.transcriptions.create(
19+ # model="base",
20+ # file="fixtures/test_audio.wav",
21+ # timestamp_granularities=["word", "segment"],
22+ # response_format="verbose_json"
23+ # )
24+ #
25+ # assert len(response["text"]) > 0
26+ # assert len(response["segments"]) > 0
27+ # assert len(response["segments"][0]["text"]) > 0
28+ # assert len(response["words"]) > 0
29+ # assert len(response["words"][0]["text"]) > 0
30+ #
31+ # def test_audio_transcribe_diarization_success():
32+ # test_client = PredictionGuard()
33+ #
34+ # response = test_client.audio.transcriptions.create(
35+ # model="base",
36+ # file="fixtures/test_audio.wav",
37+ # diarization=True,
38+ # response_format="verbose_json"
39+ # )
40+ #
41+ # assert len(response["text"]) > 0
42+ # assert len(response["segments"]) > 0
43+ # assert len(response["segments"][0]["text"]) > 0
44+ #
45+ # def test_audio_transcribe_diarization_timestamps_success():
46+ # test_client = PredictionGuard()
47+ #
48+ # response = test_client.audio.transcriptions.create(
49+ # model="base",
50+ # file="fixtures/test_audio.wav",
51+ # diarization=True,
52+ # timestamp_granularities=["word"],
53+ # response_format="verbose_json"
54+ # )
55+ #
56+ # assert len(response["text"]) > 0
57+ # assert len(response["words"]) > 0
58+ # assert len(response["words"][0]["text"]) > 0
0 commit comments