test_inputtransformer.py
499 lines
| 16.5 KiB
| text/x-python
|
PythonLexer
Thomas Kluyver
|
r10103 | import tokenize | ||
Thomas Kluyver
|
r10090 | import nose.tools as nt | ||
from IPython.testing import tools as tt | ||||
from IPython.utils import py3compat | ||||
Thomas Kluyver
|
r10095 | u_fmt = py3compat.u_format | ||
Thomas Kluyver
|
r10090 | |||
Thomas Kluyver
|
r10095 | from IPython.core import inputtransformer as ipt | ||
Thomas Kluyver
|
r10090 | |||
Thomas Kluyver
|
r10095 | def transform_and_reset(transformer): | ||
Thomas Kluyver
|
r10097 | transformer = transformer() | ||
Thomas Kluyver
|
r10090 | def transform(inp): | ||
Thomas Kluyver
|
r10095 | try: | ||
return transformer.push(inp) | ||||
finally: | ||||
transformer.reset() | ||||
Thomas Kluyver
|
r10090 | |||
return transform | ||||
Thomas Kluyver
|
r10095 | # Transformer tests | ||
Thomas Kluyver
|
r10252 | def transform_checker(tests, transformer, **kwargs): | ||
Thomas Kluyver
|
r10095 | """Utility to loop over test inputs""" | ||
Thomas Kluyver
|
r10252 | transformer = transformer(**kwargs) | ||
Thomas Kluyver
|
r10095 | try: | ||
for inp, tr in tests: | ||||
Thomas Kluyver
|
r10098 | if inp is None: | ||
out = transformer.reset() | ||||
else: | ||||
out = transformer.push(inp) | ||||
nt.assert_equal(out, tr) | ||||
Thomas Kluyver
|
r10095 | finally: | ||
transformer.reset() | ||||
# Data for all the syntax tests in the form of lists of pairs of | ||||
# raw/transformed input. We store it here as a global dict so that we can use | ||||
# it both within single-function tests and also to validate the behavior of the | ||||
# larger objects | ||||
Matthias Bussonnier
|
r27630 | syntax = dict( | ||
assign_system=[ | ||||
(i, py3compat.u_format(o)) | ||||
for i, o in [ | ||||
(u"a =! ls", "a = get_ipython().getoutput('ls')"), | ||||
(u"b = !ls", "b = get_ipython().getoutput('ls')"), | ||||
(u"c= !ls", "c = get_ipython().getoutput('ls')"), | ||||
(u"d == !ls", u"d == !ls"), # Invalid syntax, but we leave == alone. | ||||
("x=1", "x=1"), # normal input is unmodified | ||||
(" ", " "), # blank lines are kept intact | ||||
# Tuple unpacking | ||||
( | ||||
u"a, b = !echo 'a\\nb'", | ||||
u"a, b = get_ipython().getoutput(\"echo 'a\\\\nb'\")", | ||||
), | ||||
(u"a,= !echo 'a'", u"a, = get_ipython().getoutput(\"echo 'a'\")"), | ||||
( | ||||
u"a, *bc = !echo 'a\\nb\\nc'", | ||||
u"a, *bc = get_ipython().getoutput(\"echo 'a\\\\nb\\\\nc'\")", | ||||
), | ||||
# Tuple unpacking with regular Python expressions, not our syntax. | ||||
(u"a, b = range(2)", u"a, b = range(2)"), | ||||
(u"a, = range(1)", u"a, = range(1)"), | ||||
(u"a, *bc = range(3)", u"a, *bc = range(3)"), | ||||
] | ||||
], | ||||
assign_magic=[ | ||||
(i, py3compat.u_format(o)) | ||||
for i, o in [ | ||||
(u"a =% who", "a = get_ipython().run_line_magic('who', '')"), | ||||
(u"b = %who", "b = get_ipython().run_line_magic('who', '')"), | ||||
(u"c= %ls", "c = get_ipython().run_line_magic('ls', '')"), | ||||
(u"d == %ls", u"d == %ls"), # Invalid syntax, but we leave == alone. | ||||
("x=1", "x=1"), # normal input is unmodified | ||||
(" ", " "), # blank lines are kept intact | ||||
(u"a, b = %foo", u"a, b = get_ipython().run_line_magic('foo', '')"), | ||||
] | ||||
], | ||||
classic_prompt=[ | ||||
(">>> x=1", "x=1"), | ||||
("x=1", "x=1"), # normal input is unmodified | ||||
(" ", " "), # blank lines are kept intact | ||||
], | ||||
ipy_prompt=[ | ||||
("In [1]: x=1", "x=1"), | ||||
("x=1", "x=1"), # normal input is unmodified | ||||
(" ", " "), # blank lines are kept intact | ||||
], | ||||
# Tests for the escape transformer to leave normal code alone | ||||
escaped_noesc=[ | ||||
(" ", " "), | ||||
("x=1", "x=1"), | ||||
], | ||||
# System calls | ||||
escaped_shell=[ | ||||
(i, py3compat.u_format(o)) | ||||
for i, o in [ | ||||
(u"!ls", "get_ipython().system('ls')"), | ||||
# Double-escape shell, this means to capture the output of the | ||||
# subprocess and return it | ||||
(u"!!ls", "get_ipython().getoutput('ls')"), | ||||
] | ||||
], | ||||
# Help/object info | ||||
escaped_help=[ | ||||
(i, py3compat.u_format(o)) | ||||
for i, o in [ | ||||
(u"?", "get_ipython().show_usage()"), | ||||
(u"?x1", "get_ipython().run_line_magic('pinfo', 'x1')"), | ||||
(u"??x2", "get_ipython().run_line_magic('pinfo2', 'x2')"), | ||||
(u"?a.*s", "get_ipython().run_line_magic('psearch', 'a.*s')"), | ||||
(u"?%hist1", "get_ipython().run_line_magic('pinfo', '%hist1')"), | ||||
(u"?%%hist2", "get_ipython().run_line_magic('pinfo', '%%hist2')"), | ||||
(u"?abc = qwe", "get_ipython().run_line_magic('pinfo', 'abc')"), | ||||
] | ||||
], | ||||
end_help=[ | ||||
(i, py3compat.u_format(o)) | ||||
for i, o in [ | ||||
(u"x3?", "get_ipython().run_line_magic('pinfo', 'x3')"), | ||||
(u"x4??", "get_ipython().run_line_magic('pinfo2', 'x4')"), | ||||
(u"%hist1?", "get_ipython().run_line_magic('pinfo', '%hist1')"), | ||||
(u"%hist2??", "get_ipython().run_line_magic('pinfo2', '%hist2')"), | ||||
(u"%%hist3?", "get_ipython().run_line_magic('pinfo', '%%hist3')"), | ||||
(u"%%hist4??", "get_ipython().run_line_magic('pinfo2', '%%hist4')"), | ||||
(u"Ï€.foo?", "get_ipython().run_line_magic('pinfo', 'Ï€.foo')"), | ||||
(u"f*?", "get_ipython().run_line_magic('psearch', 'f*')"), | ||||
(u"ax.*aspe*?", "get_ipython().run_line_magic('psearch', 'ax.*aspe*')"), | ||||
(u"a = abc?", "get_ipython().run_line_magic('pinfo', 'abc')"), | ||||
(u"a = abc.qe??", "get_ipython().run_line_magic('pinfo2', 'abc.qe')"), | ||||
(u"a = *.items?", "get_ipython().run_line_magic('psearch', '*.items')"), | ||||
(u"plot(a?", "get_ipython().run_line_magic('pinfo', 'a')"), | ||||
(u"a*2 #comment?", "a*2 #comment?"), | ||||
] | ||||
], | ||||
# Explicit magic calls | ||||
escaped_magic=[ | ||||
(i, py3compat.u_format(o)) | ||||
for i, o in [ | ||||
(u"%cd", "get_ipython().run_line_magic('cd', '')"), | ||||
(u"%cd /home", "get_ipython().run_line_magic('cd', '/home')"), | ||||
# Backslashes need to be escaped. | ||||
(u"%cd C:\\User", "get_ipython().run_line_magic('cd', 'C:\\\\User')"), | ||||
(u" %magic", " get_ipython().run_line_magic('magic', '')"), | ||||
] | ||||
], | ||||
# Quoting with separate arguments | ||||
escaped_quote=[ | ||||
(",f", 'f("")'), | ||||
(",f x", 'f("x")'), | ||||
(" ,f y", ' f("y")'), | ||||
(",f a b", 'f("a", "b")'), | ||||
], | ||||
# Quoting with single argument | ||||
escaped_quote2=[ | ||||
(";f", 'f("")'), | ||||
(";f x", 'f("x")'), | ||||
(" ;f y", ' f("y")'), | ||||
(";f a b", 'f("a b")'), | ||||
], | ||||
# Simply apply parens | ||||
escaped_paren=[ | ||||
("/f", "f()"), | ||||
("/f x", "f(x)"), | ||||
(" /f y", " f(y)"), | ||||
("/f a b", "f(a, b)"), | ||||
], | ||||
# Check that we transform prompts before other transforms | ||||
mixed=[ | ||||
(i, py3compat.u_format(o)) | ||||
for i, o in [ | ||||
(u"In [1]: %lsmagic", "get_ipython().run_line_magic('lsmagic', '')"), | ||||
(u">>> %lsmagic", "get_ipython().run_line_magic('lsmagic', '')"), | ||||
(u"In [2]: !ls", "get_ipython().system('ls')"), | ||||
(u"In [3]: abs?", "get_ipython().run_line_magic('pinfo', 'abs')"), | ||||
(u"In [4]: b = %who", "b = get_ipython().run_line_magic('who', '')"), | ||||
] | ||||
], | ||||
) | ||||
Thomas Kluyver
|
r10090 | |||
Thomas Kluyver
|
r10095 | # multiline syntax examples. Each of these should be a list of lists, with | ||
# each entry itself having pairs of raw/transformed input. The union (with | ||||
# '\n'.join() of the transformed inputs is what the splitter should produce | ||||
# when fed the raw lines one at a time via push. | ||||
syntax_ml = \ | ||||
dict(classic_prompt = | ||||
[ [('>>> for i in range(10):','for i in range(10):'), | ||||
('... print i',' print i'), | ||||
('... ', ''), | ||||
], | ||||
[('>>> a="""','a="""'), | ||||
('... 123"""','123"""'), | ||||
], | ||||
[('a="""','a="""'), | ||||
MinRK
|
r12309 | ('... 123','123'), | ||
('... 456"""','456"""'), | ||||
], | ||||
[('a="""','a="""'), | ||||
('>>> 123','123'), | ||||
('... 456"""','456"""'), | ||||
Thomas Kluyver
|
r10652 | ], | ||
[('a="""','a="""'), | ||||
('123','123'), | ||||
('... 456"""','... 456"""'), | ||||
Thomas Kluyver
|
r10095 | ], | ||
MinRK
|
r12231 | [('....__class__','....__class__'), | ||
], | ||||
[('a=5', 'a=5'), | ||||
MinRK
|
r12309 | ('...', ''), | ||
MinRK
|
r12231 | ], | ||
Thomas Kluyver
|
r11056 | [('>>> def f(x):', 'def f(x):'), | ||
('...', ''), | ||||
('... return x', ' return x'), | ||||
], | ||||
Thomas Kluyver
|
r17037 | [('board = """....', 'board = """....'), | ||
('....', '....'), | ||||
('...."""', '...."""'), | ||||
], | ||||
Thomas Kluyver
|
r10095 | ], | ||
Thomas Kluyver
|
r10090 | |||
Thomas Kluyver
|
r10095 | ipy_prompt = | ||
[ [('In [24]: for i in range(10):','for i in range(10):'), | ||||
(' ....: print i',' print i'), | ||||
(' ....: ', ''), | ||||
], | ||||
Thomas Kluyver
|
r13934 | [('In [24]: for i in range(10):','for i in range(10):'), | ||
# Qt console prompts expand with spaces, not dots | ||||
(' ...: print i',' print i'), | ||||
(' ...: ', ''), | ||||
], | ||||
Pascal Bugnion
|
r19036 | [('In [24]: for i in range(10):','for i in range(10):'), | ||
Pascal Bugnion
|
r19038 | # Sometimes whitespace preceding '...' has been removed | ||
Pascal Bugnion
|
r19036 | ('...: print i',' print i'), | ||
('...: ', ''), | ||||
], | ||||
Pascal Bugnion
|
r19037 | [('In [24]: for i in range(10):','for i in range(10):'), | ||
# Space after last continuation prompt has been removed (issue #6674) | ||||
('...: print i',' print i'), | ||||
('...:', ''), | ||||
], | ||||
Thomas Kluyver
|
r10095 | [('In [2]: a="""','a="""'), | ||
(' ...: 123"""','123"""'), | ||||
], | ||||
[('a="""','a="""'), | ||||
MinRK
|
r12309 | (' ...: 123','123'), | ||
(' ...: 456"""','456"""'), | ||||
], | ||||
[('a="""','a="""'), | ||||
('In [1]: 123','123'), | ||||
(' ...: 456"""','456"""'), | ||||
Thomas Kluyver
|
r10652 | ], | ||
[('a="""','a="""'), | ||||
('123','123'), | ||||
(' ...: 456"""',' ...: 456"""'), | ||||
Thomas Kluyver
|
r10095 | ], | ||
], | ||||
Thomas Kluyver
|
r10090 | |||
Thomas Kluyver
|
r10112 | multiline_datastructure_prompt = | ||
Thomas Kluyver
|
r10095 | [ [('>>> a = [1,','a = [1,'), | ||
('... 2]','2]'), | ||||
], | ||||
], | ||||
Thomas Kluyver
|
r10112 | |||
multiline_datastructure = | ||||
[ [('b = ("%s"', None), | ||||
('# comment', None), | ||||
('%foo )', 'b = ("%s"\n# comment\n%foo )'), | ||||
], | ||||
], | ||||
Thomas Kluyver
|
r10095 | |||
Thomas Kluyver
|
r12352 | multiline_string = | ||
[ [("'''foo?", None), | ||||
("bar'''", "'''foo?\nbar'''"), | ||||
], | ||||
], | ||||
Thomas Kluyver
|
r10095 | leading_indent = | ||
[ [(' print "hi"','print "hi"'), | ||||
Thomas Kluyver
|
r10097 | ], | ||
[(' for a in range(5):','for a in range(5):'), | ||||
Thomas Kluyver
|
r10095 | (' a*2',' a*2'), | ||
], | ||||
[(' a="""','a="""'), | ||||
(' 123"""','123"""'), | ||||
], | ||||
[('a="""','a="""'), | ||||
(' 123"""',' 123"""'), | ||||
], | ||||
], | ||||
cellmagic = | ||||
Thomas Kluyver
|
r10097 | [ [(u'%%foo a', None), | ||
adityausathe
|
r23764 | (None, u_fmt("get_ipython().run_cell_magic('foo', 'a', '')")), | ||
Thomas Kluyver
|
r10095 | ], | ||
Thomas Kluyver
|
r10097 | [(u'%%bar 123', None), | ||
(u'hello', None), | ||||
adityausathe
|
r23764 | (None , u_fmt("get_ipython().run_cell_magic('bar', '123', 'hello')")), | ||
Thomas Kluyver
|
r10095 | ], | ||
MinRK
|
r11473 | [(u'a=5', 'a=5'), | ||
(u'%%cellmagic', '%%cellmagic'), | ||||
], | ||||
Thomas Kluyver
|
r10095 | ], | ||
escaped = | ||||
[ [('%abc def \\', None), | ||||
adityausathe
|
r23764 | ('ghi', u_fmt("get_ipython().run_line_magic('abc', 'def ghi')")), | ||
Thomas Kluyver
|
r10095 | ], | ||
[('%abc def \\', None), | ||||
('ghi\\', None), | ||||
adityausathe
|
r23764 | (None, u_fmt("get_ipython().run_line_magic('abc', 'def ghi')")), | ||
Thomas Kluyver
|
r10095 | ], | ||
], | ||||
assign_magic = | ||||
Thomas Kluyver
|
r10097 | [ [(u'a = %bc de \\', None), | ||
adityausathe
|
r23764 | (u'fg', u_fmt("a = get_ipython().run_line_magic('bc', 'de fg')")), | ||
Thomas Kluyver
|
r10095 | ], | ||
Thomas Kluyver
|
r10097 | [(u'a = %bc de \\', None), | ||
(u'fg\\', None), | ||||
adityausathe
|
r23764 | (None, u_fmt("a = get_ipython().run_line_magic('bc', 'de fg')")), | ||
Thomas Kluyver
|
r10095 | ], | ||
], | ||||
assign_system = | ||||
Thomas Kluyver
|
r10097 | [ [(u'a = !bc de \\', None), | ||
adityausathe
|
r23764 | (u'fg', u_fmt("a = get_ipython().getoutput('bc de fg')")), | ||
Thomas Kluyver
|
r10095 | ], | ||
Thomas Kluyver
|
r10097 | [(u'a = !bc de \\', None), | ||
(u'fg\\', None), | ||||
adityausathe
|
r23764 | (None, u_fmt("a = get_ipython().getoutput('bc de fg')")), | ||
Thomas Kluyver
|
r10095 | ], | ||
], | ||||
) | ||||
Thomas Kluyver
|
r10090 | |||
Thomas Kluyver
|
r10091 | |||
Thomas Kluyver
|
r10095 | def test_assign_system(): | ||
tt.check_pairs(transform_and_reset(ipt.assign_from_system), syntax['assign_system']) | ||||
def test_assign_magic(): | ||||
tt.check_pairs(transform_and_reset(ipt.assign_from_magic), syntax['assign_magic']) | ||||
Thomas Kluyver
|
r10091 | |||
def test_classic_prompt(): | ||||
Thomas Kluyver
|
r10095 | tt.check_pairs(transform_and_reset(ipt.classic_prompt), syntax['classic_prompt']) | ||
for example in syntax_ml['classic_prompt']: | ||||
transform_checker(example, ipt.classic_prompt) | ||||
Thomas Kluyver
|
r10112 | for example in syntax_ml['multiline_datastructure_prompt']: | ||
Thomas Kluyver
|
r10097 | transform_checker(example, ipt.classic_prompt) | ||
Thomas Kluyver
|
r10091 | |||
Thomas Kluyver
|
r21954 | # Check that we don't transform the second line if the first is obviously | ||
# IPython syntax | ||||
transform_checker([ | ||||
(u'%foo', '%foo'), | ||||
(u'>>> bar', '>>> bar'), | ||||
], ipt.classic_prompt) | ||||
Thomas Kluyver
|
r10091 | |||
def test_ipy_prompt(): | ||||
Thomas Kluyver
|
r10095 | tt.check_pairs(transform_and_reset(ipt.ipy_prompt), syntax['ipy_prompt']) | ||
for example in syntax_ml['ipy_prompt']: | ||||
transform_checker(example, ipt.ipy_prompt) | ||||
Thomas Kluyver
|
r10091 | |||
Thomas Kluyver
|
r21954 | # Check that we don't transform the second line if we're inside a cell magic | ||
transform_checker([ | ||||
(u'%%foo', '%%foo'), | ||||
(u'In [1]: bar', 'In [1]: bar'), | ||||
], ipt.ipy_prompt) | ||||
Thomas Kluyver
|
r10107 | def test_assemble_logical_lines(): | ||
tests = \ | ||||
[ [(u"a = \\", None), | ||||
Thomas Kluyver
|
r10112 | (u"123", u"a = 123"), | ||
Thomas Kluyver
|
r10107 | ], | ||
[(u"a = \\", None), # Test resetting when within a multi-line string | ||||
(u"12 *\\", None), | ||||
Thomas Kluyver
|
r10112 | (None, u"a = 12 *"), | ||
], | ||||
[(u"# foo\\", u"# foo\\"), # Comments can't be continued like this | ||||
Thomas Kluyver
|
r10107 | ], | ||
] | ||||
for example in tests: | ||||
transform_checker(example, ipt.assemble_logical_lines) | ||||
def test_assemble_python_lines(): | ||||
tests = \ | ||||
[ [(u"a = '''", None), | ||||
(u"abc'''", u"a = '''\nabc'''"), | ||||
], | ||||
[(u"a = '''", None), # Test resetting when within a multi-line string | ||||
(u"def", None), | ||||
(None, u"a = '''\ndef"), | ||||
], | ||||
[(u"a = [1,", None), | ||||
(u"2]", u"a = [1,\n2]"), | ||||
], | ||||
[(u"a = [1,", None), # Test resetting when within a multi-line string | ||||
(u"2,", None), | ||||
(None, u"a = [1,\n2,"), | ||||
], | ||||
Thomas Kluyver
|
r23183 | [(u"a = '''", None), # Test line continuation within a multi-line string | ||
(u"abc\\", None), | ||||
(u"def", None), | ||||
(u"'''", u"a = '''\nabc\\\ndef\n'''"), | ||||
], | ||||
Thomas Kluyver
|
r10112 | ] + syntax_ml['multiline_datastructure'] | ||
Thomas Kluyver
|
r10107 | for example in tests: | ||
transform_checker(example, ipt.assemble_python_lines) | ||||
Thomas Kluyver
|
r10095 | def test_help_end(): | ||
tt.check_pairs(transform_and_reset(ipt.help_end), syntax['end_help']) | ||||
Thomas Kluyver
|
r10091 | |||
Thomas Kluyver
|
r10095 | def test_escaped_noesc(): | ||
Thomas Kluyver
|
r10107 | tt.check_pairs(transform_and_reset(ipt.escaped_commands), syntax['escaped_noesc']) | ||
Thomas Kluyver
|
r10092 | |||
Thomas Kluyver
|
r10095 | def test_escaped_shell(): | ||
Thomas Kluyver
|
r10107 | tt.check_pairs(transform_and_reset(ipt.escaped_commands), syntax['escaped_shell']) | ||
Thomas Kluyver
|
r10092 | |||
Thomas Kluyver
|
r10095 | def test_escaped_help(): | ||
Thomas Kluyver
|
r10107 | tt.check_pairs(transform_and_reset(ipt.escaped_commands), syntax['escaped_help']) | ||
Thomas Kluyver
|
r10095 | |||
def test_escaped_magic(): | ||||
Thomas Kluyver
|
r10107 | tt.check_pairs(transform_and_reset(ipt.escaped_commands), syntax['escaped_magic']) | ||
Thomas Kluyver
|
r10095 | |||
def test_escaped_quote(): | ||||
Thomas Kluyver
|
r10107 | tt.check_pairs(transform_and_reset(ipt.escaped_commands), syntax['escaped_quote']) | ||
Thomas Kluyver
|
r10095 | |||
def test_escaped_quote2(): | ||||
Thomas Kluyver
|
r10107 | tt.check_pairs(transform_and_reset(ipt.escaped_commands), syntax['escaped_quote2']) | ||
Thomas Kluyver
|
r10095 | |||
def test_escaped_paren(): | ||||
Thomas Kluyver
|
r10107 | tt.check_pairs(transform_and_reset(ipt.escaped_commands), syntax['escaped_paren']) | ||
Thomas Kluyver
|
r10095 | |||
def test_cellmagic(): | ||||
for example in syntax_ml['cellmagic']: | ||||
transform_checker(example, ipt.cellmagic) | ||||
Thomas Kluyver
|
r10252 | |||
line_example = [(u'%%bar 123', None), | ||||
(u'hello', None), | ||||
adityausathe
|
r23764 | (u'' , u_fmt("get_ipython().run_cell_magic('bar', '123', 'hello')")), | ||
Thomas Kluyver
|
r10252 | ] | ||
transform_checker(line_example, ipt.cellmagic, end_on_blank_line=True) | ||||
Thomas Kluyver
|
r10097 | |||
def test_has_comment(): | ||||
tests = [('text', False), | ||||
('text #comment', True), | ||||
('text #comment\n', True), | ||||
('#comment', True), | ||||
('#comment\n', True), | ||||
('a = "#string"', False), | ||||
('a = "#string" # comment', True), | ||||
('a #comment not "string"', True), | ||||
] | ||||
tt.check_pairs(ipt.has_comment, tests) | ||||
Thomas Kluyver
|
r10103 | |||
@ipt.TokenInputTransformer.wrap | ||||
def decistmt(tokens): | ||||
"""Substitute Decimals for floats in a string of statements. | ||||
Based on an example from the tokenize module docs. | ||||
""" | ||||
result = [] | ||||
for toknum, tokval, _, _, _ in tokens: | ||||
if toknum == tokenize.NUMBER and '.' in tokval: # replace NUMBER tokens | ||||
for newtok in [ | ||||
(tokenize.NAME, 'Decimal'), | ||||
(tokenize.OP, '('), | ||||
(tokenize.STRING, repr(tokval)), | ||||
(tokenize.OP, ')') | ||||
]: | ||||
yield newtok | ||||
else: | ||||
yield (toknum, tokval) | ||||
def test_token_input_transformer(): | ||||
adityausathe
|
r23764 | tests = [(u'1.2', u_fmt(u"Decimal ('1.2')")), | ||
Thomas Kluyver
|
r10103 | (u'"1.2"', u'"1.2"'), | ||
] | ||||
tt.check_pairs(transform_and_reset(decistmt), tests) | ||||
ml_tests = \ | ||||
[ [(u"a = 1.2; b = '''x", None), | ||||
adityausathe
|
r23764 | (u"y'''", u_fmt(u"a =Decimal ('1.2');b ='''x\ny'''")), | ||
Thomas Kluyver
|
r10103 | ], | ||
Thomas Kluyver
|
r10106 | [(u"a = [1.2,", None), | ||
adityausathe
|
r23764 | (u"3]", u_fmt(u"a =[Decimal ('1.2'),\n3 ]")), | ||
Thomas Kluyver
|
r10103 | ], | ||
[(u"a = '''foo", None), # Test resetting when within a multi-line string | ||||
(u"bar", None), | ||||
(None, u"a = '''foo\nbar"), | ||||
], | ||||
] | ||||
for example in ml_tests: | ||||
transform_checker(example, decistmt) | ||||