def testViewDataOnSqueeze(self):
a = ones((1, 4, 1), chunk_size=2)
b = squeeze(a, axis=0)
b[:3] = 10
npa = np.ones((1, 4, 1))
npb = np.squeeze(npa, axis=0)
npb[:3] = 10
np.testing.assert_array_equal(b.execute(), npb)
np.testing.assert_array_equal(a.execute(), npa)
def testViewDataOnSqueeze(self):
data = np.random.rand(1, 4, 1)
a = tensor(data, chunk_size=2)
b = squeeze(a, axis=0)
b[:3] = 10
npa = data.copy()
npb = np.squeeze(npa, axis=0)
npb[:3] = 10
np.testing.assert_array_equal(b.execute(), npb)
np.testing.assert_array_equal(a.execute(), npa)
