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

Add benchmarks

Browse source
This commit is contained in:
Chidi Williams 2023-04-25 09:45:13 +01:00
commit 9d3d52ed2a
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/transcriber_benchmarks_test.py
View file

@ -1,3 +1,4 @@
import platform
from unittest.mock import Mock
import pytest
@ -43,7 +44,10 @@ def transcribe(qtbot, transcriber: FileTranscriber):
pytest.param(
WhisperFileTranscriber(task=(get_task(
TranscriptionModel(model_type=ModelType.FASTER_WHISPER, whisper_model_size=WhisperModelSize.TINY)))),
id="Faster Whisper - Tiny"),
id="Faster Whisper - Tiny",
marks=pytest.mark.skipif(platform.system() == 'Darwin',
reason='Error with libiomp5 already initialized on GH action runner: https://github.com/chidiwilliams/buzz/actions/runs/4657331262/jobs/8241832087')
),
])
def test_should_transcribe_and_benchmark(qtbot, benchmark, transcriber):
segments = benchmark(transcribe, qtbot, transcriber)