Updated unit tests

1 files changed, 8 insertions, 4 deletions

diff --git a/tests/test_run_pyscript.py b/tests/test_run_pyscript.py
index 60a6798e..4c30d888 100644
--- a/tests/test_run_pyscript.py
+++ b/tests/test_run_pyscript.py
@@ -59,19 +59,23 @@ def test_run_pyscript_with_odd_file_names(base_app):
     """
     python_script = utils.quote_string('nothingweird.py')
     out, err = run_cmd(base_app, "run_pyscript {}".format(python_script))
-    assert "'nothingweird.py'" in err[0]
+    assert "No such file or directory: 'nothingweird.py'" in err[0]
 
     python_script = utils.quote_string('has   spaces.py')
     out, err = run_cmd(base_app, "run_pyscript {}".format(python_script))
-    assert "'has   spaces.py'" in err[0]
+    assert "No such file or directory: 'has   spaces.py'" in err[0]
+
+    # For remaining tests, mock input to get us passed the warning about not ending in .py
+    input_mock = mock.MagicMock(name='input', return_value='1')
+    builtins.input = input_mock
 
     python_script = utils.quote_string('"is_double_quoted.py"')
     out, err = run_cmd(base_app, "run_pyscript {}".format(python_script))
-    assert "'\"is_double_quoted.py\"'" in err[0]
+    assert "No such file or directory: '\"is_double_quoted.py\"'" in err[1]
 
     python_script = utils.quote_string("'is_single_quoted.py'")
     out, err = run_cmd(base_app, "run_pyscript {}".format(python_script))
-    assert "''is_single_quoted.py''" in err[0]
+    assert 'No such file or directory: "\'is_single_quoted.py\'"' in err[1]
 
 def test_run_pyscript_with_exception(base_app, request):
     test_dir = os.path.dirname(request.module.__file__)