admin/scribe
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Merge pull request #129 from JSchmie/fix-pytest-action-2

Browse Source 
added relative Path to pytests
This commit is contained in:
Jacob Schmieder
2024-10-01 11:08:38 +02:00
committed by GitHub
parent df9c5109f3 81fb9af461
commit a4b8546033
7 changed files with 6 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/audio_test_1.mp4 → tests/audio_test_1.mp4
View File
test/audio_test_2.mp4 → tests/audio_test_2.mp4
View File
test/test_audio.py → tests/test_audio.py
View File
test/test_autotranscript.py → tests/test_autotranscript.py
+4 -4
View File
@@ -6,7 +6,7 @@ import os
@pytest.fixture @pytest.fixture
def create_scraibe_instance(): def create_scraibe_instance():
if "HF_TOKEN" in os.environ: if "HF_TOKEN" in os.environ:
return Scraibe(use_auth_token=os.environ["HF_TOKEN"]) return Scraibe(use_auth_token=os.environ["HF_TOKEN"], whisper_model= "tiny")
else: else:
return Scraibe() return Scraibe()
@@ -19,19 +19,19 @@ def test_scraibe_init(create_scraibe_instance):
def test_scraibe_autotranscribe(create_scraibe_instance): def test_scraibe_autotranscribe(create_scraibe_instance):
model = create_scraibe_instance model = create_scraibe_instance
transcript = model.autotranscribe('test/audio_test_2.mp4') transcript = model.autotranscribe('tests/audio_test_2.mp4')
assert isinstance(transcript, Transcript) assert isinstance(transcript, Transcript)
def test_scraibe_diarization(create_scraibe_instance): def test_scraibe_diarization(create_scraibe_instance):
model = create_scraibe_instance model = create_scraibe_instance
diarisation_result = model.diarization('test/audio_test_2.mp4') diarisation_result = model.diarization('tests/audio_test_2.mp4')
assert isinstance(diarisation_result, dict) assert isinstance(diarisation_result, dict)
def test_scraibe_transcribe(create_scraibe_instance): def test_scraibe_transcribe(create_scraibe_instance):
model = create_scraibe_instance model = create_scraibe_instance
transcription_result = model.transcribe('test/audio_test_2.mp4') transcription_result = model.transcribe('tests/audio_test_2.mp4')
assert isinstance(transcription_result, str) assert isinstance(transcription_result, str)
test/test_diarisation.py → tests/test_diarisation.py
View File
tests/test_diarization.py
-10
View File
@@ -1,10 +0,0 @@
from os import environ
environ["AUTOT_CACHE"] = "/mnt/disk1/Projekte/ScrAIbe/tests"
# environ["PYANNOTE_CACHE"] = "/mnt/disk1/Projekte/ScrAIbe/tests/pyannote"
# environ["TORCH_HOME"] = "/mnt/disk1/Projekte/ScrAIbe/tests/torch"
from scraibe import Scraibe
scraibe = Scraibe(whisper_type = "faster-whisper", whisper_model = "tiny")
print(scraibe.autotranscribe('/mnt/disk1/Projekte/ScrAIbe/test/audio_test_1.mp4'))
test/test_transcriber.py → tests/test_transcriber.py
+2 -2
View File
@@ -69,12 +69,12 @@ def test_get_whisper_kwargs():
def test_whisper_transcribe(whisper_instance): def test_whisper_transcribe(whisper_instance):
model = whisper_instance model = whisper_instance
# mocker.patch.object(transcriber_instance.model, 'transcribe', return_value={'Hello, World !'} ) # mocker.patch.object(transcriber_instance.model, 'transcribe', return_value={'Hello, World !'} )
transcript = model.transcribe('test/audio_test_2.mp4') transcript = model.transcribe('tests/audio_test_2.mp4')
assert isinstance(transcript, str) assert isinstance(transcript, str)
def test_faster_whisper_transcribe(faster_whisper_instance): def test_faster_whisper_transcribe(faster_whisper_instance):
model = faster_whisper_instance model = faster_whisper_instance
# mocker.patch.object(transcriber_instance.model, 'transcribe', return_value={'Hello, World !'} ) # mocker.patch.object(transcriber_instance.model, 'transcribe', return_value={'Hello, World !'} )
transcript = model.transcribe('test/audio_test_2.mp4') transcript = model.transcribe('tests/audio_test_2.mp4')
assert isinstance(transcript, str) assert isinstance(transcript, str)