Update CI & test-suite.

zh-plus · Sep 3, 2024 · 0908a65 · 0908a65
1 parent ee867ca
commit 0908a65
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 17 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -53,7 +53,7 @@ jobs:
           poetry install
           poetry run pip uninstall -y faster-whisper
           poetry run pip install "faster-whisper @ https://github.com/SYSTRAN/faster-whisper/archive/d57c5b40b06e59ec44240d93485a95799548af50.tar.gz"
-          poetry run pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
+          poetry run pip install --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
 
       - name: Install ffmpeg
         uses: FedericoCarboni/setup-ffmpeg@v2

diff --git a/tests/test_transcribe.py b/tests/test_transcribe.py
@@ -27,19 +27,21 @@ class TestTranscriber(unittest.TestCase):
     def setUp(self) -> None:
         self.audio_path = Path('data/test_audio.wav')
 
-    def test_transcribe_success(self):
-        with patch('openlrc.transcribe.BatchedInferencePipeline') as MockModel:
-            MockModel.return_value.transcribe.return_value = return_tuple
-
-            transcriber = Transcriber(model_name='tiny')
-            result, info = transcriber.transcribe(self.audio_path)
-            self.assertIsNotNone(result)
-            self.assertEqual(round(info.duration), 30)
-
-    def test_audio_file_not_found(self):
-        with patch('openlrc.transcribe.BatchedInferencePipeline') as MockModel:
-            MockModel.return_value.transcribe.return_value = return_tuple
-
-            transcriber = Transcriber(model_name='tiny')
-            with self.assertRaises(FileNotFoundError):
-                transcriber.transcribe('audio.wav')
+    @patch('openlrc.transcribe.BatchedInferencePipeline')
+    @patch('openlrc.transcribe.WhisperModel')
+    def test_transcribe_success(self, MockWhisperModel, MockBatchedInferencePipeline):
+        MockBatchedInferencePipeline.return_value.transcribe.return_value = return_tuple
+
+        transcriber = Transcriber(model_name='tiny')
+        result, info = transcriber.transcribe(self.audio_path)
+        self.assertIsNotNone(result)
+        self.assertEqual(round(info.duration), 30)
+
+    @patch('openlrc.transcribe.BatchedInferencePipeline')
+    @patch('openlrc.transcribe.WhisperModel')
+    def test_audio_file_not_found(self, MockWhisperModel, MockBatchedInferencePipeline):
+        MockBatchedInferencePipeline.return_value.transcribe.return_value = return_tuple
+
+        transcriber = Transcriber(model_name='tiny')
+        with self.assertRaises(FileNotFoundError):
+            transcriber.transcribe('audio.wav')