def test_reshape_exceptions(): data_in = TimeData(np.random.rand(6, 256), range(256)) data_out = data_in.reshape((3, 2)) npt.assert_allclose(data_in._data.reshape(3, 2, -1), data_out._data) # test assertion for non-tuple input with pytest.raises(ValueError): data_out = data_in.reshape([3, 2]) # test assertion for wrong dimension with pytest.raises(ValueError, match='Can not reshape audio object'): data_out = data_in.reshape((3, 4))
def test_reshape(): # test reshape with tuple data_in = TimeData(np.random.rand(6, 256), range(256)) data_out = data_in.reshape((3, 2)) npt.assert_allclose(data_in._data.reshape(3, 2, -1), data_out._data) assert id(data_in) != id(data_out) data_out = data_in.reshape((3, -1)) npt.assert_allclose(data_in._data.reshape(3, 2, -1), data_out._data) assert id(data_in) != id(data_out) # test reshape with int data_in = TimeData(np.random.rand(3, 2, 256), range(256)) data_out = data_in.reshape(6) npt.assert_allclose(data_in._data.reshape(6, -1), data_out._data) assert id(data_in) != id(data_out)