test_rules.py 349 B

1234567891011121314
  1. from sympy.core.rules import Transform
  2. from sympy.testing.pytest import raises
  3. def test_Transform():
  4. add1 = Transform(lambda x: x + 1, lambda x: x % 2 == 1)
  5. assert add1[1] == 2
  6. assert (1 in add1) is True
  7. assert add1.get(1) == 2
  8. raises(KeyError, lambda: add1[2])
  9. assert (2 in add1) is False
  10. assert add1.get(2) is None