3 """A brute force test using `sink` to create and write samples to a stereo
4 file, then `source` to check the correct content is read from the files."""
10 from numpy.testing import assert_equal
11 from .utils import get_tmp_sink_path
13 class aubio_source_test_case(unittest.TestCase):
15 def test_read_from_mono(self):
16 out = get_tmp_sink_path()
21 write_samples = np.ones([channels, hop_size], dtype=aubio.float_type)
23 self.check_write_and_read(samplerate, channels, hop_size, blocks,
26 def test_read_from_stereo(self):
27 out = get_tmp_sink_path()
32 write_samples = np.ones([channels, hop_size], dtype=aubio.float_type)
34 self.check_write_and_read(samplerate, channels, hop_size, blocks,
37 def test_read_from_half_stereo(self):
42 write_samples = np.ones([channels, hop_size], dtype=aubio.float_type)
44 write_samples[1, :] = 0
45 self.check_write_and_read(samplerate, channels, hop_size, blocks,
48 def test_read_from_cancelling_channels(self):
53 write_samples = np.ones([channels, hop_size], dtype=aubio.float_type)
55 write_samples[1] *= -1
56 self.check_write_and_read(samplerate, channels, hop_size, blocks,
59 def test_read_from_strange_three_channels(self):
64 write_samples = np.ones([channels, hop_size], dtype=aubio.float_type)
66 write_samples[1, :] = 0
67 self.check_write_and_read(samplerate, channels, hop_size, blocks,
70 def check_write_and_read(self, samplerate, channels,
71 hop_size, blocks, write_samples):
72 expected_mono = np.sum(write_samples, axis=0)/write_samples.shape[0]
73 out = get_tmp_sink_path()
74 snk = aubio.sink(out, samplerate, channels=channels)
75 for i in range(blocks):
76 snk.do_multi(write_samples, hop_size)
77 # close the sink before reading from it
80 src = aubio.source(out, samplerate, hop_size)
81 for i in range(blocks):
82 read_samples, read = src.do_multi()
83 assert_equal (read_samples, write_samples)
84 assert_equal (read, hop_size)
87 for i in range(blocks):
88 read_samples, read = src()
89 assert_equal (read, hop_size)
90 assert_equal (read_samples, expected_mono)
92 if __name__ == '__main__':