Skip to content

Commit

Permalink
tests: adjust result according to new param changes
Browse files Browse the repository at this point in the history
  • Loading branch information
winstxnhdw authored Jul 12, 2024
1 parent c21c9ab commit 2134dbb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_transcribe.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,11 +18,11 @@ def test_transcribe_srt(client: TestClient):
with open('tests/test.mp3', 'rb') as file:
response = client.post('/v1/transcribe', files={'request': file}, params={'caption_format': 'srt'}).json()

assert response['result'] == '1\n00:00:00,000 --> 00:00:01,700\nHello there. My name is Bella.'
assert response['result'] == '1\n00:00:00,000 --> 00:00:01,720\nHello there, my name is Bella.'


def test_transcribe_vtt(client: TestClient):
with open('tests/test.mp3', 'rb') as file:
response = client.post('/v1/transcribe', files={'request': file}, params={'caption_format': 'vtt'}).json()

assert response['result'] == 'WEBVTT\n\n00:00:00.000 --> 00:00:01.700\nHello there. My name is Bella.'
assert response['result'] == 'WEBVTT\n\n00:00:00.000 --> 00:00:01.720\nHello there, ,y name is Bella.'

0 comments on commit 2134dbb

Please sign in to comment.