def testBaseExecute(self):
with option_context({'eager_mode': True}):
a_data = np.random.rand(10, 10)
a = mt.tensor(a_data, chunk_size=3)
np.testing.assert_array_equal(a.fetch(), a_data)
r1 = a + 1
np.testing.assert_array_equal(r1.fetch(), a_data + 1)
r2 = 2 * r1
np.testing.assert_array_equal(r2.fetch(), (a_data + 1) * 2)
# test add with out
b = mt.ones((10, 10), chunk_size=3)
np.testing.assert_array_equal(b.fetch(), np.ones((10, 10)))
mt.add(a, b, out=b)
np.testing.assert_array_equal(b.fetch(), a_data + 1)
# test tensor dot
c_data1 = np.random.rand(10, 10)
c_data2 = np.random.rand(10, 10)
c1 = mt.tensor(c_data1, chunk_size=4)
c2 = mt.tensor(c_data2, chunk_size=4)
r3 = c1.dot(c2)
np.testing.assert_array_almost_equal(r3.fetch(),
c_data1.dot(c_data2))
def testViewDataOnTranspose(self):
a = ones((10, 20), chunk_size=6)
b = a.T
add(b, 1, out=b)
np.testing.assert_array_equal(b.execute(), np.ones((20, 10)) + 1)
np.testing.assert_array_equal(a.execute(), np.ones((10, 20)) + 1)
def testViewDataOnTranspose(self):
data = np.random.rand(10, 20)
a = tensor(data, chunk_size=6)
b = a.T
add(b, 1, out=b)
np.testing.assert_array_equal(b.execute(), data.T + 1)
np.testing.assert_array_equal(a.execute(), data + 1)
def test_view_data_on_transpose(setup):
data = np.random.rand(10, 20)
a = tensor(data, chunk_size=6)
b = a.T
add(b, 1, out=b)
np.testing.assert_array_equal(b.execute(), data.T + 1)
np.testing.assert_array_equal(a.execute(), data + 1)
def testViewDataOnAtleast2d(self):
a = atleast_2d(ones(10, chunk_size=5))
b = add(a[:, :5], 1, out=a[:, 5:])
npa = np.atleast_2d(np.ones(10))
npb = np.add(npa[:, :5], 1, out=npa[:, 5:])
np.testing.assert_array_equal(b.execute(), npb)
np.testing.assert_array_equal(a.execute(), npa)
def testViewDataOnAtleast2d(self):
data = np.random.rand(10)
a = atleast_2d(tensor(data, chunk_size=5))
b = add(a[:, :5], 1, out=a[:, 5:])
npa = np.atleast_2d(data.copy())
npb = np.add(npa[:, :5], 1, out=npa[:, 5:])
np.testing.assert_array_equal(b.execute(), npb)
np.testing.assert_array_equal(a.execute(), npa)
def testViewDataOnAtleast3d(self):
a = atleast_3d(ones((10, 20), chunk_size=5))
b = a[:, :5, :10][0]
c = add(b[:4], b[1:], out=a[0, 16:])
npa = np.atleast_3d(np.ones((10, 20)))
npb = npa[:, :5, :10][0]
npc = np.add(npb[:4], npb[1:], out=npa[0, 16:])
np.testing.assert_array_equal(c.execute(), npc)
np.testing.assert_array_equal(b.execute(), npb)
np.testing.assert_array_equal(a.execute(), npa)
def testViewDataOnAtleast3d(self):
data = np.random.rand(10, 20)
a = atleast_3d(tensor(data, chunk_size=5))
b = a[:, :5, :10][0]
c = add(b[:4], b[1:], out=a[0, 16:])
npa = np.atleast_3d(data.copy())
npb = npa[:, :5, :10][0]
npc = np.add(npb[:4], npb[1:], out=npa[0, 16:])
np.testing.assert_array_equal(c.execute(), npc)
np.testing.assert_array_equal(b.execute(), npb)
np.testing.assert_array_equal(a.execute(), npa)
