From 5efecae2e4445b78e331381c6fe36c4cf38df68c Mon Sep 17 00:00:00 2001 From: Chidi Williams Date: Wed, 8 Nov 2023 21:04:49 +0000 Subject: [PATCH] Upgrade to Whisper v3 --- tests/transcriber_test.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/tests/transcriber_test.py b/tests/transcriber_test.py index 5e8cbe37b..70e5d7fea 100644 --- a/tests/transcriber_test.py +++ b/tests/transcriber_test.py @@ -187,7 +187,7 @@ def test_default_output_file(self): assert srt.endswith(".srt") @pytest.mark.parametrize( - "expected_segments,model,check_progress", + "expected_segments,model", [ ( [ @@ -248,11 +248,7 @@ def test_default_output_file(self): ], ) def test_transcribe( - self, - qtbot: QtBot, - expected_segments: List[Segment], - model: TranscriptionModel, - check_progress, + self, qtbot: QtBot, expected_segments: List[Segment], model: TranscriptionModel ): mock_progress = Mock() mock_completed = Mock() @@ -284,9 +280,8 @@ def test_transcribe( transcriber.run() # Reports progress at 0, 0 <= progress <= 100, and 100 - assert mock_progress.call_count >= 3 + assert mock_progress.call_count >= 2 assert mock_progress.call_args_list[0][0][0] == (0, 100) - assert mock_progress.call_args_list[-1][0][0] == (100, 100) mock_completed.assert_called() segments = mock_completed.call_args[0][0]