|
|
"""Tests for input handlers.
|
|
|
"""
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Module imports
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
# our own packages
|
|
|
from IPython.core import autocall
|
|
|
import pytest
|
|
|
from collections.abc import Callable
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Globals
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
# Test functions
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
class CallableIndexable(object):
|
|
|
def __getitem__(self, idx): return True
|
|
|
def __call__(self, *args, **kws): return True
|
|
|
|
|
|
|
|
|
class Autocallable(autocall.IPyAutocall):
|
|
|
def __call__(self):
|
|
|
return "called"
|
|
|
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
|
"autocall, input, output",
|
|
|
[
|
|
|
# For many of the below, we're also checking that leading whitespace
|
|
|
# turns off the esc char, which it should unless there is a continuation
|
|
|
# line.
|
|
|
("1", '"no change"', '"no change"'), # normal
|
|
|
("1", "lsmagic", "get_ipython().run_line_magic('lsmagic', '')"), # magic
|
|
|
# Only explicit escapes or instances of IPyAutocallable should get
|
|
|
# expanded
|
|
|
("0", 'len "abc"', 'len "abc"'),
|
|
|
("0", "autocallable", "autocallable()"),
|
|
|
# Don't add extra brackets (gh-1117)
|
|
|
("0", "autocallable()", "autocallable()"),
|
|
|
("1", 'len "abc"', 'len("abc")'),
|
|
|
("1", 'len "abc";', 'len("abc");'), # ; is special -- moves out of parens
|
|
|
# Autocall is turned off if first arg is [] and the object
|
|
|
# is both callable and indexable. Like so:
|
|
|
("1", "len [1,2]", "len([1,2])"), # len doesn't support __getitem__...
|
|
|
("1", "call_idx [1]", "call_idx [1]"), # call_idx *does*..
|
|
|
("1", "call_idx 1", "call_idx(1)"),
|
|
|
("1", "len", "len"), # only at 2 does it auto-call on single args
|
|
|
("2", 'len "abc"', 'len("abc")'),
|
|
|
("2", 'len "abc";', 'len("abc");'),
|
|
|
("2", "len [1,2]", "len([1,2])"),
|
|
|
("2", "call_idx [1]", "call_idx [1]"),
|
|
|
("2", "call_idx 1", "call_idx(1)"),
|
|
|
# T his is what's different:
|
|
|
("2", "len", "len()"), # only at 2 does it auto-call on single args
|
|
|
("0", "Callable[[int], None]", "Callable[[int], None]"),
|
|
|
("1", "Callable[[int], None]", "Callable[[int], None]"),
|
|
|
("1", "Callable[[int], None]", "Callable[[int], None]"),
|
|
|
],
|
|
|
)
|
|
|
def test_handlers_I(autocall, input, output):
|
|
|
autocallable = Autocallable()
|
|
|
ip.user_ns["autocallable"] = autocallable
|
|
|
|
|
|
call_idx = CallableIndexable()
|
|
|
ip.user_ns["call_idx"] = call_idx
|
|
|
|
|
|
ip.user_ns["Callable"] = Callable
|
|
|
|
|
|
ip.run_line_magic("autocall", autocall)
|
|
|
assert ip.prefilter_manager.prefilter_lines(input) == output
|
|
|
ip.run_line_magic("autocall", "1")
|
|
|
|
