def test_vec_solve():
mu_12 = Tensor('mu_12', rank=0)
p_1, p_2 = map(lambda x: Tensor(x, rank=1), ['p_1', 'p_2'])
A = Tensor('A', rank=2)
print solve_vec_eqn(
-mu_12 * (T(p_1) * A * p_2) - mu_12 * (T(p_2) * A * p_1) +
(T(p_2) * A * p_1), mu_12)
def test_0 ():
A = Tensor('A', rank=2, has_inv=True)
x = Tensor('x', rank=1)
y = Tensor('y', rank=1)
alpha = Tensor('alpha', rank=0)
beta = Tensor('beta', rank=0)
eqn = alpha * A * x + beta * y
print eqn, "rank:", expr_rank(eqn)
print "solution for y", solve_vec_eqn(eqn, y)
print "solution for x", solve_vec_eqn(eqn, x)
def test_vec_solve():
mu_12 = Tensor('mu_12', rank=0)
p_1, p_2 = map(lambda x: Tensor(x, rank=1), ['p_1', 'p_2'])
A = Tensor('A', rank=2)
print solve_vec_eqn(-mu_12 * (T(p_1) * A * p_2) - mu_12 * (T(p_2) * A * p_1) + (T(p_2) * A * p_1) , mu_12)
def test_numerator ():
a, b, c = map(lambda x: Tensor(x, rank=0), 'abc')
expr = (a + b) / c
sol = solve_vec_eqn(expr, a)
assert(sol == -b)