test_assumptions_2.py 1.0 KB

1234567891011121314151617181920212223242526272829303132333435
  1. """
  2. rename this to test_assumptions.py when the old assumptions system is deleted
  3. """
  4. from sympy.abc import x, y
  5. from sympy.assumptions.assume import global_assumptions
  6. from sympy.assumptions.ask import Q
  7. from sympy.printing import pretty
  8. def test_equal():
  9. """Test for equality"""
  10. assert Q.positive(x) == Q.positive(x)
  11. assert Q.positive(x) != ~Q.positive(x)
  12. assert ~Q.positive(x) == ~Q.positive(x)
  13. def test_pretty():
  14. assert pretty(Q.positive(x)) == "Q.positive(x)"
  15. assert pretty(
  16. {Q.positive, Q.integer}) == "{Q.integer, Q.positive}"
  17. def test_global():
  18. """Test for global assumptions"""
  19. global_assumptions.add(x > 0)
  20. assert (x > 0) in global_assumptions
  21. global_assumptions.remove(x > 0)
  22. assert not (x > 0) in global_assumptions
  23. # same with multiple of assumptions
  24. global_assumptions.add(x > 0, y > 0)
  25. assert (x > 0) in global_assumptions
  26. assert (y > 0) in global_assumptions
  27. global_assumptions.clear()
  28. assert not (x > 0) in global_assumptions
  29. assert not (y > 0) in global_assumptions