|
|
"""Tests for input manipulation machinery."""
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Imports
|
|
|
#-----------------------------------------------------------------------------
|
|
|
import pytest
|
|
|
|
|
|
from IPython.core.prefilter import AutocallChecker
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Tests
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
def test_prefilter():
|
|
|
"""Test user input conversions"""
|
|
|
|
|
|
# pairs of (raw, expected correct) input
|
|
|
pairs = [ ('2+2','2+2'),
|
|
|
]
|
|
|
|
|
|
for raw, correct in pairs:
|
|
|
assert ip.prefilter(raw) == correct
|
|
|
|
|
|
def test_prefilter_shadowed():
|
|
|
def dummy_magic(line): pass
|
|
|
|
|
|
prev_automagic_state = ip.automagic
|
|
|
ip.automagic = True
|
|
|
ip.autocall = 0
|
|
|
|
|
|
try:
|
|
|
# These should not be transformed - they are shadowed by other names
|
|
|
for name in ['if', 'zip', 'get_ipython']: # keyword, builtin, global
|
|
|
ip.register_magic_function(dummy_magic, magic_name=name)
|
|
|
res = ip.prefilter(name + " foo")
|
|
|
assert res == name + " foo"
|
|
|
del ip.magics_manager.magics["line"][name]
|
|
|
|
|
|
# These should be transformed
|
|
|
for name in ['fi', 'piz', 'nohtypi_teg']:
|
|
|
ip.register_magic_function(dummy_magic, magic_name=name)
|
|
|
res = ip.prefilter(name + " foo")
|
|
|
assert res != name + " foo"
|
|
|
del ip.magics_manager.magics["line"][name]
|
|
|
|
|
|
finally:
|
|
|
ip.automagic = prev_automagic_state
|
|
|
|
|
|
def test_autocall_binops():
|
|
|
"""See https://github.com/ipython/ipython/issues/81"""
|
|
|
ip.magic('autocall 2')
|
|
|
f = lambda x: x
|
|
|
ip.user_ns['f'] = f
|
|
|
try:
|
|
|
assert ip.prefilter("f 1") == "f(1)"
|
|
|
for t in ["f +1", "f -1"]:
|
|
|
assert ip.prefilter(t) == t
|
|
|
|
|
|
# Run tests again with a more permissive exclude_regexp, which will
|
|
|
# allow transformation of binary operations ('f -1' -> 'f(-1)').
|
|
|
pm = ip.prefilter_manager
|
|
|
ac = AutocallChecker(shell=pm.shell, prefilter_manager=pm,
|
|
|
config=pm.config)
|
|
|
try:
|
|
|
ac.priority = 1
|
|
|
ac.exclude_regexp = r'^[,&^\|\*/]|^is |^not |^in |^and |^or '
|
|
|
pm.sort_checkers()
|
|
|
|
|
|
assert ip.prefilter("f -1") == "f(-1)"
|
|
|
assert ip.prefilter("f +1") == "f(+1)"
|
|
|
finally:
|
|
|
pm.unregister_checker(ac)
|
|
|
finally:
|
|
|
ip.magic('autocall 0')
|
|
|
del ip.user_ns['f']
|
|
|
|
|
|
|
|
|
def test_issue_114():
|
|
|
"""Check that multiline string literals don't expand as magic
|
|
|
see http://github.com/ipython/ipython/issues/114"""
|
|
|
|
|
|
template = '"""\n%s\n"""'
|
|
|
# Store the current value of multi_line_specials and turn it off before
|
|
|
# running test, since it could be true (case in which the test doesn't make
|
|
|
# sense, as multiline string literals *will* expand as magic in that case).
|
|
|
msp = ip.prefilter_manager.multi_line_specials
|
|
|
ip.prefilter_manager.multi_line_specials = False
|
|
|
try:
|
|
|
for mgk in ip.magics_manager.lsmagic()['line']:
|
|
|
raw = template % mgk
|
|
|
assert ip.prefilter(raw) == raw
|
|
|
finally:
|
|
|
ip.prefilter_manager.multi_line_specials = msp
|
|
|
|
|
|
|
|
|
def test_prefilter_attribute_errors():
|
|
|
"""Capture exceptions thrown by user objects on attribute access.
|
|
|
|
|
|
See http://github.com/ipython/ipython/issues/988."""
|
|
|
|
|
|
class X(object):
|
|
|
def __getattr__(self, k):
|
|
|
raise ValueError('broken object')
|
|
|
def __call__(self, x):
|
|
|
return x
|
|
|
|
|
|
# Create a callable broken object
|
|
|
ip.user_ns['x'] = X()
|
|
|
ip.magic('autocall 2')
|
|
|
try:
|
|
|
# Even if x throws an attribute error when looking at its rewrite
|
|
|
# attribute, we should not crash. So the test here is simply making
|
|
|
# the prefilter call and not having an exception.
|
|
|
ip.prefilter('x 1')
|
|
|
finally:
|
|
|
del ip.user_ns['x']
|
|
|
ip.magic('autocall 0')
|
|
|
|
|
|
|
|
|
def test_autocall_should_support_unicode():
|
|
|
ip.magic('autocall 2')
|
|
|
ip.user_ns['π'] = lambda x: x
|
|
|
try:
|
|
|
assert ip.prefilter("π 3") == "π(3)"
|
|
|
finally:
|
|
|
ip.magic('autocall 0')
|
|
|
del ip.user_ns['π']
|
|
|
|
