test_gmpy.py 397 B

123456789101112
  1. from sympy.external.gmpy import LONG_MAX, iroot
  2. from sympy.testing.pytest import raises
  3. def test_iroot():
  4. assert iroot(2, LONG_MAX) == (1, False)
  5. assert iroot(2, LONG_MAX + 1) == (1, False)
  6. for x in range(3):
  7. assert iroot(x, 1) == (x, True)
  8. raises(ValueError, lambda: iroot(-1, 1))
  9. raises(ValueError, lambda: iroot(0, 0))
  10. raises(ValueError, lambda: iroot(0, -1))