From d258dd855074c8088a5f9fbbc3e748f6f010f9b8 2013-03-31 09:03:53
From: Thomas Kluyver <takowl@gmail.com>
Date: 2013-03-31 09:03:53
Subject: [PATCH] Correction to test framework.

---

diff --git a/IPython/core/tests/test_inputtransformer.py b/IPython/core/tests/test_inputtransformer.py
index 5839ce7..4ed0551 100644
--- a/IPython/core/tests/test_inputtransformer.py
+++ b/IPython/core/tests/test_inputtransformer.py
@@ -23,7 +23,11 @@ def transform_checker(tests, transformer):
     transformer = transformer()
     try:
         for inp, tr in tests:
-            nt.assert_equal(transformer.push(inp), tr)
+            if inp is None:
+                out = transformer.reset()
+            else:
+                out = transformer.push(inp)
+            nt.assert_equal(out, tr)
     finally:
         transformer.reset()