def test_iadd(self):
x = UTPS(numpy.array([1.,2.,3.]), P = 1)
x2 = x.copy()
y = UTPS(numpy.array([5.,7.,11.]), P = 1)
x += y
assert_array_almost_equal(add(x2,y,x2).data,x.data)
def test_iadd_vectorized(self):
x = UTPS(numpy.array([1.,2.,3, 4.,6.]),P = 2)
y = UTPS(numpy.array([5.,7.,11, 1.,2.]),P = 2)
y = add(y,x,y)
assert_array_almost_equal([6.,9.,14., 5., 8.], y.data)
def test_add(self):
""" test z = x + y"""
x = UTPS(numpy.array([1.,2.,3.]), P = 1)
y = UTPS(numpy.array([5.,7.,11.]), P = 1)
z = add(x,y)
assert_array_almost_equal([6.,9.,14.], z.data)
def test_iadd(self):
""" test y += x"""
x = UTPS(numpy.array([1.,2.,3.]), P = 1)
y = UTPS(numpy.array([5.,7.,11.]), P = 1)
y = add(y,x,y)
assert_array_almost_equal([6.,9.,14.], y.data)
def test_add_epb(self):
x = UTPS(numpy.array([1.,2.,3.]), P = 1)
y = UTPS(numpy.array([5.,7.,11.]), P = 1)
z = add(x,y)
zbar = UTPS(numpy.random.rand(3), P = 1)
xbar = UTPS(numpy.zeros(3), P = 1)
ybar = UTPS(numpy.zeros(3), P = 1)
epb_add(x,y,z, zbar, xbar, ybar)
assert_array_almost_equal(zbar.data, xbar.data)
assert_array_almost_equal(zbar.data, ybar.data)
def test_add_epb(self):
Q,P,D = 3,1,3
x = UTPS(numpy.random.rand(1+P*(D-1)), P = P)
y = UTPS(numpy.random.rand(1+P*(D-1)), P = P)
z = add(x,y)
zbar = UTPS(numpy.random.rand( Q*(1+P*(D-1))), Q = Q, P = P)
xbar = UTPS(numpy.random.rand( Q*(1+P*(D-1))), Q = Q, P = P)
ybar = UTPS(numpy.random.rand( Q*(1+P*(D-1))), Q = Q, P = P)
xbar2 = xbar.copy()
ybar2 = ybar.copy()
zbar2 = ybar.copy()
epb_add(x,y,z, zbar, xbar, ybar)
assert_array_almost_equal(xbar2.data + zbar.data, xbar.data)
assert_array_almost_equal(ybar2.data + zbar.data, ybar.data)
def test_add(self):
x = UTPS(numpy.array([1.,2.,3.]), P = 1)
y = UTPS(numpy.array([5.,7.,11.]), P = 1)
z = x + y
assert_array_almost_equal(add(x,y).data,z.data)
