/
test_webrtcvad.py
106 lines (95 loc) · 3.7 KB
/
test_webrtcvad.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
import unittest
import wave
from memory_profiler import memory_usage
import webrtcvad
class WebRtcVadTests(unittest.TestCase):
@staticmethod
def _load_wave(file_name):
fp = wave.open(file_name, 'rb')
try:
assert fp.getnchannels() == 1, (
'{0}: sound format is incorrect! Sound must be mono.'.format(
file_name))
assert fp.getsampwidth() == 2, (
'{0}: sound format is incorrect! '
'Sample width of sound must be 2 bytes.').format(file_name)
assert fp.getframerate() in (8000, 16000, 32000), (
'{0}: sound format is incorrect! '
'Sampling frequency must be 8000 Hz, 16000 Hz or 32000 Hz.')
sampling_frequency = fp.getframerate()
sound_data = fp.readframes(fp.getnframes())
finally:
fp.close()
del fp
return sound_data, sampling_frequency
def test_constructor(self):
vad = webrtcvad.Vad()
def test_set_mode(self):
vad = webrtcvad.Vad()
vad.set_mode(0)
vad.set_mode(1)
vad.set_mode(2)
vad.set_mode(3)
self.assertRaises(
ValueError,
vad.set_mode, 4)
def test_valid_rate_and_frame_length(self):
self.assertTrue(webrtcvad.valid_rate_and_frame_length(8000, 160))
self.assertTrue(webrtcvad.valid_rate_and_frame_length(16000, 160))
self.assertFalse(webrtcvad.valid_rate_and_frame_length(32000, 160))
self.assertRaises(
(ValueError, OverflowError),
webrtcvad.valid_rate_and_frame_length, 2 ** 35, 10)
def test_process_zeroes(self):
frame_len = 160
self.assertTrue(
webrtcvad.valid_rate_and_frame_length(8000, frame_len))
sample = b'\x00' * frame_len * 2
vad = webrtcvad.Vad()
self.assertFalse(vad.is_speech(sample, 16000))
def test_process_file(self):
with open('test-audio.raw', 'rb') as f:
data = f.read()
frame_ms = 30
n = int(8000 * 2 * 30 / 1000.0)
frame_len = int(n / 2)
self.assertTrue(
webrtcvad.valid_rate_and_frame_length(8000, frame_len))
chunks = list(data[pos:pos + n] for pos in range(0, len(data), n))
if len(chunks[-1]) != n:
chunks = chunks[:-1]
expecteds = [
'011110111111111111111111111100',
'011110111111111111111111111100',
'000000111111111111111111110000',
'000000111111111111111100000000'
]
for mode in (0, 1, 2, 3):
vad = webrtcvad.Vad(mode)
result = ''
for chunk in chunks:
voiced = vad.is_speech(chunk, 8000)
result += '1' if voiced else '0'
self.assertEqual(expecteds[mode], result)
def test_leak(self):
sound, fs = self._load_wave('leak-test.wav')
frame_ms = 0.010
frame_len = int(round(fs * frame_ms))
n = int(len(sound) / (2 * frame_len))
nrepeats = 1000
vad = webrtcvad.Vad(3)
used_memory_before = memory_usage(-1)[0]
for counter in range(nrepeats):
find_voice = False
for frame_ind in range(n):
slice_start = (frame_ind * 2 * frame_len)
slice_end = ((frame_ind + 1) * 2 * frame_len)
if vad.is_speech(sound[slice_start:slice_end], fs):
find_voice = True
self.assertTrue(find_voice)
used_memory_after = memory_usage(-1)[0]
self.assertGreaterEqual(
used_memory_before / 5.0,
used_memory_after - used_memory_before)
if __name__ == '__main__':
unittest.main(verbosity=2)