cjones/WhisperLive
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update vad tests after refactor

Browse Source
This commit is contained in:
makaveli10
2024-02-15 12:16:39 +05:30
parent 7bcb92c266
commit 4db91eed66
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_vad.py
+7 -8
View File
@@ -1,13 +1,12 @@
import unittest import unittest
import numpy as np import numpy as np
import torch
from whisper_live.tensorrt_utils import load_audio from whisper_live.tensorrt_utils import load_audio
from whisper_live.vad import VoiceActivityDetection from whisper_live.vad import VoiceActivityDetector
class TestVoiceActivityDetection(unittest.TestCase): class TestVoiceActivityDetection(unittest.TestCase):
def setUp(self): def setUp(self):
self.vad = VoiceActivityDetection() self.vad = VoiceActivityDetector()
self.sample_rate = 16000 self.sample_rate = 16000
def generate_silence(self, duration_seconds): def generate_silence(self, duration_seconds):
@@ -18,10 +17,10 @@ class TestVoiceActivityDetection(unittest.TestCase):
def test_vad_silence_detection(self): def test_vad_silence_detection(self):
silence = self.generate_silence(3) silence = self.generate_silence(3)
speech_prob = self.vad(torch.from_numpy(silence.copy()), self.sample_rate).item() is_speech_present = self.vad(silence.copy())
self.assertLess(speech_prob, 0.5, "VAD incorrectly identified silence as speech.") self.assertFalse(is_speech_present, "VAD incorrectly identified silence as speech.")
def test_vad_speech_detection(self): def test_vad_speech_detection(self):
audio_tensor = torch.from_numpy(load_audio("assets/jfk.flac")) audio_tensor = load_audio("assets/jfk.flac")
speech_prob = self.vad(audio_tensor, self.sample_rate).item() is_speech_present = self.vad(audio_tensor)
self.assertGreater(speech_prob, 0.5, "VAD failed to identify speech segment.") self.assertTrue(is_speech_present, "VAD failed to identify speech segment.")