aadnk commited on
Commit
25df8a0
1 Parent(s): b1d4eff

Fix unit test

Browse files
Files changed (1) hide show
  1. 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):