Fix shellapp tests

takluyver · takluyver · commit 6a2667e1b4c5 · 2016-06-01T12:14:10.000+01:00
diff --git a/IPython/core/tests/test_shellapp.py b/IPython/core/tests/test_shellapp.py
@@ -52,10 +52,9 @@ def test_py_script_file_attribute_interactively(self):
         src = "True\n"
         self.mktmp(src)
 
-        out = 'In [1]: False\n\nIn [2]:'
-        err = SQLITE_NOT_AVAILABLE_ERROR if sqlite_err_maybe else None
-        tt.ipexec_validate(self.fname, out, err, options=['-i'],
+        out, err = tt.ipexec(self.fname, options=['-i'],
                            commands=['"__file__" in globals()', 'exit()'])
+        self.assertIn("False", out)
 
     @dec.skip_win32
     @dec.skipif(PY3)
@@ -64,7 +63,6 @@ def test_py_script_file_compiler_directive(self):
         src = "from __future__ import division\n"
         self.mktmp(src)
 
-        out = 'In [1]: float\n\nIn [2]:'
-        err = SQLITE_NOT_AVAILABLE_ERROR if sqlite_err_maybe else None
-        tt.ipexec_validate(self.fname, out, err, options=['-i'],
+        out, err = tt.ipexec(self.fname, options=['-i'],
                            commands=['type(1/2)', 'exit()'])
+        self.assertIn('float', out)
diff --git a/IPython/testing/tools.py b/IPython/testing/tools.py
@@ -202,13 +202,7 @@ def ipexec(fname, options=None, commands=()):
     """
     if options is None: options = []
 
-    # For these subprocess calls, eliminate all prompt printing so we only see
-    # output from script execution
-    prompt_opts = [ '--PromptManager.in_template=""',
-                    '--PromptManager.in2_template=""',
-                    '--PromptManager.out_template=""'
-    ]
-    cmdargs = default_argv() + prompt_opts + options
+    cmdargs = default_argv() + options
 
     test_dir = os.path.dirname(__file__)
 
