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):
         | 
