|
|
import unittest
|
|
|
import nose.tools as nt
|
|
|
|
|
|
from IPython.testing import tools as tt
|
|
|
from IPython.utils import py3compat
|
|
|
|
|
|
from IPython.core import inputtransformer
|
|
|
from IPython.core.tests.test_inputsplitter import syntax
|
|
|
|
|
|
def wrap_transform(transformer):
|
|
|
def transform(inp):
|
|
|
results = []
|
|
|
for line in inp:
|
|
|
res = transformer.push(line)
|
|
|
if res is not None:
|
|
|
results.append(res)
|
|
|
transformer.reset()
|
|
|
return results
|
|
|
|
|
|
return transform
|
|
|
|
|
|
cellmagic_tests = [
|
|
|
(['%%foo a', None], ["get_ipython().run_cell_magic('foo', 'a', '')"]),
|
|
|
(['%%bar 123', 'hello', ''], ["get_ipython().run_cell_magic('bar', '123', 'hello')"]),
|
|
|
]
|
|
|
|
|
|
def test_transform_cellmagic():
|
|
|
tt.check_pairs(wrap_transform(inputtransformer.cellmagic), cellmagic_tests)
|
|
|
|
|
|
esctransform_tests = [(i, [py3compat.u_format(ol) for ol in o]) for i,o in [
|
|
|
(['%pdef zip'], ["get_ipython().magic({u}'pdef zip')"]),
|
|
|
(['%abc def \\', 'ghi'], ["get_ipython().magic({u}'abc def ghi')"]),
|
|
|
(['%abc def \\', 'ghi\\', None], ["get_ipython().magic({u}'abc def ghi')"]),
|
|
|
]]
|
|
|
|
|
|
def test_transform_escaped():
|
|
|
tt.check_pairs(wrap_transform(inputtransformer.escaped_transformer), esctransform_tests)
|
|
|
|
|
|
def endhelp_test():
|
|
|
tt.check_pairs(inputtransformer.help_end.push, syntax['end_help'])
|
|
|
|
|
|
classic_prompt_tests = [
|
|
|
(['>>> a=1'], ['a=1']),
|
|
|
(['>>> a="""','... 123"""'], ['a="""', '123"""']),
|
|
|
(['a="""','... 123"""'], ['a="""', '... 123"""']),
|
|
|
]
|
|
|
|
|
|
def test_classic_prompt():
|
|
|
tt.check_pairs(wrap_transform(inputtransformer.classic_prompt), classic_prompt_tests)
|
|
|
|
|
|
ipy_prompt_tests = [
|
|
|
(['In [1]: a=1'], ['a=1']),
|
|
|
(['In [2]: a="""',' ...: 123"""'], ['a="""', '123"""']),
|
|
|
(['a="""',' ...: 123"""'], ['a="""', ' ...: 123"""']),
|
|
|
]
|
|
|
|
|
|
def test_ipy_prompt():
|
|
|
tt.check_pairs(wrap_transform(inputtransformer.ipy_prompt), ipy_prompt_tests)
|
|
|
|
|
|
leading_indent_tests = [
|
|
|
([' print "hi"'], ['print "hi"']),
|
|
|
([' for a in range(5):', ' a*2'], ['for a in range(5):', ' a*2']),
|
|
|
([' a="""',' 123"""'], ['a="""', '123"""']),
|
|
|
(['a="""',' 123"""'], ['a="""', ' 123"""']),
|
|
|
]
|
|
|
|
|
|
def test_leading_indent():
|
|
|
tt.check_pairs(wrap_transform(inputtransformer.leading_indent), leading_indent_tests)
|
|
|
|
|
|
assign_magic_tests = [(i, [py3compat.u_format(ol) for ol in o]) for i,o in [
|
|
|
(['a = %bc de \\', 'fg'], ["a = get_ipython().magic('bc de fg')"]),
|
|
|
(['a = %bc de \\', 'fg\\', None], ["a = get_ipython().magic('bc de fg')"]),
|
|
|
]]
|
|
|
|
|
|
def test_assign_magic():
|
|
|
tt.check_pairs(wrap_transform(inputtransformer.assign_from_magic), assign_magic_tests)
|
|
|
|
|
|
assign_system_tests = [(i, [py3compat.u_format(ol) for ol in o]) for i,o in [
|
|
|
(['a = !bc de \\', 'fg'], ["a = get_ipython().getoutput('bc de fg')"]),
|
|
|
(['a = !bc de \\', 'fg\\', None], ["a = get_ipython().getoutput('bc de fg')"]),
|
|
|
]]
|
|
|
|
|
|
def test_assign_system():
|
|
|
tt.check_pairs(wrap_transform(inputtransformer.assign_from_system), assign_system_tests)
|
|
|
|
