From 3d9f7d9f95f2d847e37ec5c347682aab8cfab465 Mon Sep 17 00:00:00 2001 From: Raivis Dejus Date: Tue, 24 Jun 2025 20:17:47 +0300 Subject: [PATCH] Test adjustment --- tests/transcriber/whisper_cpp_file_transcriber_test.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/transcriber/whisper_cpp_file_transcriber_test.py b/tests/transcriber/whisper_cpp_file_transcriber_test.py index 921f58d5..60556fb2 100644 --- a/tests/transcriber/whisper_cpp_file_transcriber_test.py +++ b/tests/transcriber/whisper_cpp_file_transcriber_test.py @@ -1,3 +1,4 @@ +import os from typing import List from unittest.mock import Mock from pathlib import Path @@ -18,7 +19,6 @@ from tests.audio import test_audio_path, test_multibyte_utf8_audio_path from tests.model_loader import get_model_path -# TODO Add test for CPU mode (BUZZ_FORCE_CPU=true) class TestWhisperCppFileTranscriber: @pytest.mark.parametrize( "word_level_timings,expected_segments", @@ -33,6 +33,7 @@ class TestWhisperCppFileTranscriber: def test_transcribe( self, qtbot: QtBot, word_level_timings: bool, expected_segments: List[Segment] ): + os.environ["BUZZ_FORCE_CPU"] = "true" file_transcription_options = FileTranscriptionOptions( file_paths=[str(Path(test_audio_path).resolve())] ) @@ -93,6 +94,7 @@ class TestWhisperCppFileTranscriber: def test_transcribe_latvian( self, qtbot: QtBot, word_level_timings: bool, expected_segments: List[Segment] ): + os.environ["BUZZ_FORCE_CPU"] = "true" file_transcription_options = FileTranscriptionOptions( file_paths=[str(Path(test_multibyte_utf8_audio_path).resolve())] )