test: fix expected test value after changing audio token allowance

ref: N25B-209
This commit is contained in:
Twirre Meulenbelt
2025-11-05 12:41:48 +01:00
parent 689a0d35da
commit 1b58549c2a

View File

@@ -5,12 +5,13 @@ from control_backend.agents.transcription.speech_recognizer import OpenAIWhisper
def test_estimate_max_tokens():
"""Inputting one minute of audio, assuming 300 words per minute, expecting 400 tokens."""
"""Inputting one minute of audio, assuming 450 words per minute and adding a 10 token padding,
expecting 610 tokens."""
audio = np.empty(shape=(60 * 16_000), dtype=np.float32)
actual = SpeechRecognizer._estimate_max_tokens(audio)
assert actual == 400
assert actual == 610
assert isinstance(actual, int)