From 2de4d31c4ce452371dc9858068902db55836f11a 2018-03-10 18:23:26
From: Thomas Kluyver <thomas@kluyver.me.uk>
Date: 2018-03-10 18:23:26
Subject: [PATCH] More tests for line-based input transformers

---

diff --git a/IPython/core/tests/test_inputtransformer2_line.py b/IPython/core/tests/test_inputtransformer2_line.py
index 9139e58..1458272 100644
--- a/IPython/core/tests/test_inputtransformer2_line.py
+++ b/IPython/core/tests/test_inputtransformer2_line.py
@@ -7,7 +7,7 @@ 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 @@ 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))