Adds audio translation test

defenseunicorns · Jul 3, 2024 · 0b2fe7b · 0b2fe7b
1 parent 20a2171
commit 0b2fe7b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/tests/data/arabic-audio.wav b/tests/data/arabic-audio.wav
diff --git a/tests/e2e/test_whisper.py b/tests/e2e/test_whisper.py
@@ -46,3 +46,12 @@ def test_transcriptions():
 
     assert len(transcription.text) > 0  # The transcription should not be empty
     assert len(transcription.text) < 500  # The transcription should not be too long
+
+
+def test_translations():
+    translation = client.audio.translations.create(
+        model="whisper", file=Path("tests/data/arabic-audio.wav")
+    )
+
+    assert len(translation.text) > 0  # The translation should not be empty
+    assert len(translation.text) < 500  # The translation should not be too long