Update whisper test to match current behavior

descriptinc · Oct 23, 2024 · 101f5ce · 101f5ce
1 parent 13c61ce
commit 101f5ce
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/core/test_whisper.py b/tests/core/test_whisper.py
@@ -26,8 +26,7 @@ def test_whisper_transcript():
 
     transcript = signal.get_whisper_transcript()
 
-    assert "<|startoftranscript|>" in transcript
-    assert "<|endoftext|>" in transcript
+    assert transcript.startswith("<|startoftranscript|>")
 
 
 def test_whisper_embeddings():