89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
from sympy.unify.core import Compound, Variable, CondVariable, allcombinations
|
|
from sympy.unify import core
|
|
|
|
a,b,c = 'a', 'b', 'c'
|
|
w,x,y,z = map(Variable, 'wxyz')
|
|
|
|
C = Compound
|
|
|
|
def is_associative(x):
|
|
return isinstance(x, Compound) and (x.op in ('Add', 'Mul', 'CAdd', 'CMul'))
|
|
def is_commutative(x):
|
|
return isinstance(x, Compound) and (x.op in ('CAdd', 'CMul'))
|
|
|
|
|
|
def unify(a, b, s={}):
|
|
return core.unify(a, b, s=s, is_associative=is_associative,
|
|
is_commutative=is_commutative)
|
|
|
|
def test_basic():
|
|
assert list(unify(a, x, {})) == [{x: a}]
|
|
assert list(unify(a, x, {x: 10})) == []
|
|
assert list(unify(1, x, {})) == [{x: 1}]
|
|
assert list(unify(a, a, {})) == [{}]
|
|
assert list(unify((w, x), (y, z), {})) == [{w: y, x: z}]
|
|
assert list(unify(x, (a, b), {})) == [{x: (a, b)}]
|
|
|
|
assert list(unify((a, b), (x, x), {})) == []
|
|
assert list(unify((y, z), (x, x), {}))!= []
|
|
assert list(unify((a, (b, c)), (a, (x, y)), {})) == [{x: b, y: c}]
|
|
|
|
def test_ops():
|
|
assert list(unify(C('Add', (a,b,c)), C('Add', (a,x,y)), {})) == \
|
|
[{x:b, y:c}]
|
|
assert list(unify(C('Add', (C('Mul', (1,2)), b,c)), C('Add', (x,y,c)), {})) == \
|
|
[{x: C('Mul', (1,2)), y:b}]
|
|
|
|
def test_associative():
|
|
c1 = C('Add', (1,2,3))
|
|
c2 = C('Add', (x,y))
|
|
assert tuple(unify(c1, c2, {})) == ({x: 1, y: C('Add', (2, 3))},
|
|
{x: C('Add', (1, 2)), y: 3})
|
|
|
|
def test_commutative():
|
|
c1 = C('CAdd', (1,2,3))
|
|
c2 = C('CAdd', (x,y))
|
|
result = list(unify(c1, c2, {}))
|
|
assert {x: 1, y: C('CAdd', (2, 3))} in result
|
|
assert ({x: 2, y: C('CAdd', (1, 3))} in result or
|
|
{x: 2, y: C('CAdd', (3, 1))} in result)
|
|
|
|
def _test_combinations_assoc():
|
|
assert set(allcombinations((1,2,3), (a,b), True)) == \
|
|
{(((1, 2), (3,)), (a, b)), (((1,), (2, 3)), (a, b))}
|
|
|
|
def _test_combinations_comm():
|
|
assert set(allcombinations((1,2,3), (a,b), None)) == \
|
|
{(((1,), (2, 3)), ('a', 'b')), (((2,), (3, 1)), ('a', 'b')),
|
|
(((3,), (1, 2)), ('a', 'b')), (((1, 2), (3,)), ('a', 'b')),
|
|
(((2, 3), (1,)), ('a', 'b')), (((3, 1), (2,)), ('a', 'b'))}
|
|
|
|
def test_allcombinations():
|
|
assert set(allcombinations((1,2), (1,2), 'commutative')) ==\
|
|
{(((1,),(2,)), ((1,),(2,))), (((1,),(2,)), ((2,),(1,)))}
|
|
|
|
|
|
def test_commutativity():
|
|
c1 = Compound('CAdd', (a, b))
|
|
c2 = Compound('CAdd', (x, y))
|
|
assert is_commutative(c1) and is_commutative(c2)
|
|
assert len(list(unify(c1, c2, {}))) == 2
|
|
|
|
|
|
def test_CondVariable():
|
|
expr = C('CAdd', (1, 2))
|
|
x = Variable('x')
|
|
y = CondVariable('y', lambda a: a % 2 == 0)
|
|
z = CondVariable('z', lambda a: a > 3)
|
|
pattern = C('CAdd', (x, y))
|
|
assert list(unify(expr, pattern, {})) == \
|
|
[{x: 1, y: 2}]
|
|
|
|
z = CondVariable('z', lambda a: a > 3)
|
|
pattern = C('CAdd', (z, y))
|
|
|
|
assert list(unify(expr, pattern, {})) == []
|
|
|
|
def test_defaultdict():
|
|
assert next(unify(Variable('x'), 'foo')) == {Variable('x'): 'foo'}
|