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

only test autotranscript.py

Browse Source
This commit is contained in:
Tryndaron
2024-04-08 13:23:43 +02:00
parent eabd5734c2
commit c55c12db03
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/pytest.yaml
+1 -1
View File
@@ -39,5 +39,5 @@ jobs:
env: env:
HF_TOKEN : ${{ secrets.HF_TOKEN }} HF_TOKEN : ${{ secrets.HF_TOKEN }}
run: | run: |
pytest test/test_audio.py pytest test/test_autotrnscript.py
test/test_autotranscript.py
+4 -2
View File
@@ -11,6 +11,8 @@ def example_audio_file(audio_test_2.mp4):
""" """
@pytest.fixture @pytest.fixture
def create_scraibe_instance(): def create_scraibe_instance():
if "HF_TOKEN" in os.environ: if "HF_TOKEN" in os.environ:
@@ -27,7 +29,7 @@ def test_scraibe_init(create_scraibe_instance):
assert isinstance(model.diariser, Diariser) assert isinstance(model.diariser, Diariser)
def test_scraibe_autotranscribe(create_scraibe_instance, example_audio_file): """ def test_scraibe_autotranscribe(create_scraibe_instance, example_audio_file):
model = create_scraibe_instance model = create_scraibe_instance
transcript = model.autotranscribe(example_audio_file) transcript = model.autotranscribe(example_audio_file)
assert isinstance(transcript, Transcript) assert isinstance(transcript, Transcript)
@@ -58,4 +60,4 @@ def test_get_audio_file(create_scraibe_instance, example_audio_file):
audio_file = os.path.exist(example_audio_file) audio_file = os.path.exist(example_audio_file)
assert isinstance(audio_file, AudioProcessor) assert isinstance(audio_file, AudioProcessor)
assert isinstance(audio_file.waveform, torch.Tensor) assert isinstance(audio_file.waveform, torch.Tensor)
assert isinstance(audio_file.sr, torch.Tensor) assert isinstance(audio_file.sr, torch.Tensor) """