Upgrade to Whisper v3

chidiwilliams · Nov 8, 2023 · 5efecae · 5efecae
1 parent 35369c4
commit 5efecae
Showing 1 changed file with 3 additions and 8 deletions.
diff --git 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]