def testSqueezeExecution(self):
data = np.array([[[0], [1], [2]]])
x = tensor(data, chunk_size=1)
t = squeeze(x)
res = self.executor.execute_tensor(t, concat=True)[0]
expected = np.squeeze(data)
np.testing.assert_equal(res, expected)
t = squeeze(x, axis=2)
res = self.executor.execute_tensor(t, concat=True)[0]
expected = np.squeeze(data, axis=2)
np.testing.assert_equal(res, expected)
def test_squeeze():
data = np.array([[[0], [1], [2]]])
x = tensor(data)
t = squeeze(x)
assert t.shape == (3, )
assert t.dtype is not None
t = squeeze(x, axis=0)
assert t.shape == (3, 1)
with pytest.raises(ValueError):
squeeze(x, axis=1)
t = squeeze(x, axis=2)
assert t.shape == (1, 3)
def testSqueeze(self):
data = np.array([[[0], [1], [2]]])
x = tensor(data)
t = squeeze(x)
self.assertEqual(t.shape, (3, ))
self.assertIsNotNone(t.dtype)
t = squeeze(x, axis=0)
self.assertEqual(t.shape, (3, 1))
with self.assertRaises(ValueError):
squeeze(x, axis=1)
t = squeeze(x, axis=2)
self.assertEqual(t.shape, (1, 3))
def test_create_view():
arr = ones((10, 20, 30), chunk_size=[4, 3, 5])
arr2 = transpose(arr)
assert arr2.op.create_view is True
arr3 = transpose(arr)
assert arr3.op.create_view is True
arr4 = arr.swapaxes(0, 1)
assert arr4.op.create_view is True
arr5 = moveaxis(arr, 1, 0)
assert arr5.op.create_view is True
arr6 = atleast_1d(1)
assert arr6.op.create_view is True
arr7 = atleast_2d([1, 1])
assert arr7.op.create_view is True
arr8 = atleast_3d([1, 1])
assert arr8.op.create_view is True
arr9 = arr[:3, [1, 2, 3]]
# no view cuz of fancy indexing
assert arr9.op.create_view is False
arr9[0][0][0] = 100
assert arr9.op.create_view is False
arr10 = arr[:3, None, :5]
assert arr10.op.create_view is True
arr10[0][0][0] = 100
assert arr10.op.create_view is False
data = np.array([[[0], [1], [2]]])
x = tensor(data)
t = squeeze(x)
assert t.op.create_view is True
y = x.reshape(3)
assert y.op.create_view is True
def testCreateView(self):
arr = ones((10, 20, 30), chunk_size=[4, 3, 5])
arr2 = transpose(arr)
self.assertTrue(arr2.op.create_view)
arr3 = transpose(arr)
self.assertTrue(arr3.op.create_view)
arr4 = arr.swapaxes(0, 1)
self.assertTrue(arr4.op.create_view)
arr5 = moveaxis(arr, 1, 0)
self.assertTrue(arr5.op.create_view)
arr6 = atleast_1d(1)
self.assertTrue(arr6.op.create_view)
arr7 = atleast_2d([1, 1])
self.assertTrue(arr7.op.create_view)
arr8 = atleast_3d([1, 1])
self.assertTrue(arr8.op.create_view)
arr9 = arr[:3, [1, 2, 3]]
# no view cuz of fancy indexing
self.assertFalse(arr9.op.create_view)
arr9[0][0][0] = 100
self.assertFalse(arr9.op.create_view)
arr10 = arr[:3, None, :5]
self.assertTrue(arr10.op.create_view)
arr10[0][0][0] = 100
self.assertFalse(arr10.op.create_view)
data = np.array([[[0], [1], [2]]])
x = tensor(data)
t = squeeze(x)
self.assertTrue(t.op.create_view)
y = x.reshape(3)
self.assertTrue(y.op.create_view)
