def ipexec(fname, options=None, commands=()):
"""
Shamelessly stolen from ipython tests but without relative path requirements
"""
if options is None: options = []
cmdargs = default_argv() + options
ipython_cmd = get_ipython_cmd()
full_cmd = ipython_cmd + cmdargs + ['--', fname]
print(full_cmd)
env = os.environ.copy()
# FIXME: ignore all warnings in ipexec while we have shims
# should we keep suppressing warnings here, even after removing shims?
env['PYTHONWARNINGS'] = 'ignore'
# env.pop('PYTHONWARNINGS', None) # Avoid extraneous warnings appearing on stderr
for k, v in env.items():
# Debug a bizarre failure we've seen on Windows:
# TypeError: environment can only contain strings
if not isinstance(v, str):
print(k, v)
p = Popen(full_cmd, stdout=PIPE, stderr=PIPE, stdin=PIPE, env=env)
out, err = p.communicate(input='\n'.join(commands) or None)
return out, err
def test_IPython():
ipython = get_ipython_cmd()
SMM = b'\x1b[?1034h'
commands = b"""\
1
raise Exception
undefined
def f():
pass
f()
"""
# First the control (without iterm2_tools)
p = subprocess.Popen(ipython + ['--quick', '--colors=NoColor', '--no-banner'],
stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE)
stdout, stderr = p.communicate(input=commands)
# Different versions of readline do different things with the smm code.
stdout = stdout.replace(SMM, b'').strip()
assert stdout == b"""\
In [1]: Out[1]: 1
In [2]: \n\
In [2]: ---------------------------------------------------------------------------
Exception Traceback (most recent call last)
<ipython-input-2-fca2ab0ca76b> in <module>()
----> 1 raise Exception
Exception: \n\nIn [3]: ---------------------------------------------------------------------------
NameError Traceback (most recent call last)
<ipython-input-3-002bcaa7be0e> in <module>()
----> 1 undefined
NameError: name 'undefined' is not defined
In [4]: ...: ...: \n\
In [5]: \n\
In [6]: \n\
In [6]: \n\
Do you really want to exit ([y]/n)?\
"""
assert stderr == b''
# Now the same thing with iterm2_tools.ipython
p = subprocess.Popen(ipython + ['--quick', '--colors=NoColor',
'--no-banner', '--ext=iterm2_tools.ipython'], stdout=subprocess.PIPE,
stderr=subprocess.PIPE, stdin=subprocess.PIPE)
# Things of note here:
# - There are 8 prompts (3 with no input, the second, sixth, and
# seventh). The sixth is empty because f() returns None, and the eighth
# is not empty because of the exit confirmation.
# - The color codes are outside of the iterm2 codes. This is because of
# the way IPython handles color codes. See the note in ipython.py.
# - The D code (after_output) should always go right before the A code
# (before_prompt).
# - The fourth and fifth D code (after_output), corresponding to the third
# and fourth prompt, should have D;1 (exceptions). The rest should have
# D;0.
# - The A, B, and D codes should be surrounded by \001 and \002 (C,
# before_output) does not need it because it is not in the prompt.
stdout, stderr = p.communicate(input=commands)
# Different versions of readline do different things with the smm code.
stdout = stdout.replace(SMM, b'').strip()
# Note: this test will fail in versions of IPython < 4.1.0 because of a
# bug. See https://github.com/ipython/ipython/issues/8724 and
# https://github.com/ipython/ipython/pull/8738.
assert stdout == b"""\
\x01\x1b]133;D;0\x07\x02\x01\x1b]133;A\x07\x02In [1]: \x01\x1b]133;B\x07\x02\x1b]133;C\x07Out[1]: 1
\x01\x1b]133;D;0\x07\x02\x01\x1b]133;A\x07\x02In [2]: \x01\x1b]133;B\x07\x02
\x01\x1b]133;D;0\x07\x02\x01\x1b]133;A\x07\x02In [2]: \x01\x1b]133;B\x07\x02\x1b]133;C\x07---------------------------------------------------------------------------
Exception Traceback (most recent call last)
<ipython-input-2-fca2ab0ca76b> in <module>()
----> 1 raise Exception
Exception: \n\n\x01\x1b]133;D;1\x07\x02\x01\x1b]133;A\x07\x02In [3]: \x01\x1b]133;B\x07\x02\x1b]133;C\x07---------------------------------------------------------------------------
NameError Traceback (most recent call last)
<ipython-input-3-002bcaa7be0e> in <module>()
----> 1 undefined
NameError: name 'undefined' is not defined
\x01\x1b]133;D;1\x07\x02\x01\x1b]133;A\x07\x02In [4]: \x01\x1b]133;B\x07\x02 ...: ...: \x1b]133;C\x07
\x01\x1b]133;D;0\x07\x02\x01\x1b]133;A\x07\x02In [5]: \x01\x1b]133;B\x07\x02\x1b]133;C\x07
\x01\x1b]133;D;0\x07\x02\x01\x1b]133;A\x07\x02In [6]: \x01\x1b]133;B\x07\x02
\x01\x1b]133;D;0\x07\x02\x01\x1b]133;A\x07\x02In [6]: \x01\x1b]133;B\x07\x02
Do you really want to exit ([y]/n)?\
"""
assert stderr == b''
# Ideally all the codes would be bytes in Python 3, but bytes don't have a
# format (even in Python 3.5).
stdout = stdout.decode('ascii')
AFTER_OUTPUT0 = AFTER_OUTPUT.format(command_status=0)
AFTER_OUTPUT1 = AFTER_OUTPUT.format(command_status=1)
assert (stdout.count(AFTER_OUTPUT0) ==
stdout.count(readline_invisible(AFTER_OUTPUT0)) ==
stdout.count(readline_invisible(AFTER_OUTPUT0) +
readline_invisible(BEFORE_PROMPT)) == 6)
assert (stdout.count(AFTER_OUTPUT1) ==
stdout.count(readline_invisible(AFTER_OUTPUT1)) ==
stdout.count(readline_invisible(AFTER_OUTPUT1) +
readline_invisible(BEFORE_PROMPT)) == 2)
assert (stdout.count(BEFORE_PROMPT) ==
stdout.count(readline_invisible(BEFORE_PROMPT)) == 8)
assert (stdout.count(AFTER_PROMPT) ==
stdout.count(readline_invisible(AFTER_PROMPT)) == 8)
assert stdout.count(BEFORE_OUTPUT) == 5
assert stdout.count(readline_invisible(BEFORE_OUTPUT)) == 0
AFTER_OUTPUT_RE = re.compile(re.escape(AFTER_OUTPUT.format(command_status='DUMMY')).replace("DUMMY",
r'\d'))
assert re.findall(AFTER_OUTPUT_RE, stdout) == [
AFTER_OUTPUT0,
AFTER_OUTPUT0,
AFTER_OUTPUT0,
AFTER_OUTPUT1,
AFTER_OUTPUT1,
AFTER_OUTPUT0,
AFTER_OUTPUT0,
AFTER_OUTPUT0,
]
AFTER_PROMPT_RE = re.escape(AFTER_PROMPT)
BEFORE_OUTPUT_RE = re.escape(BEFORE_OUTPUT)
AFTER_PROMPT_OR_BEFORE_OUTPUT_RE = re.compile('(%s|%s)' % (AFTER_PROMPT_RE, BEFORE_OUTPUT_RE))
assert re.findall(AFTER_PROMPT_OR_BEFORE_OUTPUT_RE, stdout) == [
AFTER_PROMPT, BEFORE_OUTPUT, # non-empty prompt
AFTER_PROMPT, # empty prompt
AFTER_PROMPT, BEFORE_OUTPUT,
AFTER_PROMPT, BEFORE_OUTPUT,
AFTER_PROMPT, BEFORE_OUTPUT,
AFTER_PROMPT, BEFORE_OUTPUT,
AFTER_PROMPT,
AFTER_PROMPT,
]
# Copyright (c) IPython Development Team.
# Distributed under the terms of the Modified BSD License.
import io
import os
import glob
import shutil
import unittest
from IPython.nbformat import v4, write
from IPython.utils.tempdir import TemporaryWorkingDirectory
from IPython.utils.process import get_output_error_code
from IPython.testing.tools import get_ipython_cmd
# a trailing space allows for simpler concatenation with the other arguments
ipy_cmd = get_ipython_cmd(as_string=True) + " "
class TestsBase(unittest.TestCase):
"""Base tests class. Contains useful fuzzy comparison and nbconvert
functions."""
def fuzzy_compare(self, a, b, newlines_are_spaces=True, tabs_are_spaces=True,
fuzzy_spacing=True, ignore_spaces=False,
ignore_newlines=False, case_sensitive=False, leave_padding=False):
"""
Performs a fuzzy comparison of two strings. A fuzzy comparison is a
comparison that ignores insignificant differences in the two comparands.
The significance of certain differences can be specified via the keyword
parameters of this method.
#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------
import os
import glob
import shutil
import unittest
import IPython
from IPython.utils.tempdir import TemporaryWorkingDirectory
from IPython.utils.process import get_output_error_code
from IPython.testing.tools import get_ipython_cmd
# a trailing space allows for simpler concatenation with the other arguments
ipy_cmd = get_ipython_cmd(as_string=True) + " "
#-----------------------------------------------------------------------------
# Classes and functions
#-----------------------------------------------------------------------------
class TestsBase(unittest.TestCase):
"""Base tests class. Contains useful fuzzy comparison and nbconvert
functions."""
def fuzzy_compare(self,
a,
b,
newlines_are_spaces=True,
tabs_are_spaces=True,
fuzzy_spacing=True,