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)