jhj0517 commited on
Commit
5f8203e
·
1 Parent(s): 44a8350

Fix pytest parameterize

Browse files
Files changed (1) hide show
  1. 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("whisper_type", "vad_filter", "bgm_separation", [
13
- ("whisper", False, False),
14
- ("faster-whisper", False, False),
15
- ("insanely_fast_whisper", False, False)
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