Fix unit test
Browse files- tests/vad_test.py +3 -3
tests/vad_test.py
CHANGED
@@ -5,7 +5,7 @@ import sys
|
|
5 |
|
6 |
sys.path.append('../whisper-webui')
|
7 |
|
8 |
-
from vad import AbstractTranscription, VadSileroTranscription
|
9 |
|
10 |
class TestVad(unittest.TestCase):
|
11 |
def __init__(self, *args, **kwargs):
|
@@ -24,8 +24,8 @@ class TestVad(unittest.TestCase):
|
|
24 |
])
|
25 |
|
26 |
self.assertListEqual(result['segments'],
|
27 |
-
[{'end': 50.0, 'start': 40.0, 'text': 'Hello world'},
|
28 |
-
{'end': 120.0, 'start': 110.0, 'text': 'Hello world'}]
|
29 |
)
|
30 |
|
31 |
def transcribe_segments(self, segment):
|
|
|
5 |
|
6 |
sys.path.append('../whisper-webui')
|
7 |
|
8 |
+
from src.vad import AbstractTranscription, VadSileroTranscription
|
9 |
|
10 |
class TestVad(unittest.TestCase):
|
11 |
def __init__(self, *args, **kwargs):
|
|
|
24 |
])
|
25 |
|
26 |
self.assertListEqual(result['segments'],
|
27 |
+
[{'end': 50.0, 'start': 40.0, 'text': 'Hello world '},
|
28 |
+
{'end': 120.0, 'start': 110.0, 'text': 'Hello world '}]
|
29 |
)
|
30 |
|
31 |
def transcribe_segments(self, segment):
|