def read_block_mono_test(self):
wfile, infos = wav.open_read_mode(STEPS_MONO_16B)
samples = wav.read_block(wfile, 4410)
self.assertEqual(samples.shape, (4410, 1))
self.assertEqual(infos['frame_rate'], 44100)
self.assertEqual(infos['channel_count'], 1)
expected = numpy.ones([4410, 1]) * -1
numpy.testing.assert_array_equal(expected.round(3), samples.round(3))
def read_all_stereo_test(self):
wfile, infos = wav.open_read_mode(STEPS_STEREO_16B)
samples = wav.read_all(wfile)
self.assertEqual(samples.shape, (92610, 2))
self.assertEqual(infos['frame_rate'], 44100)
self.assertEqual(infos['channel_count'], 2)
# Sanity check
frame_rate, samples_test = sp_wavfile.read(open(STEPS_STEREO_16B, 'r'))
numpy.testing.assert_array_equal(samples[:10,:].round(4), (samples_test[:10,:] / float(2**15)).round(4))
def read_block_mono_test(self):
wfile, infos = wav.open_read_mode(STEPS_MONO_16B)
samples = wav.read_block(wfile, 4410)
self.assertEqual(samples.shape, (4410, 1))
self.assertEqual(infos['frame_rate'], 44100)
self.assertEqual(infos['channel_count'], 1)
expected = numpy.ones([4410, 1]) * -1
numpy.testing.assert_array_equal(expected.round(3), samples.round(3))
def seek_and_read_mono_block_test(self):
wfile, infos = wav.open_read_mode(STEPS_MONO_16B)
frames = wav.seek(wfile, 0.4, 0.5)
samples = wav.read_block(wfile, 100)
self.assertEqual(frames, numpy.round((0.5 - 0.4) * 44100, 6))
self.assertEqual(samples.shape, (100, 1))
self.assertEqual(infos['frame_rate'], 44100)
self.assertEqual(infos['channel_count'], 1)
expected = numpy.ones([100, 1]) * -0.6
numpy.testing.assert_array_equal(expected.round(3), samples.round(3))
def seek_and_read_mono_block_test(self):
wfile, infos = wav.open_read_mode(STEPS_MONO_16B)
frames = wav.seek(wfile, 0.4, 0.5)
samples = wav.read_block(wfile, 100)
self.assertEqual(frames, numpy.round((0.5 - 0.4) * 44100, 6))
self.assertEqual(samples.shape, (100, 1))
self.assertEqual(infos['frame_rate'], 44100)
self.assertEqual(infos['channel_count'], 1)
expected = numpy.ones([100, 1]) * -0.6
numpy.testing.assert_array_equal(expected.round(3), samples.round(3))
def read_all_stereo_test(self):
wfile, infos = wav.open_read_mode(STEPS_STEREO_16B)
samples = wav.read_all(wfile)
self.assertEqual(samples.shape, (92610, 2))
self.assertEqual(infos['frame_rate'], 44100)
self.assertEqual(infos['channel_count'], 2)
# Sanity check
frame_rate, samples_test = sp_wavfile.read(open(STEPS_STEREO_16B, 'r'))
numpy.testing.assert_array_equal(samples[:10, :].round(4),
(samples_test[:10, :] /
float(2**15)).round(4))
def seek_and_read_stereo_block_test(self):
wfile, infos = wav.open_read_mode(STEPS_STEREO_16B)
frames = wav.seek(wfile, 0.6)
samples = wav.read_block(wfile, 100)
self.assertEqual(frames, (infos['duration'] - 0.6) * 44100)
self.assertEqual(samples.shape, (100, 2))
self.assertEqual(infos['frame_rate'], 44100)
self.assertEqual(infos['channel_count'], 2)
expected = numpy.ones([100, 2])
expected[:, 0] *= -0.4
expected[:, 1] *= 0.4
numpy.testing.assert_array_equal(expected.round(3), samples.round(3))
def seek_and_read_stereo_block_test(self):
wfile, infos = wav.open_read_mode(STEPS_STEREO_16B)
frames = wav.seek(wfile, 0.6)
samples = wav.read_block(wfile, 100)
self.assertEqual(frames, (infos['duration'] - 0.6) * 44100)
self.assertEqual(samples.shape, (100, 2))
self.assertEqual(infos['frame_rate'], 44100)
self.assertEqual(infos['channel_count'], 2)
expected = numpy.ones([100, 2])
expected[:, 0] *= -0.4
expected[:, 1] *= 0.4
numpy.testing.assert_array_equal(expected.round(3), samples.round(3))
