12 lines
397 B
Python
12 lines
397 B
Python
from sympy.external.gmpy import LONG_MAX, iroot
|
|
from sympy.testing.pytest import raises
|
|
|
|
|
|
def test_iroot():
|
|
assert iroot(2, LONG_MAX) == (1, False)
|
|
assert iroot(2, LONG_MAX + 1) == (1, False)
|
|
for x in range(3):
|
|
assert iroot(x, 1) == (x, True)
|
|
raises(ValueError, lambda: iroot(-1, 1))
|
|
raises(ValueError, lambda: iroot(0, 0))
|
|
raises(ValueError, lambda: iroot(0, -1))
|