More tests for line-based input transformers

IPython/core/tests/test_inputtransformer2_line.py

@@ -7,7 +7,7 b' import nose.tools as nt'
 
 from IPython.core import inputtransformer2 as ipt2
 
-SIMPLE = ("""\
+CELL_MAGIC = ("""\
 %%foo arg
 body 1
 body 2
@@ -16,6 +16,73 b" get_ipython().run_cell_magic('foo', 'arg', 'body 1\\\\nbody 2\\\\n')"
 """)
 
 def test_cell_magic():
-    for sample, expected in [SIMPLE]:
+    for sample, expected in [CELL_MAGIC]:
         nt.assert_equal(ipt2.cell_magic(sample.splitlines(keepends=True)),
                         expected.splitlines(keepends=True))
+
+CLASSIC_PROMPT = ("""\
+>>> for a in range(5):
+...     print(a)
+""", """\
+for a in range(5):
+    print(a)
+""")
+
+CLASSIC_PROMPT_L2 = ("""\
+for a in range(5):
+...     print(a)
+...     print(a ** 2)
+""", """\
+for a in range(5):
+    print(a)
+    print(a ** 2)
+""")
+
+def test_classic_prompt():
+    for sample, expected in [CLASSIC_PROMPT, CLASSIC_PROMPT_L2]:
+        nt.assert_equal(ipt2.classic_prompt(sample.splitlines(keepends=True)),
+                        expected.splitlines(keepends=True))
+
+IPYTHON_PROMPT = ("""\
+In [1]: for a in range(5):
+   ...:     print(a)
+""", """\
+for a in range(5):
+    print(a)
+""")
+
+IPYTHON_PROMPT_L2 = ("""\
+for a in range(5):
+   ...:     print(a)
+   ...:     print(a ** 2)
+""", """\
+for a in range(5):
+    print(a)
+    print(a ** 2)
+""")
+
+def test_ipython_prompt():
+    for sample, expected in [IPYTHON_PROMPT, IPYTHON_PROMPT_L2]:
+        nt.assert_equal(ipt2.ipython_prompt(sample.splitlines(keepends=True)),
+                        expected.splitlines(keepends=True))
+
+INDENT_SPACES = ("""\
+     if True:
+        a = 3
+""", """\
+if True:
+   a = 3
+""")
+
+INDENT_TABS = ("""\
+\tif True:
+\t\tb = 4
+""", """\
+if True:
+\tb = 4
+""")
+
+def test_leading_indent():
+    for sample, expected in [INDENT_SPACES, INDENT_TABS]:
+        nt.assert_equal(ipt2.leading_indent(sample.splitlines(keepends=True)),
+                        expected.splitlines(keepends=True))