deblan-mirror/buzz
1
0
Fork 0
mirror of https://github.com/chidiwilliams/buzz.git synced 2026-03-14 14:45:46 +01:00
Code Issues Projects Releases Packages Wiki Activity

Test adjustment

Browse source
This commit is contained in:
Raivis Dejus 2025-06-24 20:17:47 +03:00
commit 3d9f7d9f95
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/transcriber/whisper_cpp_file_transcriber_test.py
View file

@ -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())]
)