def test_distribute_add_mul():
from sympy import Add, Mul, symbols
x, y = symbols('x, y')
expr = Mul(2, Add(x, y), evaluate=False)
expected = Add(Mul(2, x), Mul(2, y))
distribute_mul = distribute(Mul, Add)
assert distribute_mul(expr) == expected
def test_distribute_add_mul():
from sympy import Add, Mul, symbols
x, y = symbols('x, y')
expr = Mul(2, Add(x, y), evaluate=False)
expected = Add(Mul(2, x), Mul(2, y))
distribute_mul = distribute(Mul, Add)
assert distribute_mul(expr) == expected
def test_distribute():
class T1(Basic): pass
class T2(Basic): pass
distribute_t12 = distribute(T1, T2)
assert distribute_t12(T1(1, 2, T2(3, 4), 5)) == \
T2(T1(1, 2, 3, 5),
T1(1, 2, 4, 5))
assert distribute_t12(T1(1, 2, 3)) == T1(1, 2, 3)
def test_distribute():
class T1(Basic):
pass
class T2(Basic):
pass
distribute_t12 = distribute(T1, T2)
assert distribute_t12(T1(1, 2, T2(3, 4), 5)) == \
T2(T1(1, 2, 3, 5),
T1(1, 2, 4, 5))
assert distribute_t12(T1(1, 2, 3)) == T1(1, 2, 3)
