def setUp(self):
self.signal = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789abcdefgh"
self.audio_source = BufferAudioSource(data_buffer=self.signal,
sampling_rate=11,
sample_width=4,
channels=1)
self.audio_source.open()
def test_set_readonly_short_properties_exception(self):
data = ""
sampling_rate = 8000
sample_width = 2
channels = 1
a_source = BufferAudioSource(data, sampling_rate, sample_width,
channels)
with self.assertRaises(AttributeError):
a_source.sr = 16000
a_source.sw = 1
a_source.ch = 2
def test_Limiter_Overlap_Deco_read(self):
block_size = 256
hop_size = 200
ads = ADSFactory.ads(audio_source=self.audio_source, max_time=0.50, block_size=block_size, hop_size=hop_size)
# Read all available data overlapping blocks
ads.open()
ads_data = []
while True:
block = ads.read()
if block is None:
break
ads_data.append(block)
ads.close()
# Read all data from file and build a BufferAudioSource
fp = wave.open(dataset.one_to_six_arabic_16000_mono_bc_noise, "r")
wave_data = fp.readframes(fp.getnframes())
fp.close()
audio_source = BufferAudioSource(wave_data, ads.get_sampling_rate(),
ads.get_sample_width(), ads.get_channels())
audio_source.open()
# Compare all blocks read from OverlapADS to those read
# from an audio source with a manual set_position
for i,block in enumerate(ads_data):
tmp = audio_source.read(block_size)
self.assertEqual(block, tmp, "Unexpected block (N={0}) read from OverlapADS".format(i))
audio_source.set_position((i+1) * hop_size)
audio_source.close()
def test_Limiter_Recorder_Overlap_Deco_rewind_and_read_alias(self):
# Use arbitrary valid block_size and hop_size
block_size = 5
hop_size = 4
ads = ADSFactory.ads(db=self.signal,
sr=16,
sw=2,
ch=1,
mt=0.80,
bs=block_size,
hs=hop_size,
rec=True)
# Read all available data overlapping blocks
ads.open()
i = 0
while True:
block = ads.read()
if block is None:
break
i += 1
ads.rewind()
# Build a BufferAudioSource
audio_source = BufferAudioSource(self.signal, ads.get_sampling_rate(),
ads.get_sample_width(),
ads.get_channels())
audio_source.open()
# Compare all blocks read from OverlapADS to those read
# from an audio source with a manual set_position
for j in range(i):
tmp = audio_source.read(block_size)
block = ads.read()
self.assertEqual(
block, tmp,
"Unexpected block (N={0}) read from OverlapADS".format(i))
audio_source.set_position((j + 1) * hop_size)
ads.close()
audio_source.close()
def test_Limiter_Recorder_Overlap_Deco_rewind_and_read(self):
# Use arbitrary valid block_size and hop_size
block_size = 5
hop_size = 4
ads = ADSFactory.ads(
data_buffer=self.signal,
sampling_rate=16,
sample_width=2,
channels=1,
max_time=0.80,
block_size=block_size,
hop_size=hop_size,
record=True,
)
# Read all available data overlapping blocks
ads.open()
i = 0
while True:
block = ads.read()
if block is None:
break
i += 1
ads.rewind()
# Build a BufferAudioSource
audio_source = BufferAudioSource(self.signal, ads.sampling_rate,
ads.sample_width, ads.channels)
audio_source.open()
# Compare all blocks read from OverlapADS to those read
# from an audio source with a manual position setting
for j in range(i):
tmp = audio_source.read(block_size)
block = ads.read()
self.assertEqual(
block,
tmp,
"Unexpected block '{}' (N={}) read from OverlapADS".format(
block, i),
)
audio_source.position = (j + 1) * hop_size
ads.close()
audio_source.close()
def test_read_short_properties(self):
data = ""
sampling_rate = 8000
sample_width = 2
channels = 1
a_source = BufferAudioSource(data, sampling_rate, sample_width,
channels)
self.assertEqual(a_source.sr, sampling_rate)
self.assertEqual(a_source.sw, sample_width)
self.assertEqual(a_source.ch, channels)
def test_Limiter_Recorder_Overlap_Deco_rewind_and_read(self):
# Use arbitrary valid block_size and hop_size
block_size = 1600
hop_size = 400
ads = ADSFactory.ads(audio_source=self.audio_source,
max_time=1.50,
block_size=block_size,
hop_size=hop_size,
record=True)
# Read all available data overlapping blocks
ads.open()
i = 0
while True:
block = ads.read()
if block is None:
break
i += 1
ads.rewind()
# Read all data from file and build a BufferAudioSource
fp = wave.open(dataset.one_to_six_arabic_16000_mono_bc_noise, "r")
wave_data = fp.readframes(fp.getnframes())
fp.close()
audio_source = BufferAudioSource(wave_data, ads.get_sampling_rate(),
ads.get_sample_width(),
ads.get_channels())
audio_source.open()
# Compare all blocks read from OverlapADS to those read
# from an audio source with a manual set_position
for j in range(i):
tmp = audio_source.read(block_size)
self.assertEqual(
ads.read(), tmp,
"Unexpected block (N={0}) read from OverlapADS".format(i))
audio_source.set_position((j + 1) * hop_size)
ads.close()
audio_source.close()
def _detect(asegment):
asource = BufferAudioSource(data_buffer=asegment,
sampling_rate=FRAME_RATE,
sample_width=bytes_per_frame,
channels=1)
ads = ADSFactory.ads(audio_source=asource, block_dur=1./sample_rate)
ads.open()
tokens = tokenizer.tokenize(ads)
length = (len(asegment)//bytes_per_frame + frames_per_window - 1)//frames_per_window
media_bstring = np.zeros(length+1, dtype=int)
for token in tokens:
media_bstring[token[1]] += 1
media_bstring[token[2]+1] -= 1
return (np.cumsum(media_bstring)[:-1] > 0)
def test_Overlap_Deco_read(self):
# Use arbitrary valid block_size and hop_size
block_size = 1714
hop_size = 313
ads = ADSFactory.ads(audio_source=self.audio_source,
block_size=block_size,
hop_size=hop_size)
# Read all available data overlapping blocks
ads.open()
ads_data = []
while True:
block = ads.read()
if block is None:
break
ads_data.append(block)
ads.close()
# Read all data from file and build a BufferAudioSource
fp = wave.open(dataset.one_to_six_arabic_16000_mono_bc_noise, "r")
wave_data = fp.readframes(fp.getnframes())
fp.close()
audio_source = BufferAudioSource(wave_data, ads.get_sampling_rate(),
ads.get_sample_width(),
ads.get_channels())
audio_source.open()
# Compare all blocks read from OverlapADS to those read
# from an audio source with a manual set_position
for i, block in enumerate(ads_data):
tmp = audio_source.read(block_size)
self.assertEqual(
block, tmp,
"Unexpected block (N={0}) read from OverlapADS".format(i))
audio_source.set_position((i + 1) * hop_size)
audio_source.close()
def _detect(asegment):
asource = BufferAudioSource(data_buffer=asegment,
sampling_rate=frame_rate,
sample_width=bytes_per_frame,
channels=1)
ads = ADSFactory.ads(audio_source=asource, block_dur=1. / sample_rate)
ads.open()
tokens = tokenizer.tokenize(ads)
length = (len(asegment) // bytes_per_frame + frames_per_window -
1) // frames_per_window
media_bstring = np.zeros(length + 1)
for token in tokens:
media_bstring[token[1]] = 1.
media_bstring[token[2] + 1] = non_speech_label - 1.
return np.clip(np.cumsum(media_bstring)[:-1], 0., 1.)
def test_Limiter_Overlap_Deco_read(self):
block_size = 256
hop_size = 200
ads = ADSFactory.ads(
audio_source=self.audio_source,
max_time=0.50,
block_size=block_size,
hop_size=hop_size,
)
# Read all available data overlapping blocks
ads.open()
ads_data = []
while True:
block = ads.read()
if block is None:
break
ads_data.append(block)
ads.close()
# Read all data from file and build a BufferAudioSource
fp = wave.open(dataset.one_to_six_arabic_16000_mono_bc_noise, "r")
wave_data = fp.readframes(fp.getnframes())
fp.close()
audio_source = BufferAudioSource(wave_data, ads.sampling_rate,
ads.sample_width, ads.channels)
audio_source.open()
# Compare all blocks read from OverlapADS to those read
# from an audio source with a manual position setting
for i, block in enumerate(ads_data):
tmp = audio_source.read(len(block) // (ads.sw * ads.ch))
self.assertEqual(
len(block),
len(tmp),
"Unexpected block (N={0}) read from OverlapADS".format(i),
)
audio_source.position = (i + 1) * hop_size
audio_source.close()
def test_Limiter_Recorder_Overlap_Deco_rewind_and_read_alias(self):
# Use arbitrary valid block_size and hop_size
block_size = 5
hop_size = 4
ads = ADSFactory.ads(db=self.signal, sr=16,
sw=2, ch=1, mt = 0.80,
bs=block_size, hs=hop_size,
rec=True)
# Read all available data overlapping blocks
ads.open()
i = 0
while True:
block = ads.read()
if block is None:
break
i += 1
ads.rewind()
# Build a BufferAudioSource
audio_source = BufferAudioSource(self.signal, ads.get_sampling_rate(),
ads.get_sample_width(), ads.get_channels())
audio_source.open()
# Compare all blocks read from OverlapADS to those read
# from an audio source with a manual set_position
for j in range(i):
tmp = audio_source.read(block_size)
block = ads.read()
self.assertEqual(block, tmp, "Unexpected block (N={0}) read from OverlapADS".format(i))
audio_source.set_position((j+1) * hop_size)
ads.close()
audio_source.close()
def test_Limiter_Recorder_Overlap_Deco_rewind_and_read(self):
# Use arbitrary valid block_size and hop_size
block_size = 1600
hop_size = 400
ads = ADSFactory.ads(audio_source=self.audio_source, max_time = 1.50, block_size=block_size, hop_size=hop_size, record=True)
# Read all available data overlapping blocks
ads.open()
i = 0
while True:
block = ads.read()
if block is None:
break
i += 1
ads.rewind()
# Read all data from file and build a BufferAudioSource
fp = wave.open(dataset.one_to_six_arabic_16000_mono_bc_noise, "r")
wave_data = fp.readframes(fp.getnframes())
fp.close()
audio_source = BufferAudioSource(wave_data, ads.get_sampling_rate(),
ads.get_sample_width(), ads.get_channels())
audio_source.open()
# Compare all blocks read from OverlapADS to those read
# from an audio source with a manual set_position
for j in range(i):
tmp = audio_source.read(block_size)
self.assertEqual(ads.read(), tmp, "Unexpected block (N={0}) read from OverlapADS".format(i))
audio_source.set_position((j+1) * hop_size)
ads.close()
audio_source.close()
def setUp(self):
self.signal = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789abcdefgh"
self.audio_source = BufferAudioSource(data_buffer=self.signal,
sampling_rate=11, sample_width=4, channels=1)
self.audio_source.open()
class TestBufferAudioSource_SR11_SW4_CH1(unittest.TestCase):
def setUp(self):
self.signal = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789abcdefgh"
self.audio_source = BufferAudioSource(data_buffer=self.signal,
sampling_rate=11, sample_width=4, channels=1)
self.audio_source.open()
def tearDown(self):
self.audio_source.close()
def test_sr11_sw4_ch1_read_1(self):
block = self.audio_source.read(1)
self.assertEqual(block, "ABCD", msg="wrong block, expected: 'ABCD', found: {0} ".format(block))
def test_sr11_sw4_ch1_read_6(self):
block = self.audio_source.read(6)
self.assertEqual(block, "ABCDEFGHIJKLMNOPQRSTUVWX", msg="wrong block, expected: 'ABCDEFGHIJKLMNOPQRSTUVWX', found: {0} ".format(block))
def test_sr11_sw4_ch1_read_multiple(self):
block = self.audio_source.read(1)
self.assertEqual(block, "ABCD", msg="wrong block, expected: 'AB', found: {0} ".format(block))
block = self.audio_source.read(6)
self.assertEqual(block, "EFGHIJKLMNOPQRSTUVWXYZ01", msg="wrong block, expected: 'EFGHIJKLMNOPQRSTUVWXYZ01', found: {0} ".format(block))
block = self.audio_source.read(3)
self.assertEqual(block, "23456789abcd", msg="wrong block, expected: '23456789abcd', found: {0} ".format(block))
block = self.audio_source.read(9999)
self.assertEqual(block, "efgh", msg="wrong block, expected: 'efgh', found: {0} ".format(block))
def test_sr11_sw4_ch1_read_all(self):
block = self.audio_source.read(9999)
self.assertEqual(block, self.signal, msg="wrong block, expected: {0}, found: {1} ".format(self.signal, block))
block = self.audio_source.read(1)
self.assertEqual(block, None, msg="wrong block, expected: {0}, found: {1} ".format(None, block))
def test_sr11_sw4_ch1_get_sampling_rate(self):
srate = self.audio_source.get_sampling_rate()
self.assertEqual(srate, 11, msg="wrong sampling rate, expected: 10, found: {0} ".format(srate))
def test_sr11_sw4_ch1_get_sample_width(self):
swidth = self.audio_source.get_sample_width()
self.assertEqual(swidth, 4, msg="wrong sample width, expected: 1, found: {0} ".format(swidth))
def test_sr11_sw4_ch1_get_channels(self):
channels = self.audio_source.get_channels()
self.assertEqual(channels, 1, msg="wrong number of channels, expected: 1, found: {0} ".format(channels))
def test_sr11_sw4_ch1_get_position_0(self):
pos = self.audio_source.get_position()
self.assertEqual(pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr11_sw4_ch1_get_position_5(self):
self.audio_source.read(5)
pos = self.audio_source.get_position()
self.assertEqual(pos, 5, msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr11_sw4_ch1_get_position_9(self):
self.audio_source.read(5)
self.audio_source.read(4)
pos = self.audio_source.get_position()
self.assertEqual(pos, 9, msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr11_sw4_ch1_set_position_0(self):
self.audio_source.read(10)
self.audio_source.set_position(0)
pos = self.audio_source.get_position()
self.assertEqual(pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr11_sw4_ch1_set_position_10(self):
self.audio_source.set_position(10)
pos = self.audio_source.get_position()
self.assertEqual(pos, 10, msg="wrong position, expected: 10, found: {0} ".format(pos))
def test_sr11_sw4_ch1_get_time_position_0(self):
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 0.0, msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_get_time_position_1(self):
srate = self.audio_source.get_sampling_rate()
# read one second
self.audio_source.read(srate)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 1.0, msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_get_time_position_0_63(self):
# read 2.5 seconds
self.audio_source.read(7)
tp = self.audio_source.get_time_position()
self.assertAlmostEqual(tp, 0.636363636364, msg="wrong time position, expected: 0.636363636364, found: {0} ".format(tp))
def test_sr11_sw4_ch1_set_time_position_0(self):
self.audio_source.read(10)
self.audio_source.set_time_position(0)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 0.0, msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_set_time_position_1(self):
self.audio_source.set_time_position(1)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 1.0, msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_set_time_position_end(self):
self.audio_source.set_time_position(100)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 1.0, msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_rewind(self):
self.audio_source.read(10)
self.audio_source.rewind()
tp = self.audio_source.get_position()
self.assertEqual(tp, 0, msg="wrong position, expected: 0.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_set_data(self):
self.audio_source.set_data("abcdefgh")
block = self.audio_source.read(9999)
self.assertEqual(block, "abcdefgh", msg="wrong block, expected: 'abcdef', found: {0} ".format(block))
def test_sr11_sw4_ch1_set_data_exception(self):
with self.assertRaises(Exception):
self.assertRaises(ValueError, self.audio_source.set_data("abcdef"))
class TestBufferAudioSource_SR16_SW2_CH1(unittest.TestCase):
def setUp(self):
self.signal = "ABCDEFGHIJKLMNOPQRSTUVWXYZ012345"
self.audio_source = BufferAudioSource(data_buffer=self.signal,
sampling_rate=16, sample_width=2, channels=1)
self.audio_source.open()
def tearDown(self):
self.audio_source.close()
def test_sr16_sw2_ch1_read_1(self):
block = self.audio_source.read(1)
self.assertEqual(block, "AB", msg="wrong block, expected: 'AB', found: {0} ".format(block))
def test_sr16_sw2_ch1_read_6(self):
block = self.audio_source.read(6)
self.assertEqual(block, "ABCDEFGHIJKL", msg="wrong block, expected: 'ABCDEFGHIJKL', found: {0} ".format(block))
def test_sr16_sw2_ch1_read_multiple(self):
block = self.audio_source.read(1)
self.assertEqual(block, "AB", msg="wrong block, expected: 'AB', found: {0} ".format(block))
block = self.audio_source.read(6)
self.assertEqual(block, "CDEFGHIJKLMN", msg="wrong block, expected: 'CDEFGHIJKLMN', found: {0} ".format(block))
block = self.audio_source.read(5)
self.assertEqual(block, "OPQRSTUVWX", msg="wrong block, expected: 'OPQRSTUVWX', found: {0} ".format(block))
block = self.audio_source.read(9999)
self.assertEqual(block, "YZ012345", msg="wrong block, expected: 'YZ012345', found: {0} ".format(block))
def test_sr16_sw2_ch1_read_all(self):
block = self.audio_source.read(9999)
self.assertEqual(block, self.signal, msg="wrong block, expected: {0}, found: {1} ".format(self.signal, block))
block = self.audio_source.read(1)
self.assertEqual(block, None, msg="wrong block, expected: {0}, found: {1} ".format(None, block))
def test_sr16_sw2_ch1_get_sampling_rate(self):
srate = self.audio_source.get_sampling_rate()
self.assertEqual(srate, 16, msg="wrong sampling rate, expected: 10, found: {0} ".format(srate))
def test_sr16_sw2_ch1_get_sample_width(self):
swidth = self.audio_source.get_sample_width()
self.assertEqual(swidth, 2, msg="wrong sample width, expected: 1, found: {0} ".format(swidth))
def test_sr16_sw2_ch1_get_channels(self):
channels = self.audio_source.get_channels()
self.assertEqual(channels, 1, msg="wrong number of channels, expected: 1, found: {0} ".format(channels))
def test_sr16_sw2_ch1_get_position_0(self):
pos = self.audio_source.get_position()
self.assertEqual(pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr16_sw2_ch1_get_position_5(self):
self.audio_source.read(5)
pos = self.audio_source.get_position()
self.assertEqual(pos, 5, msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr16_sw2_ch1_get_position_15(self):
self.audio_source.read(5)
self.audio_source.read(10)
pos = self.audio_source.get_position()
self.assertEqual(pos, 15, msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr16_sw2_ch1_set_position_0(self):
self.audio_source.read(10)
self.audio_source.set_position(0)
pos = self.audio_source.get_position()
self.assertEqual(pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr16_sw2_ch1_set_position_10(self):
self.audio_source.set_position(10)
pos = self.audio_source.get_position()
self.assertEqual(pos, 10, msg="wrong position, expected: 10, found: {0} ".format(pos))
def test_sr16_sw2_ch1_get_time_position_0(self):
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 0.0, msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_get_time_position_1(self):
srate = self.audio_source.get_sampling_rate()
# read one second
self.audio_source.read(srate)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 1.0, msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_get_time_position_0_75(self):
# read 2.5 seconds
self.audio_source.read(12)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 0.75, msg="wrong time position, expected: 0.75, found: {0} ".format(tp))
def test_sr16_sw2_ch1_set_time_position_0(self):
self.audio_source.read(10)
self.audio_source.set_time_position(0)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 0.0, msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_set_time_position_1(self):
self.audio_source.set_time_position(1)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 1.0, msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_set_time_position_end(self):
self.audio_source.set_time_position(100)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 1.0, msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_rewind(self):
self.audio_source.read(10)
self.audio_source.rewind()
tp = self.audio_source.get_position()
self.assertEqual(tp, 0, msg="wrong position, expected: 0.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_set_data(self):
self.audio_source.set_data("abcdef")
block = self.audio_source.read(9999)
self.assertEqual(block, "abcdef", msg="wrong block, expected: 'abcdef', found: {0} ".format(block))
def test_sr16_sw2_ch1_set_data_exception(self):
with self.assertRaises(Exception):
self.assertRaises(ValueError, self.audio_source.set_data("abcde"))
class TestBufferAudioSource_SR10_SW1_CH1(unittest.TestCase):
def setUp(self):
self.signal = "ABCDEFGHIJKLMNOPQRSTUVWXYZ012345"
self.audio_source = BufferAudioSource(data_buffer=self.signal,
sampling_rate=10, sample_width=1, channels=1)
self.audio_source.open()
def tearDown(self):
self.audio_source.close()
def test_sr10_sw1_ch1_read_1(self):
block = self.audio_source.read(1)
self.assertEqual(block, "A", msg="wrong block, expected: 'A', found: {0} ".format(block))
def test_sr10_sw1_ch1_read_6(self):
block = self.audio_source.read(6)
self.assertEqual(block, "ABCDEF", msg="wrong block, expected: 'ABCDEF', found: {0} ".format(block))
def test_sr10_sw1_ch1_read_multiple(self):
block = self.audio_source.read(1)
self.assertEqual(block, "A", msg="wrong block, expected: 'A', found: {0} ".format(block))
block = self.audio_source.read(6)
self.assertEqual(block, "BCDEFG", msg="wrong block, expected: 'BCDEFG', found: {0} ".format(block))
block = self.audio_source.read(13)
self.assertEqual(block, "HIJKLMNOPQRST", msg="wrong block, expected: 'HIJKLMNOPQRST', found: {0} ".format(block))
block = self.audio_source.read(9999)
self.assertEqual(block, "UVWXYZ012345", msg="wrong block, expected: 'UVWXYZ012345', found: {0} ".format(block))
def test_sr10_sw1_ch1_read_all(self):
block = self.audio_source.read(9999)
self.assertEqual(block, self.signal, msg="wrong block, expected: {0}, found: {1} ".format(self.signal, block))
block = self.audio_source.read(1)
self.assertEqual(block, None, msg="wrong block, expected: {0}, found: {1} ".format(None, block))
def test_sr10_sw1_ch1_get_sampling_rate(self):
srate = self.audio_source.get_sampling_rate()
self.assertEqual(srate, 10, msg="wrong sampling rate, expected: 10, found: {0} ".format(srate))
def test_sr10_sw1_ch1_get_sample_width(self):
swidth = self.audio_source.get_sample_width()
self.assertEqual(swidth, 1, msg="wrong sample width, expected: 1, found: {0} ".format(swidth))
def test_sr10_sw1_ch1_get_channels(self):
channels = self.audio_source.get_channels()
self.assertEqual(channels, 1, msg="wrong number of channels, expected: 1, found: {0} ".format(channels))
def test_sr10_sw1_ch1_get_position_0(self):
pos = self.audio_source.get_position()
self.assertEqual(pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr10_sw1_ch1_get_position_5(self):
self.audio_source.read(5)
pos = self.audio_source.get_position()
self.assertEqual(pos, 5, msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr10_sw1_ch1_get_position_25(self):
self.audio_source.read(5)
self.audio_source.read(20)
pos = self.audio_source.get_position()
self.assertEqual(pos, 25, msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr10_sw1_ch1_set_position_0(self):
self.audio_source.read(10)
self.audio_source.set_position(0)
pos = self.audio_source.get_position()
self.assertEqual(pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr10_sw1_ch1_set_position_10(self):
self.audio_source.set_position(10)
pos = self.audio_source.get_position()
self.assertEqual(pos, 10, msg="wrong position, expected: 10, found: {0} ".format(pos))
def test_sr10_sw1_ch1_get_time_position_0(self):
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 0.0, msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr10_sw1_ch1_get_time_position_1(self):
srate = self.audio_source.get_sampling_rate()
# read one second
self.audio_source.read(srate)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 1.0, msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr10_sw1_ch1_get_time_position_2_5(self):
# read 2.5 seconds
self.audio_source.read(25)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 2.5, msg="wrong time position, expected: 2.5, found: {0} ".format(tp))
def test_sr10_sw1_ch1_set_time_position_0(self):
self.audio_source.read(10)
self.audio_source.set_time_position(0)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 0.0, msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr10_sw1_ch1_set_time_position_1(self):
self.audio_source.set_time_position(1)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 1.0, msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr10_sw1_ch1_set_time_position_end(self):
self.audio_source.set_time_position(100)
tp = self.audio_source.get_time_position()
self.assertEqual(tp, 3.2, msg="wrong time position, expected: 3.2, found: {0} ".format(tp))
def test_sr10_sw1_ch1_rewind(self):
self.audio_source.read(10)
self.audio_source.rewind()
tp = self.audio_source.get_position()
self.assertEqual(tp, 0, msg="wrong position, expected: 0.0, found: {0} ".format(tp))
def test_sr10_sw1_ch1_set_data(self):
self.audio_source.set_data("12345")
block = self.audio_source.read(9999)
self.assertEqual(block, "12345", msg="wrong block, expected: '12345', found: {0} ".format(block))
def test_sr10_sw1_ch1_read_closed(self):
self.audio_source.close()
with self.assertRaises(Exception):
self.audio_source.read(1)
class TestBufferAudioSource_SR10_SW1_CH1(unittest.TestCase):
def setUp(self):
self.signal = "ABCDEFGHIJKLMNOPQRSTUVWXYZ012345"
self.audio_source = BufferAudioSource(data_buffer=self.signal,
sampling_rate=10,
sample_width=1,
channels=1)
self.audio_source.open()
def tearDown(self):
self.audio_source.close()
def test_sr10_sw1_ch1_read_1(self):
block = self.audio_source.read(1)
self.assertEqual(
block,
"A",
msg="wrong block, expected: 'A', found: {0} ".format(block))
def test_sr10_sw1_ch1_read_6(self):
block = self.audio_source.read(6)
self.assertEqual(
block,
"ABCDEF",
msg="wrong block, expected: 'ABCDEF', found: {0} ".format(block))
def test_sr10_sw1_ch1_read_multiple(self):
block = self.audio_source.read(1)
self.assertEqual(
block,
"A",
msg="wrong block, expected: 'A', found: {0} ".format(block))
block = self.audio_source.read(6)
self.assertEqual(
block,
"BCDEFG",
msg="wrong block, expected: 'BCDEFG', found: {0} ".format(block))
block = self.audio_source.read(13)
self.assertEqual(
block,
"HIJKLMNOPQRST",
msg="wrong block, expected: 'HIJKLMNOPQRST', found: {0} ".format(
block))
block = self.audio_source.read(9999)
self.assertEqual(
block,
"UVWXYZ012345",
msg="wrong block, expected: 'UVWXYZ012345', found: {0} ".format(
block))
def test_sr10_sw1_ch1_read_all(self):
block = self.audio_source.read(9999)
self.assertEqual(block,
self.signal,
msg="wrong block, expected: {0}, found: {1} ".format(
self.signal, block))
block = self.audio_source.read(1)
self.assertEqual(block,
None,
msg="wrong block, expected: {0}, found: {1} ".format(
None, block))
def test_sr10_sw1_ch1_get_sampling_rate(self):
srate = self.audio_source.get_sampling_rate()
self.assertEqual(
srate,
10,
msg="wrong sampling rate, expected: 10, found: {0} ".format(srate))
def test_sr10_sw1_ch1_get_sample_width(self):
swidth = self.audio_source.get_sample_width()
self.assertEqual(
swidth,
1,
msg="wrong sample width, expected: 1, found: {0} ".format(swidth))
def test_sr10_sw1_ch1_get_channels(self):
channels = self.audio_source.get_channels()
self.assertEqual(
channels,
1,
msg="wrong number of channels, expected: 1, found: {0} ".format(
channels))
def test_sr10_sw1_ch1_get_position_0(self):
pos = self.audio_source.get_position()
self.assertEqual(
pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr10_sw1_ch1_get_position_5(self):
self.audio_source.read(5)
pos = self.audio_source.get_position()
self.assertEqual(
pos, 5, msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr10_sw1_ch1_get_position_25(self):
self.audio_source.read(5)
self.audio_source.read(20)
pos = self.audio_source.get_position()
self.assertEqual(
pos,
25,
msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr10_sw1_ch1_set_position_0(self):
self.audio_source.read(10)
self.audio_source.set_position(0)
pos = self.audio_source.get_position()
self.assertEqual(
pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr10_sw1_ch1_set_position_10(self):
self.audio_source.set_position(10)
pos = self.audio_source.get_position()
self.assertEqual(
pos,
10,
msg="wrong position, expected: 10, found: {0} ".format(pos))
def test_sr10_sw1_ch1_get_time_position_0(self):
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
0.0,
msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr10_sw1_ch1_get_time_position_1(self):
srate = self.audio_source.get_sampling_rate()
# read one second
self.audio_source.read(srate)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
1.0,
msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr10_sw1_ch1_get_time_position_2_5(self):
# read 2.5 seconds
self.audio_source.read(25)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
2.5,
msg="wrong time position, expected: 2.5, found: {0} ".format(tp))
def test_sr10_sw1_ch1_set_time_position_0(self):
self.audio_source.read(10)
self.audio_source.set_time_position(0)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
0.0,
msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr10_sw1_ch1_set_time_position_1(self):
self.audio_source.set_time_position(1)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
1.0,
msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr10_sw1_ch1_set_time_position_end(self):
self.audio_source.set_time_position(100)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
3.2,
msg="wrong time position, expected: 3.2, found: {0} ".format(tp))
def test_sr10_sw1_ch1_rewind(self):
self.audio_source.read(10)
self.audio_source.rewind()
tp = self.audio_source.get_position()
self.assertEqual(
tp, 0, msg="wrong position, expected: 0.0, found: {0} ".format(tp))
def test_sr10_sw1_ch1_set_data(self):
self.audio_source.set_data("12345")
block = self.audio_source.read(9999)
self.assertEqual(
block,
"12345",
msg="wrong block, expected: '12345', found: {0} ".format(block))
def test_sr10_sw1_ch1_read_closed(self):
self.audio_source.close()
with self.assertRaises(Exception):
self.audio_source.read(1)
class TestBufferAudioSource_SR16_SW2_CH1(unittest.TestCase):
def setUp(self):
self.signal = "ABCDEFGHIJKLMNOPQRSTUVWXYZ012345"
self.audio_source = BufferAudioSource(data_buffer=self.signal,
sampling_rate=16,
sample_width=2,
channels=1)
self.audio_source.open()
def tearDown(self):
self.audio_source.close()
def test_sr16_sw2_ch1_read_1(self):
block = self.audio_source.read(1)
self.assertEqual(
block,
"AB",
msg="wrong block, expected: 'AB', found: {0} ".format(block))
def test_sr16_sw2_ch1_read_6(self):
block = self.audio_source.read(6)
self.assertEqual(
block,
"ABCDEFGHIJKL",
msg="wrong block, expected: 'ABCDEFGHIJKL', found: {0} ".format(
block))
def test_sr16_sw2_ch1_read_multiple(self):
block = self.audio_source.read(1)
self.assertEqual(
block,
"AB",
msg="wrong block, expected: 'AB', found: {0} ".format(block))
block = self.audio_source.read(6)
self.assertEqual(
block,
"CDEFGHIJKLMN",
msg="wrong block, expected: 'CDEFGHIJKLMN', found: {0} ".format(
block))
block = self.audio_source.read(5)
self.assertEqual(
block,
"OPQRSTUVWX",
msg="wrong block, expected: 'OPQRSTUVWX', found: {0} ".format(
block))
block = self.audio_source.read(9999)
self.assertEqual(
block,
"YZ012345",
msg="wrong block, expected: 'YZ012345', found: {0} ".format(block))
def test_sr16_sw2_ch1_read_all(self):
block = self.audio_source.read(9999)
self.assertEqual(block,
self.signal,
msg="wrong block, expected: {0}, found: {1} ".format(
self.signal, block))
block = self.audio_source.read(1)
self.assertEqual(block,
None,
msg="wrong block, expected: {0}, found: {1} ".format(
None, block))
def test_sr16_sw2_ch1_get_sampling_rate(self):
srate = self.audio_source.get_sampling_rate()
self.assertEqual(
srate,
16,
msg="wrong sampling rate, expected: 10, found: {0} ".format(srate))
def test_sr16_sw2_ch1_get_sample_width(self):
swidth = self.audio_source.get_sample_width()
self.assertEqual(
swidth,
2,
msg="wrong sample width, expected: 1, found: {0} ".format(swidth))
def test_sr16_sw2_ch1_get_channels(self):
channels = self.audio_source.get_channels()
self.assertEqual(
channels,
1,
msg="wrong number of channels, expected: 1, found: {0} ".format(
channels))
def test_sr16_sw2_ch1_get_position_0(self):
pos = self.audio_source.get_position()
self.assertEqual(
pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr16_sw2_ch1_get_position_5(self):
self.audio_source.read(5)
pos = self.audio_source.get_position()
self.assertEqual(
pos, 5, msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr16_sw2_ch1_get_position_15(self):
self.audio_source.read(5)
self.audio_source.read(10)
pos = self.audio_source.get_position()
self.assertEqual(
pos,
15,
msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr16_sw2_ch1_set_position_0(self):
self.audio_source.read(10)
self.audio_source.set_position(0)
pos = self.audio_source.get_position()
self.assertEqual(
pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr16_sw2_ch1_set_position_10(self):
self.audio_source.set_position(10)
pos = self.audio_source.get_position()
self.assertEqual(
pos,
10,
msg="wrong position, expected: 10, found: {0} ".format(pos))
def test_sr16_sw2_ch1_get_time_position_0(self):
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
0.0,
msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_get_time_position_1(self):
srate = self.audio_source.get_sampling_rate()
# read one second
self.audio_source.read(srate)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
1.0,
msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_get_time_position_0_75(self):
# read 2.5 seconds
self.audio_source.read(12)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
0.75,
msg="wrong time position, expected: 0.75, found: {0} ".format(tp))
def test_sr16_sw2_ch1_set_time_position_0(self):
self.audio_source.read(10)
self.audio_source.set_time_position(0)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
0.0,
msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_set_time_position_1(self):
self.audio_source.set_time_position(1)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
1.0,
msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_set_time_position_end(self):
self.audio_source.set_time_position(100)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
1.0,
msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_rewind(self):
self.audio_source.read(10)
self.audio_source.rewind()
tp = self.audio_source.get_position()
self.assertEqual(
tp, 0, msg="wrong position, expected: 0.0, found: {0} ".format(tp))
def test_sr16_sw2_ch1_set_data(self):
self.audio_source.set_data("abcdef")
block = self.audio_source.read(9999)
self.assertEqual(
block,
"abcdef",
msg="wrong block, expected: 'abcdef', found: {0} ".format(block))
def test_sr16_sw2_ch1_set_data_exception(self):
with self.assertRaises(Exception):
self.assertRaises(ValueError, self.audio_source.set_data("abcde"))
class TestBufferAudioSource_SR11_SW4_CH1(unittest.TestCase):
def setUp(self):
self.signal = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789abcdefgh"
self.audio_source = BufferAudioSource(data_buffer=self.signal,
sampling_rate=11,
sample_width=4,
channels=1)
self.audio_source.open()
def tearDown(self):
self.audio_source.close()
def test_sr11_sw4_ch1_read_1(self):
block = self.audio_source.read(1)
self.assertEqual(
block,
"ABCD",
msg="wrong block, expected: 'ABCD', found: {0} ".format(block))
def test_sr11_sw4_ch1_read_6(self):
block = self.audio_source.read(6)
self.assertEqual(
block,
"ABCDEFGHIJKLMNOPQRSTUVWX",
msg="wrong block, expected: 'ABCDEFGHIJKLMNOPQRSTUVWX', found: {0} "
.format(block))
def test_sr11_sw4_ch1_read_multiple(self):
block = self.audio_source.read(1)
self.assertEqual(
block,
"ABCD",
msg="wrong block, expected: 'AB', found: {0} ".format(block))
block = self.audio_source.read(6)
self.assertEqual(
block,
"EFGHIJKLMNOPQRSTUVWXYZ01",
msg="wrong block, expected: 'EFGHIJKLMNOPQRSTUVWXYZ01', found: {0} "
.format(block))
block = self.audio_source.read(3)
self.assertEqual(
block,
"23456789abcd",
msg="wrong block, expected: '23456789abcd', found: {0} ".format(
block))
block = self.audio_source.read(9999)
self.assertEqual(
block,
"efgh",
msg="wrong block, expected: 'efgh', found: {0} ".format(block))
def test_sr11_sw4_ch1_read_all(self):
block = self.audio_source.read(9999)
self.assertEqual(block,
self.signal,
msg="wrong block, expected: {0}, found: {1} ".format(
self.signal, block))
block = self.audio_source.read(1)
self.assertEqual(block,
None,
msg="wrong block, expected: {0}, found: {1} ".format(
None, block))
def test_sr11_sw4_ch1_get_sampling_rate(self):
srate = self.audio_source.get_sampling_rate()
self.assertEqual(
srate,
11,
msg="wrong sampling rate, expected: 10, found: {0} ".format(srate))
def test_sr11_sw4_ch1_get_sample_width(self):
swidth = self.audio_source.get_sample_width()
self.assertEqual(
swidth,
4,
msg="wrong sample width, expected: 1, found: {0} ".format(swidth))
def test_sr11_sw4_ch1_get_channels(self):
channels = self.audio_source.get_channels()
self.assertEqual(
channels,
1,
msg="wrong number of channels, expected: 1, found: {0} ".format(
channels))
def test_sr11_sw4_ch1_get_position_0(self):
pos = self.audio_source.get_position()
self.assertEqual(
pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr11_sw4_ch1_get_position_5(self):
self.audio_source.read(5)
pos = self.audio_source.get_position()
self.assertEqual(
pos, 5, msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr11_sw4_ch1_get_position_9(self):
self.audio_source.read(5)
self.audio_source.read(4)
pos = self.audio_source.get_position()
self.assertEqual(
pos, 9, msg="wrong position, expected: 5, found: {0} ".format(pos))
def test_sr11_sw4_ch1_set_position_0(self):
self.audio_source.read(10)
self.audio_source.set_position(0)
pos = self.audio_source.get_position()
self.assertEqual(
pos, 0, msg="wrong position, expected: 0, found: {0} ".format(pos))
def test_sr11_sw4_ch1_set_position_10(self):
self.audio_source.set_position(10)
pos = self.audio_source.get_position()
self.assertEqual(
pos,
10,
msg="wrong position, expected: 10, found: {0} ".format(pos))
def test_sr11_sw4_ch1_get_time_position_0(self):
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
0.0,
msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_get_time_position_1(self):
srate = self.audio_source.get_sampling_rate()
# read one second
self.audio_source.read(srate)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
1.0,
msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_get_time_position_0_63(self):
# read 2.5 seconds
self.audio_source.read(7)
tp = self.audio_source.get_time_position()
self.assertAlmostEqual(
tp,
0.636363636364,
msg="wrong time position, expected: 0.636363636364, found: {0} ".
format(tp))
def test_sr11_sw4_ch1_set_time_position_0(self):
self.audio_source.read(10)
self.audio_source.set_time_position(0)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
0.0,
msg="wrong time position, expected: 0.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_set_time_position_1(self):
self.audio_source.set_time_position(1)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
1.0,
msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_set_time_position_end(self):
self.audio_source.set_time_position(100)
tp = self.audio_source.get_time_position()
self.assertEqual(
tp,
1.0,
msg="wrong time position, expected: 1.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_rewind(self):
self.audio_source.read(10)
self.audio_source.rewind()
tp = self.audio_source.get_position()
self.assertEqual(
tp, 0, msg="wrong position, expected: 0.0, found: {0} ".format(tp))
def test_sr11_sw4_ch1_set_data(self):
self.audio_source.set_data("abcdefgh")
block = self.audio_source.read(9999)
self.assertEqual(
block,
"abcdefgh",
msg="wrong block, expected: 'abcdef', found: {0} ".format(block))
def test_sr11_sw4_ch1_set_data_exception(self):
with self.assertRaises(Exception):
self.assertRaises(ValueError, self.audio_source.set_data("abcdef"))
