wehub-resource-sync
139 lines
4.9 KiB
Python
139 lines
4.9 KiB
Python
"""Tests for the Command-Line Interface."""
|
|
|
|
import platform
|
|
import sys
|
|
|
|
import pexpect
|
|
import pytest
|
|
|
|
from mpmath.tests.test_demos import Console
|
|
|
|
|
|
if platform.python_implementation() == 'PyPy':
|
|
pytest.skip("Don't run CLI tests on PyPy.",
|
|
allow_module_level=True)
|
|
|
|
|
|
if sys.version_info >= (3, 15):
|
|
pytestmark = pytest.mark.filterwarnings("ignore:.*:DeprecationWarning")
|
|
|
|
|
|
def test_bare_console_no_bare_division():
|
|
c = Console(f'{sys.executable} -m mpmath --no-ipython '
|
|
'--no-wrap-floats --int-limits') # for coverage
|
|
|
|
assert c.expect_exact('>>> ') == 0
|
|
assert c.send('1 + 2\r\n') == 7
|
|
assert c.expect_exact('3\r\n>>> ') == 0
|
|
assert c.send('1/2\r\n') == 5
|
|
assert c.expect_exact('Fraction(1, 2)\r\n>>> ') == 0
|
|
assert c.send('-1/2\r\n') == 6
|
|
assert c.expect_exact('Fraction(-1, 2)\r\n>>> ') == 0
|
|
assert c.send('2**3/7\r\n') == 8
|
|
assert c.expect_exact('Fraction(8, 7)\r\n>>> ') == 0
|
|
assert c.send('(3 + 5)/7\r\n') == 11
|
|
assert c.expect_exact('Fraction(8, 7)\r\n>>> ') == 0
|
|
assert c.send('(0.5 + 1)/2\r\n') == 13
|
|
assert c.expect_exact('0.75\r\n>>> ') == 0
|
|
|
|
|
|
def test_bare_console_bare_division():
|
|
c = Console(f'{sys.executable} -m mpmath --no-ipython --no-wrap-division '
|
|
'--no-wrap-floats')
|
|
|
|
assert c.expect_exact('>>> ') == 0
|
|
assert c.send('1/2\r\n') == 5
|
|
assert c.expect_exact('0.5\r\n>>> ') == 0
|
|
|
|
|
|
def test_bare_console_without_ipython():
|
|
try:
|
|
import IPython
|
|
del IPython
|
|
pytest.skip('IPython is available')
|
|
except ImportError:
|
|
pass
|
|
|
|
c = Console(f'{sys.executable} -m mpmath')
|
|
|
|
assert c.expect_exact('>>> ') == 0
|
|
assert c.send('1 + 2\r\n') == 7
|
|
assert c.expect_exact('3\r\n>>> ') == 0
|
|
assert c.send('1/2\r\n') == 5
|
|
assert c.expect_exact('\r\nFraction(1, 2)\r\n>>> ') == 0
|
|
|
|
|
|
def test_ipython_console_bare_division_noauto():
|
|
pytest.importorskip('IPython')
|
|
|
|
c = Console(f'{sys.executable} -m mpmath --simple-prompt --no-wrap-floats '
|
|
"--no-wrap-division --colors 'NoColor' ")
|
|
|
|
assert c.expect_exact('\r\nIn [1]: ') == 0
|
|
assert c.send('1/2\r\n') == 5
|
|
assert c.expect_exact('\r\nOut[1]: 0.5\r\n\r\nIn [2]: ') == 0
|
|
|
|
|
|
def test_ipython_console_wrap_floats():
|
|
pytest.importorskip('IPython')
|
|
|
|
c = Console(f'{sys.executable} -m mpmath --simple-prompt --prec 100 '
|
|
"--colors 'NoColor' --no-pretty")
|
|
|
|
assert c.expect_exact('\r\nIn [1]: ') == 0
|
|
assert c.send('10.9\r\n') == 6
|
|
assert c.expect_exact("\r\nOut[1]: mpf('10.899999999999999999999999999995')\r\n\r\nIn [2]: ") == 0
|
|
assert c.send('def f():\r\n x = 1.1\n return x + 1\n\r\n\n') == 42
|
|
assert c.expect_exact("\r\n\r\nIn [3]: ") == 0
|
|
assert c.send('f()\r\n') == 5
|
|
assert c.expect_exact("\r\nOut[3]: mpf('2.0999999999999999999999999999987')\r\n\r\nIn [4]: ") == 0
|
|
|
|
|
|
def test_bare_console_wrap_floats():
|
|
c = Console(f'{sys.executable} -m mpmath --simple-prompt --no-ipython --prec 100 '
|
|
"--colors 'NoColor' --no-pretty")
|
|
|
|
assert c.expect_exact('>>> ') == 0
|
|
assert c.send("10.9\r\n") == 6
|
|
assert c.expect_exact("mpf('10.899999999999999999999999999995')\r\n>>> ") == 0
|
|
assert c.send("1e100\r\n") == 7
|
|
assert c.expect_exact("mpf('9.9999999999999999999999999999997e+99')\r\n>>> ") == 0
|
|
assert c.send("1E100\r\n") == 7
|
|
assert c.expect_exact("mpf('9.9999999999999999999999999999997e+99')\r\n>>> ") == 0
|
|
assert c.send("1+10.9j\r\n") == 9
|
|
assert c.expect_exact("mpc(real='1.0', imag='10.899999999999999999999999999995')\r\n>>> ") == 0
|
|
assert c.send("1+10.9J\r\n") == 9
|
|
assert c.expect_exact("mpc(real='1.0', imag='10.899999999999999999999999999995')\r\n>>> ") == 0
|
|
assert c.send('mpf(10.9)\r\n') == 11
|
|
assert c.expect_exact("mpf('10.899999999999999999999999999995')\r\n>>> ") == 0
|
|
assert c.send('0x1p-1\r\n') == 8
|
|
assert c.expect_exact("mpf('0.5')\r\n>>> ") == 0
|
|
assert c.send('0b1p+1\r\n') == 8
|
|
assert c.expect_exact("mpf('2.0')\r\n>>> ") == 0
|
|
|
|
|
|
@pytest.mark.skipif(sys.version_info < (3, 13),
|
|
reason="XXX: uses new REPL")
|
|
def test_bare_console_pretty():
|
|
c = Console(f'{sys.executable} -m mpmath --simple-prompt --no-ipython --prec 100 '
|
|
"--colors 'NoColor'", _dumb=False)
|
|
|
|
assert c.expect('>>> ') == 0
|
|
assert c.send("10.9\r\n") == 6
|
|
assert c.expect("10.899999999999999999999999999995") == 0
|
|
assert c.send("def f():\r\n x = ?\r\n\r\n") == 21
|
|
assert c.expect('SyntaxError:') == 0
|
|
assert c.send('def f():\r\n return 1.1\r\n\r\n') == 26
|
|
assert c.expect('>>> ') == 0
|
|
assert c.send("f()\r\n") == 5
|
|
assert c.expect('1.1000000000000000000000000000003') == 0
|
|
assert c.send("a = 2.1; a\r\n") == 12
|
|
assert c.expect('2.0999999999999999999999999999987') == 0
|
|
|
|
|
|
def test_mpmath_version():
|
|
c = Console(f'{sys.executable} -m mpmath --version')
|
|
|
|
assert c.expect(pexpect.EOF) == 0
|
|
assert c.before.startswith('1.')
|