def testAtleast2dExecution(self): x = 1 y = ones(3, chunk_size=2) z = ones((3, 4), chunk_size=2) t = atleast_2d(x, y, z) res = [self.executor.execute_tensor(i, concat=True)[0] for i in t] self.assertTrue(np.array_equal(res[0], np.array([[1]]))) self.assertTrue(np.array_equal(res[1], np.atleast_2d(np.ones(3)))) self.assertTrue(np.array_equal(res[2], np.ones((3, 4))))
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) arr10 = arr[:3, None, :5] self.assertTrue(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)