Spaces:
Running
Running
jhj0517
commited on
Commit
·
5f8203e
1
Parent(s):
44a8350
Fix pytest parameterize
Browse files- tests/test_transcription.py +11 -5
tests/test_transcription.py
CHANGED
@@ -9,11 +9,14 @@ import gradio as gr
|
|
9 |
import os
|
10 |
|
11 |
|
12 |
-
@pytest.mark.parametrize(
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
|
|
|
|
|
|
17 |
def test_transcribe(
|
18 |
whisper_type: str,
|
19 |
vad_filter: bool,
|
@@ -70,6 +73,9 @@ def test_transcribe(
|
|
70 |
|
71 |
|
72 |
def download_file(url, save_dir):
|
|
|
|
|
|
|
73 |
if not os.path.exists(save_dir):
|
74 |
os.makedirs(save_dir)
|
75 |
|
|
|
9 |
import os
|
10 |
|
11 |
|
12 |
+
@pytest.mark.parametrize(
|
13 |
+
"whisper_type,vad_filter,bgm_separation",
|
14 |
+
[
|
15 |
+
("whisper", False, False),
|
16 |
+
("faster-whisper", False, False),
|
17 |
+
("insanely_fast_whisper", False, False)
|
18 |
+
]
|
19 |
+
)
|
20 |
def test_transcribe(
|
21 |
whisper_type: str,
|
22 |
vad_filter: bool,
|
|
|
73 |
|
74 |
|
75 |
def download_file(url, save_dir):
|
76 |
+
if os.path.exists(TEST_FILE_PATH):
|
77 |
+
return
|
78 |
+
|
79 |
if not os.path.exists(save_dir):
|
80 |
os.makedirs(save_dir)
|
81 |
|