diff --git a/test/audio_test_1.mp4 b/tests/audio_test_1.mp4 similarity index 100% rename from test/audio_test_1.mp4 rename to tests/audio_test_1.mp4 diff --git a/test/audio_test_2.mp4 b/tests/audio_test_2.mp4 similarity index 100% rename from test/audio_test_2.mp4 rename to tests/audio_test_2.mp4 diff --git a/test/test_audio.py b/tests/test_audio.py similarity index 100% rename from test/test_audio.py rename to tests/test_audio.py diff --git a/test/test_autotranscript.py b/tests/test_autotranscript.py similarity index 83% rename from test/test_autotranscript.py rename to tests/test_autotranscript.py index 78442b3..fbf18ab 100644 --- a/test/test_autotranscript.py +++ b/tests/test_autotranscript.py @@ -6,7 +6,7 @@ import os @pytest.fixture def create_scraibe_instance(): 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: return Scraibe() @@ -19,19 +19,19 @@ def test_scraibe_init(create_scraibe_instance): def test_scraibe_autotranscribe(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) def test_scraibe_diarization(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) def test_scraibe_transcribe(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) diff --git a/test/test_diarisation.py b/tests/test_diarisation.py similarity index 100% rename from test/test_diarisation.py rename to tests/test_diarisation.py diff --git a/tests/test_diarization.py b/tests/test_diarization.py deleted file mode 100644 index f9e81a5..0000000 --- a/tests/test_diarization.py +++ /dev/null @@ -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')) \ No newline at end of file diff --git a/test/test_transcriber.py b/tests/test_transcriber.py similarity index 95% rename from test/test_transcriber.py rename to tests/test_transcriber.py index 5bfe3cf..ba9d99a 100644 --- a/test/test_transcriber.py +++ b/tests/test_transcriber.py @@ -69,12 +69,12 @@ def test_get_whisper_kwargs(): def test_whisper_transcribe(whisper_instance): model = whisper_instance # 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) def test_faster_whisper_transcribe(faster_whisper_instance): model = faster_whisper_instance # 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)