def test_to_mono(): data = np.arange(6, dtype=np.float32).reshape(2, 3) stereo = Stereo(data, 44100, 'foo.wav') actual = stereo.to_mono() expected = np.array([1.5, 2.5, 3.5]) assert np.array_equal(actual.data, expected) assert actual.sr == stereo.sr assert actual.filename == stereo.filename
def test_drop(): data = np.arange(20).reshape(2, 10) stereo = Stereo(data, 44100, 'foo.wav') actual = stereo.drop(5) expected = np.array([ [5, 6, 7, 8, 9], [15, 16, 17, 18, 19], ]) assert np.array_equal(actual.data, expected) assert not np.array_equal(actual.data, stereo.data) assert actual.sr == stereo.sr assert actual.filename == stereo.filename
def test_resize_down(): data = np.arange(20).reshape(2, 10) stereo = Stereo(data, 44100, 'foo.wav') actual = stereo.resize(5) expected = np.array([ [0, 1, 2, 3, 4], [10, 11, 12, 13, 14], ]) assert np.array_equal(actual.data, expected) assert not np.array_equal(actual.data, stereo.data) assert actual.sr == stereo.sr assert actual.filename == stereo.filename
def test_resize_up(): data = np.arange(10).reshape(2, 5) stereo = Stereo(data, 44100, 'foo.wav') actual = stereo.resize(10) expected = np.array([ [0, 1, 2, 3, 4, 0, 0, 0, 0, 0], [5, 6, 7, 8, 9, 0, 0, 0, 0, 0], ]) assert np.array_equal(actual.data, expected) assert not np.array_equal(actual.data, stereo.data) assert actual.sr == stereo.sr assert actual.filename == stereo.filename
def test_tile(): data = np.arange(4).reshape(2, 2) stereo = Stereo(data, 44100, 'foo.wav') actual = stereo.tile(3) expected = np.array([ [0, 1, 0, 1, 0, 1], [2, 3, 2, 3, 2, 3], ]) assert np.array_equal(actual.data, expected) assert not np.array_equal(actual.data, stereo.data) assert actual.sr == stereo.sr assert actual.filename == stereo.filename
def test_trim(): data1 = np.zeros(2 * 4096, dtype=np.float32).reshape(2, 4096) stereo1 = Stereo(data1, 44100, 'foo.wav') data2 = np.arange(4, dtype=np.float32).reshape(2, 2) stereo2 = Stereo(data2, 44100, 'foo.wav') stereo = stereo1.append(stereo2) actual = stereo.trim() assert actual.data.shape == (2, 512 + 2) # TODO assert not np.array_equal(actual.data, stereo.data) assert actual.sr == stereo.sr assert actual.filename == stereo.filename
def test_empty(): stereo = Stereo.empty(44100) assert stereo.data.size == 0 assert stereo.data.shape == (2, 0) assert stereo.data.dtype == np.float32 assert stereo.sr == 44100 assert stereo.filename is None
def test_append(): data1 = np.arange(6).reshape(2, 3) stereo1 = Stereo(data1, 44100, 'foo.wav') data2 = np.arange(4).reshape(2, 2) stereo2 = Stereo(data2, 44100, 'bar.wav') actual = stereo1.append(stereo2) expected = np.array([ [0, 1, 2, 0, 1], [3, 4, 5, 2, 3], ]) assert np.array_equal(actual.data, expected) assert not np.array_equal(actual.data, stereo1.data) assert not np.array_equal(actual.data, stereo2.data) assert actual.sr == stereo1.sr assert actual.filename == stereo1.filename
def test_init(): data = np.arange(100).reshape(2, 50) stereo = Stereo(data, 44100, 'foo.wav') assert np.array_equal(stereo.data, data) assert stereo.sr == 44100 assert stereo.filename == 'foo.wav'
def test_init_assertion(): with pytest.raises(AssertionError): data = np.arange(100) Stereo(data, 44100, 'foo.wav')
def test_is_mono(): data = np.arange(4).reshape(2, 2) stereo = Stereo(data, 44100, 'foo.wav') assert not stereo.is_mono()