##// END OF EJS Templates
Merge pull request #5652 from jhamrick/placeholder...
Merge pull request #5652 from jhamrick/placeholder Add placeholder attribute to text widgets

File last commit:

r15886:7bba2551
r16352:04f4cf16 merge
Show More
test_interactiveshell.py
734 lines | 25.1 KiB | text/x-python | PythonLexer
/ IPython / core / tests / test_interactiveshell.py
Jörgen Stenarson
Adding test for safe_execfile call with non-ascii path
r5094 # -*- coding: utf-8 -*-
Fernando Perez
Fix bug with execution of naked multiline strings....
r3300 """Tests for the key interactiveshell module.
Historically the main classes in interactiveshell have been under-tested. This
module should grow as many single-method tests as possible to trap many of the
recurring bugs we seem to encounter with high-level interaction.
Authors
-------
* Fernando Perez
"""
#-----------------------------------------------------------------------------
# Copyright (C) 2011 The IPython Development Team
#
# Distributed under the terms of the BSD License. The full license is in
# the file COPYING, distributed as part of this software.
#-----------------------------------------------------------------------------
#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------
# stdlib
Thomas Kluyver
Add framework for AST transformations of input code.
r8220 import ast
Jörgen Stenarson
Adding test for safe_execfile call with non-ascii path
r5094 import os
Thomas Kluyver
Test process exit codes with terminating signal
r12774 import signal
Jörgen Stenarson
Adding test for safe_execfile call with non-ascii path
r5094 import shutil
Fernando Perez
Add support for finding cell magics with ?/??....
r6997 import sys
Jörgen Stenarson
Adding test for safe_execfile call with non-ascii path
r5094 import tempfile
Fernando Perez
Fix bug with execution of naked multiline strings....
r3300 import unittest
Thomas Kluyver
Start of new callback system
r15597 try:
from unittest import mock
except ImportError:
import mock
Jörgen Stenarson
Adding test for safe_execfile call with non-ascii path
r5094 from os.path import join
MinRK
short error message on AliasError in run_cell...
r3822
Fernando Perez
Add support for finding cell magics with ?/??....
r6997 # third-party
import nose.tools as nt
# Our own
Volker Braun
Also catch SyntaxErrors from InputTransformers in run_cell()...
r13525 from IPython.core.inputtransformer import InputTransformer
Thomas Kluyver
Add test for exit status from system_piped
r12767 from IPython.testing.decorators import skipif, skip_win32, onlyif_unicode_paths
Thomas Kluyver
Add test that numpy, IPython.parallel and IPython.zmq aren't imported on startup.
r8085 from IPython.testing import tools as tt
MinRK
short error message on AliasError in run_cell...
r3822 from IPython.utils import io
Thomas Kluyver
Python 3 compatibility for os.getcwdu()
r13447 from IPython.utils import py3compat
Thomas Kluyver
Use StringIO.StringIO on Python 2....
r13366 from IPython.utils.py3compat import unicode_type, PY3
if PY3:
from io import StringIO
else:
from StringIO import StringIO
Fernando Perez
Fix bug with execution of naked multiline strings....
r3300
#-----------------------------------------------------------------------------
Fernando Perez
Add support for finding cell magics with ?/??....
r6997 # Globals
#-----------------------------------------------------------------------------
# This is used by every single test, no point repeating it ad nauseam
ip = get_ipython()
#-----------------------------------------------------------------------------
Fernando Perez
Fix bug with execution of naked multiline strings....
r3300 # Tests
#-----------------------------------------------------------------------------
class InteractiveShellTestCase(unittest.TestCase):
def test_naked_string_cells(self):
"""Test that cells with only naked strings are fully executed"""
# First, single-line inputs
ip.run_cell('"a"\n')
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ip.user_ns['_'], 'a')
Fernando Perez
Fix bug with execution of naked multiline strings....
r3300 # And also multi-line cells
ip.run_cell('"""a\nb"""\n')
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ip.user_ns['_'], 'a\nb')
Paul Ivanov
added test for GH-306
r3499
Thomas Kluyver
Fix blank input crashing IPython, +unittest
r3441 def test_run_empty_cell(self):
"""Just make sure we don't get a horrible error with a blank
cell of input. Yes, I did overlook that."""
Thomas Kluyver
Don't increment execution_count on empty cells. +test.
r3706 old_xc = ip.execution_count
Thomas Kluyver
Fix blank input crashing IPython, +unittest
r3441 ip.run_cell('')
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ip.execution_count, old_xc)
Fernando Perez
BUG: multi-line, multi-block cells were broken. Test added....
r3467
Paul Ivanov
added test for GH-306
r3499 def test_run_cell_multiline(self):
Fernando Perez
BUG: multi-line, multi-block cells were broken. Test added....
r3467 """Multi-block, multi-line cells must execute correctly.
"""
src = '\n'.join(["x=1",
"y=2",
"if 1:",
" x += 1",
" y += 1",])
ip.run_cell(src)
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ip.user_ns['x'], 2)
self.assertEqual(ip.user_ns['y'], 3)
Paul Ivanov
added test for GH-306
r3499
def test_multiline_string_cells(self):
Paul Ivanov
added test for GH-307
r3500 "Code sprinkled with multiline strings should execute (GH-306)"
Paul Ivanov
added test for GH-306
r3499 ip.run_cell('tmp=0')
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ip.user_ns['tmp'], 0)
Paul Ivanov
added test for GH-306
r3499 ip.run_cell('tmp=1;"""a\nb"""\n')
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ip.user_ns['tmp'], 1)
Paul Ivanov
added test for GH-307
r3500
def test_dont_cache_with_semicolon(self):
"Ending a line with semicolon should not cache the returned object (GH-307)"
oldlen = len(ip.user_ns['Out'])
Thomas Kluyver
Revert PR #5388...
r15886 for cell in ['1;', '1;1;']:
Shashi Gowda
Suppress output even when a comment follows ;. Fixes #4525.
r15883 ip.run_cell(cell, store_history=True)
newlen = len(ip.user_ns['Out'])
self.assertEqual(oldlen, newlen)
i = 0
Paul Ivanov
added test for GH-307
r3500 #also test the default caching behavior
Thomas Kluyver
Revert PR #5388...
r15886 for cell in ['1', '1;1']:
Shashi Gowda
Suppress output even when a comment follows ;. Fixes #4525.
r15883 ip.run_cell(cell, store_history=True)
newlen = len(ip.user_ns['Out'])
i += 1
self.assertEqual(oldlen+i, newlen)
Paul Ivanov
added the skip_known decorator
r3503
Paul Ivanov
added test for GH-284: ensure In variable is works
r3501 def test_In_variable(self):
"Verify that In variable grows with user input (GH-284)"
oldlen = len(ip.user_ns['In'])
Thomas Kluyver
Change run_cell to not store history by default.
r4995 ip.run_cell('1;', store_history=True)
Paul Ivanov
added test for GH-284: ensure In variable is works
r3501 newlen = len(ip.user_ns['In'])
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(oldlen+1, newlen)
self.assertEqual(ip.user_ns['In'][-1],'1;')
Thomas Kluyver
Test for magic names in multiline strings.
r3707
def test_magic_names_in_string(self):
Thomas Kluyver
Fix bug with magic names in multi-line strings. run_cell now uses inputsplitter for static transformations.
r3709 ip.run_cell('a = """\n%exit\n"""')
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ip.user_ns['a'], '\n%exit\n')
MinRK
prevent errors in prefilter from crashing IPython...
r3821
MinRK
fix SyntaxError on !(command)...
r3908 def test_trailing_newline(self):
"""test that running !(command) does not raise a SyntaxError"""
ip.run_cell('!(true)\n', False)
ip.run_cell('!(true)\n\n\n', False)
Julian Taylor
Add test for non-ascii characters in PlainTextFormatter
r4281
def test_gh_597(self):
Thomas Kluyver
Brief docstring to explain test.
r4375 """Pretty-printing lists of objects with non-ascii reprs may cause
problems."""
Julian Taylor
Add test for non-ascii characters in PlainTextFormatter
r4281 class Spam(object):
def __repr__(self):
return "\xe9"*50
import IPython.core.formatters
f = IPython.core.formatters.PlainTextFormatter()
f([Spam(),Spam()])
Thomas Kluyver
Parse user code to AST using compiler flags....
r4795
Thomas Kluyver
Use user_ns as global namespace if it is passed without user_module, and add test for pickling interactively defined objects....
r5456
Thomas Kluyver
Parse user code to AST using compiler flags....
r4795 def test_future_flags(self):
"""Check that future flags are used for parsing code (gh-777)"""
ip.run_cell('from __future__ import print_function')
try:
ip.run_cell('prfunc_return_val = print(1,2, sep=" ")')
assert 'prfunc_return_val' in ip.user_ns
finally:
# Reset compiler flags so we don't mess up other tests.
ip.compile.reset_compiler_flags()
Olivier Verdier
TST: add future unicode_literals test (#786)
r4811
def test_future_unicode(self):
"""Check that unicode_literals is imported from __future__ (gh #786)"""
try:
Olivier Verdier
TST: simpler check for unicode literals (#790)...
r4820 ip.run_cell(u'byte_str = "a"')
Olivier Verdier
TST: remove assert messages (#790)...
r4821 assert isinstance(ip.user_ns['byte_str'], str) # string literals are byte strings by default
Olivier Verdier
TST: add future unicode_literals test (#786)
r4811 ip.run_cell('from __future__ import unicode_literals')
Olivier Verdier
TST: simpler check for unicode literals (#790)...
r4820 ip.run_cell(u'unicode_str = "a"')
Thomas Kluyver
Replace references to unicode and basestring
r13353 assert isinstance(ip.user_ns['unicode_str'], unicode_type) # strings literals are now unicode
Olivier Verdier
TST: add future unicode_literals test (#786)
r4811 finally:
# Reset compiler flags so we don't mess up other tests.
ip.compile.reset_compiler_flags()
Thomas Kluyver
Use user_ns as global namespace if it is passed without user_module, and add test for pickling interactively defined objects....
r5456
def test_can_pickle(self):
"Can we pickle objects defined interactively (GH-29)"
ip = get_ipython()
ip.reset()
ip.run_cell(("class Mylist(list):\n"
" def __init__(self,x=[]):\n"
" list.__init__(self,x)"))
ip.run_cell("w=Mylist([1,2,3])")
Thomas Kluyver
Update imports for Python 3...
r13354 from pickle import dumps
Thomas Kluyver
Use user_ns as global namespace if it is passed without user_module, and add test for pickling interactively defined objects....
r5456
# We need to swap in our main module - this is only necessary
# inside the test framework, because IPython puts the interactive module
# in place (but the test framework undoes this).
_main = sys.modules['__main__']
sys.modules['__main__'] = ip.user_module
try:
res = dumps(ip.user_ns["w"])
finally:
sys.modules['__main__'] = _main
self.assertTrue(isinstance(res, bytes))
def test_global_ns(self):
"Code in functions must be able to access variables outside them."
ip = get_ipython()
ip.run_cell("a = 10")
Thomas Kluyver
Tiny correction to test for interactiveshell scope.
r5465 ip.run_cell(("def f(x):\n"
Thomas Kluyver
Use user_ns as global namespace if it is passed without user_module, and add test for pickling interactively defined objects....
r5456 " return x + a"))
ip.run_cell("b = f(12)")
self.assertEqual(ip.user_ns["b"], 22)
MinRK
protect IPython from bad custom exception handlers...
r4991
def test_bad_custom_tb(self):
"""Check that InteractiveShell is protected from bad custom exception handlers"""
from IPython.utils import io
save_stderr = io.stderr
try:
# capture stderr
io.stderr = StringIO()
MinRK
protect against bad return type of CustomTB...
r4999 ip.set_custom_exc((IOError,), lambda etype,value,tb: 1/0)
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ip.custom_exceptions, (IOError,))
MinRK
protect IPython from bad custom exception handlers...
r4991 ip.run_cell(u'raise IOError("foo")')
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ip.custom_exceptions, ())
MinRK
protect IPython from bad custom exception handlers...
r4991 self.assertTrue("Custom TB Handler failed" in io.stderr.getvalue())
finally:
io.stderr = save_stderr
MinRK
protect against bad return type of CustomTB...
r4999 def test_bad_custom_tb_return(self):
"""Check that InteractiveShell is protected from bad return types in custom exception handlers"""
from IPython.utils import io
save_stderr = io.stderr
try:
# capture stderr
io.stderr = StringIO()
ip.set_custom_exc((NameError,),lambda etype,value,tb, tb_offset=None: 1)
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ip.custom_exceptions, (NameError,))
MinRK
protect against bad return type of CustomTB...
r4999 ip.run_cell(u'a=abracadabra')
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ip.custom_exceptions, ())
MinRK
protect against bad return type of CustomTB...
r4999 self.assertTrue("Custom TB Handler failed" in io.stderr.getvalue())
finally:
io.stderr = save_stderr
Thomas Kluyver
Add drop_by_id method to shell, to remove variables added by extensions.
r5068 def test_drop_by_id(self):
myvars = {"a":object(), "b":object(), "c": object()}
ip.push(myvars, interactive=False)
for name in myvars:
assert name in ip.user_ns, name
assert name in ip.user_ns_hidden, name
ip.user_ns['b'] = 12
ip.drop_by_id(myvars)
for name in ["a", "c"]:
assert name not in ip.user_ns, name
assert name not in ip.user_ns_hidden, name
assert ip.user_ns['b'] == 12
ip.reset()
Jörgen Stenarson
Adding test for safe_execfile call with non-ascii path
r5094
Fernando Perez
Add failing test for #822, will be fixed next
r5357 def test_var_expand(self):
Thomas Kluyver
Sort out unicode test for shell expansion.
r5366 ip.user_ns['f'] = u'Ca\xf1o'
self.assertEqual(ip.var_expand(u'echo $f'), u'echo Ca\xf1o')
MinRK
ignore errors in shell.var_expand...
r6124 self.assertEqual(ip.var_expand(u'echo {f}'), u'echo Ca\xf1o')
self.assertEqual(ip.var_expand(u'echo {f[:-1]}'), u'echo Ca\xf1')
self.assertEqual(ip.var_expand(u'echo {1*2}'), u'echo 2')
Thomas Kluyver
Sort out unicode test for shell expansion.
r5366
ip.user_ns['f'] = b'Ca\xc3\xb1o'
Fernando Perez
Add failing test for #822, will be fixed next
r5357 # This should not raise any exception:
ip.var_expand(u'echo $f')
MinRK
ignore errors in shell.var_expand...
r6124
Thomas Kluyver
Add failing test for issue gh-1878
r7331 def test_var_expand_local(self):
Thomas Kluyver
Test for local variable expansion in %magic commands.
r7333 """Test local variable expansion in !system and %magic calls"""
# !system
Thomas Kluyver
Add failing test for issue gh-1878
r7331 ip.run_cell('def test():\n'
' lvar = "ttt"\n'
' ret = !echo {lvar}\n'
' return ret[0]\n')
res = ip.user_ns['test']()
nt.assert_in('ttt', res)
Thomas Kluyver
Test for local variable expansion in %magic commands.
r7333
# %magic
ip.run_cell('def makemacro():\n'
' macroname = "macro_var_expand_locals"\n'
' %macro {macroname} codestr\n')
ip.user_ns['codestr'] = "str(12)"
ip.run_cell('makemacro()')
nt.assert_in('macro_var_expand_locals', ip.user_ns)
Thomas Kluyver
Add failing test for issue gh-1878
r7331
Thomas Kluyver
Add failing test for variable expansion with self (reopened issue #1878).
r8224 def test_var_expand_self(self):
"""Test variable expansion with the name 'self', which was failing.
See https://github.com/ipython/ipython/issues/1878#issuecomment-7698218
"""
ip.run_cell('class cTest:\n'
' classvar="see me"\n'
' def test(self):\n'
' res = !echo Variable: {self.classvar}\n'
Thomas Kluyver
Fix variable expansion on 'self'...
r8225 ' return res[0]\n')
Thomas Kluyver
Add failing test for variable expansion with self (reopened issue #1878).
r8224 nt.assert_in('see me', ip.user_ns['cTest']().test())
MinRK
ignore errors in shell.var_expand...
r6124 def test_bad_var_expand(self):
"""var_expand on invalid formats shouldn't raise"""
# SyntaxError
self.assertEqual(ip.var_expand(u"{'a':5}"), u"{'a':5}")
# NameError
self.assertEqual(ip.var_expand(u"{asdf}"), u"{asdf}")
# ZeroDivisionError
self.assertEqual(ip.var_expand(u"{1/0}"), u"{1/0}")
MinRK
add silent kwarg to run_cell...
r6802
Thomas Kluyver
Start of new callback system
r15597 def test_silent_postexec(self):
Thomas Kluyver
Rename pre/post_execute_explicit events to pre/post_run_cell
r15607 """run_cell(silent=True) doesn't invoke pre/post_run_cell callbacks"""
Thomas Kluyver
Start of new callback system
r15597 pre_explicit = mock.Mock()
pre_always = mock.Mock()
post_explicit = mock.Mock()
post_always = mock.Mock()
MinRK
add silent kwarg to run_cell...
r6802
Thomas Kluyver
Rename pre/post_execute_explicit events to pre/post_run_cell
r15607 ip.events.register('pre_run_cell', pre_explicit)
Thomas Kluyver
Rename callbacks -> events (mostly), fire -> trigger
r15605 ip.events.register('pre_execute', pre_always)
Thomas Kluyver
Rename pre/post_execute_explicit events to pre/post_run_cell
r15607 ip.events.register('post_run_cell', post_explicit)
Thomas Kluyver
Rename callbacks -> events (mostly), fire -> trigger
r15605 ip.events.register('post_execute', post_always)
Thomas Kluyver
Start of new callback system
r15597
try:
ip.run_cell("1", silent=True)
assert pre_always.called
assert not pre_explicit.called
assert post_always.called
assert not post_explicit.called
# double-check that non-silent exec did what we expected
# silent to avoid
ip.run_cell("1")
assert pre_explicit.called
assert post_explicit.called
finally:
# remove post-exec
Thomas Kluyver
Rename callbacks -> events (mostly), fire -> trigger
r15605 ip.events.reset_all()
MinRK
add silent kwarg to run_cell...
r6802
def test_silent_noadvance(self):
"""run_cell(silent=True) doesn't advance execution_count"""
ec = ip.execution_count
# silent should force store_history=False
ip.run_cell("1", store_history=True, silent=True)
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ec, ip.execution_count)
MinRK
add silent kwarg to run_cell...
r6802 # double-check that non-silent exec did what we expected
# silent to avoid
ip.run_cell("1", store_history=True)
Bradley M. Froehle
s/assertEquals/assertEqual/
r7874 self.assertEqual(ec+1, ip.execution_count)
MinRK
add silent kwarg to run_cell...
r6802
def test_silent_nodisplayhook(self):
"""run_cell(silent=True) doesn't trigger displayhook"""
d = dict(called=False)
trap = ip.display_trap
save_hook = trap.hook
def failing_hook(*args, **kwargs):
d['called'] = True
try:
trap.hook = failing_hook
ip.run_cell("1", silent=True)
self.assertFalse(d['called'])
# double-check that non-silent exec did what we expected
# silent to avoid
ip.run_cell("1")
self.assertTrue(d['called'])
finally:
trap.hook = save_hook
Fernando Perez
Add failing test for #822, will be fixed next
r5357
Bradley M. Froehle
Add flush softspace test, i.e., print 1,; print 2
r6643 @skipif(sys.version_info[0] >= 3, "softspace removed in py3")
def test_print_softspace(self):
"""Verify that softspace is handled correctly when executing multiple
statements.
In [1]: print 1; print 2
1
2
In [2]: print 1,; print 2
1 2
"""
Fernando Perez
Add support for finding cell magics with ?/??....
r6997
def test_ofind_line_magic(self):
from IPython.core.magic import register_line_magic
@register_line_magic
def lmagic(line):
"A line magic"
# Get info on line magic
lfind = ip._ofind('lmagic')
info = dict(found=True, isalias=False, ismagic=True,
namespace = 'IPython internal', obj= lmagic.__wrapped__,
parent = None)
nt.assert_equal(lfind, info)
def test_ofind_cell_magic(self):
from IPython.core.magic import register_cell_magic
@register_cell_magic
def cmagic(line, cell):
"A cell magic"
# Get info on cell magic
find = ip._ofind('cmagic')
info = dict(found=True, isalias=False, ismagic=True,
namespace = 'IPython internal', obj= cmagic.__wrapped__,
parent = None)
nt.assert_equal(find, info)
Thomas Kluyver
Add test for custom exception hook.
r7110
def test_custom_exception(self):
called = []
def my_handler(shell, etype, value, tb, tb_offset=None):
called.append(etype)
shell.showtraceback((etype, value, tb), tb_offset=tb_offset)
ip.set_custom_exc((ValueError,), my_handler)
try:
ip.run_cell("raise ValueError('test')")
# Check that this was called, and only once.
self.assertEqual(called, [ValueError])
finally:
# Reset the custom exception hook
ip.set_custom_exc((), None)
Thomas Kluyver
Test __future__ environments
r9139
@skipif(sys.version_info[0] >= 3, "no differences with __future__ in py3")
def test_future_environment(self):
"Can we run code with & without the shell's __future__ imports?"
ip.run_cell("from __future__ import division")
ip.run_cell("a = 1/2", shell_futures=True)
self.assertEqual(ip.user_ns['a'], 0.5)
ip.run_cell("b = 1/2", shell_futures=False)
self.assertEqual(ip.user_ns['b'], 0)
ip.compile.reset_compiler_flags()
# This shouldn't leak to the shell's compiler
ip.run_cell("from __future__ import division \nc=1/2", shell_futures=False)
self.assertEqual(ip.user_ns['c'], 0.5)
ip.run_cell("d = 1/2", shell_futures=True)
self.assertEqual(ip.user_ns['d'], 0)
Fernando Perez
Add support for finding cell magics with ?/??....
r6997
Fernando Perez
Add failing test for #822, will be fixed next
r5357
Jörgen Stenarson
Adding test for safe_execfile call with non-ascii path
r5094 class TestSafeExecfileNonAsciiPath(unittest.TestCase):
Thomas Kluyver
Skip some more tests that require unicode paths
r12168 @onlyif_unicode_paths
Jörgen Stenarson
Adding test for safe_execfile call with non-ascii path
r5094 def setUp(self):
self.BASETESTDIR = tempfile.mkdtemp()
self.TESTDIR = join(self.BASETESTDIR, u"åäö")
os.mkdir(self.TESTDIR)
with open(join(self.TESTDIR, u"åäötestscript.py"), "w") as sfile:
sfile.write("pass\n")
Thomas Kluyver
Python 3 compatibility for os.getcwdu()
r13447 self.oldpath = py3compat.getcwd()
Jörgen Stenarson
Adding test for safe_execfile call with non-ascii path
r5094 os.chdir(self.TESTDIR)
self.fname = u"åäötestscript.py"
def tearDown(self):
os.chdir(self.oldpath)
shutil.rmtree(self.BASETESTDIR)
Thomas Kluyver
Skip some more tests that require unicode paths
r12168 @onlyif_unicode_paths
Jörgen Stenarson
Adding test for safe_execfile call with non-ascii path
r5094 def test_1(self):
"""Test safe_execfile with non-ascii path
"""
Fernando Perez
Add support for finding cell magics with ?/??....
r6997 ip.safe_execfile(self.fname, {}, raise_exceptions=True)
Jörgen Stenarson
Add simple test for non-ascii characters in system_raw call.
r5314
Thomas Kluyver
Test process exit codes with terminating signal
r12774 class ExitCodeChecks(tt.TempFileMixin):
def test_exit_code_ok(self):
self.system('exit 0')
self.assertEqual(ip.user_ns['_exit_code'], 0)
def test_exit_code_error(self):
self.system('exit 1')
self.assertEqual(ip.user_ns['_exit_code'], 1)
@skipif(not hasattr(signal, 'SIGALRM'))
def test_exit_code_signal(self):
self.mktmp("import signal, time\n"
"signal.setitimer(signal.ITIMER_REAL, 0.1)\n"
"time.sleep(1)\n")
self.system("%s %s" % (sys.executable, self.fname))
self.assertEqual(ip.user_ns['_exit_code'], -signal.SIGALRM)
class TestSystemRaw(unittest.TestCase, ExitCodeChecks):
system = ip.system_raw
Jörgen Stenarson
Add simple test for non-ascii characters in system_raw call.
r5314
Thomas Kluyver
Skip some more tests that require unicode paths
r12168 @onlyif_unicode_paths
Jörgen Stenarson
Add simple test for non-ascii characters in system_raw call.
r5314 def test_1(self):
"""Test system_raw with non-ascii cmd
"""
Thomas Kluyver
Test process exit codes with terminating signal
r12774 cmd = u'''python -c "'åäö'" '''
Fernando Perez
Add support for finding cell magics with ?/??....
r6997 ip.system_raw(cmd)
Fernando Perez
Add simple test for __IPYTHON__ in builtins.
r5493
Thomas Kluyver
Test process exit codes with terminating signal
r12774 # TODO: Exit codes are currently ignored on Windows.
class TestSystemPipedExitCode(unittest.TestCase, ExitCodeChecks):
system = ip.system_piped
Thomas Kluyver
Add test for exit status from system_piped
r12767 @skip_win32
Thomas Kluyver
Test process exit codes with terminating signal
r12774 def test_exit_code_ok(self):
ExitCodeChecks.test_exit_code_ok(self)
@skip_win32
def test_exit_code_error(self):
ExitCodeChecks.test_exit_code_error(self)
@skip_win32
def test_exit_code_signal(self):
ExitCodeChecks.test_exit_code_signal(self)
Fernando Perez
Add simple test for __IPYTHON__ in builtins.
r5493
Thomas Kluyver
Add test that numpy, IPython.parallel and IPython.zmq aren't imported on startup.
r8085 class TestModules(unittest.TestCase, tt.TempFileMixin):
def test_extraneous_loads(self):
"""Test we're not loading modules on startup that we shouldn't.
"""
self.mktmp("import sys\n"
"print('numpy' in sys.modules)\n"
"print('IPython.parallel' in sys.modules)\n"
MinRK
mv IPython.zmq to IPython.kernel.zmq
r9372 "print('IPython.kernel.zmq' in sys.modules)\n"
Thomas Kluyver
Add test that numpy, IPython.parallel and IPython.zmq aren't imported on startup.
r8085 )
out = "False\nFalse\nFalse\n"
tt.ipexec_validate(self.fname, out)
Thomas Kluyver
Add framework for AST transformations of input code.
r8220 class Negator(ast.NodeTransformer):
"""Negates all number literals in an AST."""
def visit_Num(self, node):
node.n = -node.n
return node
class TestAstTransform(unittest.TestCase):
def setUp(self):
self.negator = Negator()
ip.ast_transformers.append(self.negator)
def tearDown(self):
ip.ast_transformers.remove(self.negator)
def test_run_cell(self):
with tt.AssertPrints('-34'):
ip.run_cell('print (12 + 22)')
# A named reference to a number shouldn't be transformed.
ip.user_ns['n'] = 55
with tt.AssertNotPrints('-55'):
ip.run_cell('print (n)')
Thomas Kluyver
Add tests for %timeit with AST transformations
r8485
def test_timeit(self):
called = set()
def f(x):
called.add(x)
ip.push({'f':f})
with tt.AssertPrints("best of "):
ip.run_line_magic("timeit", "-n1 f(1)")
self.assertEqual(called, set([-1]))
called.clear()
with tt.AssertPrints("best of "):
ip.run_cell_magic("timeit", "-n1 f(2)", "f(3)")
self.assertEqual(called, set([-2, -3]))
Thomas Kluyver
Make %time magic work with AST transformations
r8486
def test_time(self):
called = []
def f(x):
called.append(x)
ip.push({'f':f})
# Test with an expression
Thomas Kluyver
Fix test for Windows...
r9901 with tt.AssertPrints("Wall time: "):
Thomas Kluyver
Make %time magic work with AST transformations
r8486 ip.run_line_magic("time", "f(5+9)")
self.assertEqual(called, [-14])
called[:] = []
# Test with a statement (different code path)
Thomas Kluyver
Fix test for Windows...
r9901 with tt.AssertPrints("Wall time: "):
Thomas Kluyver
Make %time magic work with AST transformations
r8486 ip.run_line_magic("time", "a = f(-3 + -2)")
self.assertEqual(called, [5])
Thomas Kluyver
Add test that macros are affected by AST transformations
r8488
def test_macro(self):
ip.push({'a':10})
# The AST transformation makes this do a+=-1
ip.define_macro("amacro", "a+=1\nprint(a)")
with tt.AssertPrints("9"):
ip.run_cell("amacro")
with tt.AssertPrints("8"):
ip.run_cell("amacro")
Fernando Perez
Add simple test for __IPYTHON__ in builtins.
r5493
Thomas Kluyver
Fix missing locations in modified AST....
r8479 class IntegerWrapper(ast.NodeTransformer):
"""Wraps all integers in a call to Integer()"""
def visit_Num(self, node):
if isinstance(node.n, int):
return ast.Call(func=ast.Name(id='Integer', ctx=ast.Load()),
args=[node], keywords=[])
Thomas Kluyver
Fix Integer() wrapper transformer, thanks to @asmeurer
r8508 return node
Thomas Kluyver
Fix missing locations in modified AST....
r8479
class TestAstTransform2(unittest.TestCase):
def setUp(self):
self.intwrapper = IntegerWrapper()
ip.ast_transformers.append(self.intwrapper)
self.calls = []
def Integer(*args):
self.calls.append(args)
Thomas Kluyver
Add tests for %timeit with AST transformations
r8485 return args
Thomas Kluyver
Fix missing locations in modified AST....
r8479 ip.push({"Integer": Integer})
def tearDown(self):
ip.ast_transformers.remove(self.intwrapper)
del ip.user_ns['Integer']
def test_run_cell(self):
ip.run_cell("n = 2")
self.assertEqual(self.calls, [(2,)])
Thomas Kluyver
Fix Integer() wrapper transformer, thanks to @asmeurer
r8508
# This shouldn't throw an error
ip.run_cell("o = 2.0")
self.assertEqual(ip.user_ns['o'], 2.0)
Thomas Kluyver
Add tests for %timeit with AST transformations
r8485
def test_timeit(self):
called = set()
def f(x):
called.add(x)
ip.push({'f':f})
with tt.AssertPrints("best of "):
ip.run_line_magic("timeit", "-n1 f(1)")
self.assertEqual(called, set([(1,)]))
called.clear()
with tt.AssertPrints("best of "):
ip.run_cell_magic("timeit", "-n1 f(2)", "f(3)")
self.assertEqual(called, set([(2,), (3,)]))
Thomas Kluyver
Fix missing locations in modified AST....
r8479
Thomas Kluyver
Add test that a failing ast_transformer is unregistered.
r8221 class ErrorTransformer(ast.NodeTransformer):
"""Throws an error when it sees a number."""
def visit_Num(self):
raise ValueError("test")
class TestAstTransformError(unittest.TestCase):
def test_unregistering(self):
err_transformer = ErrorTransformer()
ip.ast_transformers.append(err_transformer)
with tt.AssertPrints("unregister", channel='stderr'):
ip.run_cell("1 + 2")
# This should have been removed.
nt.assert_not_in(err_transformer, ip.ast_transformers)
Fernando Perez
Add simple test for __IPYTHON__ in builtins.
r5493
def test__IPYTHON__():
# This shouldn't raise a NameError, that's all
__IPYTHON__
MinRK
user_variables and user_expressions use rich repr API...
r10636
class DummyRepr(object):
def __repr__(self):
return "DummyRepr"
def _repr_html_(self):
return "<b>dummy</b>"
def _repr_javascript_(self):
return "console.log('hi');", {'key': 'value'}
def test_user_variables():
# enable all formatters
ip.display_formatter.active_types = ip.display_formatter.format_types
ip.user_ns['dummy'] = d = DummyRepr()
MinRK
use sets when testing user expressions...
r10682 keys = set(['dummy', 'doesnotexist'])
MinRK
user_variables and user_expressions use rich repr API...
r10636 r = ip.user_variables(keys)
MinRK
use sets when testing user expressions...
r10682 nt.assert_equal(keys, set(r.keys()))
MinRK
user_variables and user_expressions use rich repr API...
r10636 dummy = r['dummy']
MinRK
use sets when testing user expressions...
r10682 nt.assert_equal(set(['status', 'data', 'metadata']), set(dummy.keys()))
MinRK
user_variables and user_expressions use rich repr API...
r10636 nt.assert_equal(dummy['status'], 'ok')
data = dummy['data']
metadata = dummy['metadata']
nt.assert_equal(data.get('text/html'), d._repr_html_())
js, jsmd = d._repr_javascript_()
nt.assert_equal(data.get('application/javascript'), js)
nt.assert_equal(metadata.get('application/javascript'), jsmd)
dne = r['doesnotexist']
nt.assert_equal(dne['status'], 'error')
nt.assert_equal(dne['ename'], 'KeyError')
# back to text only
ip.display_formatter.active_types = ['text/plain']
def test_user_expression():
# enable all formatters
ip.display_formatter.active_types = ip.display_formatter.format_types
query = {
'a' : '1 + 2',
'b' : '1/0',
}
r = ip.user_expressions(query)
import pprint
pprint.pprint(r)
Julian Taylor
sort dictionary keys before comparison, ordering is not guaranteed
r14839 nt.assert_equal(set(r.keys()), set(query.keys()))
MinRK
user_variables and user_expressions use rich repr API...
r10636 a = r['a']
MinRK
use sets when testing user expressions...
r10682 nt.assert_equal(set(['status', 'data', 'metadata']), set(a.keys()))
MinRK
user_variables and user_expressions use rich repr API...
r10636 nt.assert_equal(a['status'], 'ok')
data = a['data']
metadata = a['metadata']
nt.assert_equal(data.get('text/plain'), '3')
b = r['b']
nt.assert_equal(b['status'], 'error')
nt.assert_equal(b['ename'], 'ZeroDivisionError')
# back to text only
ip.display_formatter.active_types = ['text/plain']
Volker Braun
Also catch SyntaxErrors from InputTransformers in run_cell()...
r13525 class TestSyntaxErrorTransformer(unittest.TestCase):
"""Check that SyntaxError raised by an input transformer is handled by run_cell()"""
class SyntaxErrorTransformer(InputTransformer):
def push(self, line):
pos = line.find('syntaxerror')
if pos >= 0:
e = SyntaxError('input contains "syntaxerror"')
e.text = line
e.offset = pos + 1
raise e
return line
def reset(self):
pass
def setUp(self):
self.transformer = TestSyntaxErrorTransformer.SyntaxErrorTransformer()
ip.input_splitter.python_line_transforms.append(self.transformer)
ip.input_transformer_manager.python_line_transforms.append(self.transformer)
def tearDown(self):
ip.input_splitter.python_line_transforms.remove(self.transformer)
ip.input_transformer_manager.python_line_transforms.remove(self.transformer)
def test_syntaxerror_input_transformer(self):
with tt.AssertPrints('1234'):
ip.run_cell('1234')
with tt.AssertPrints('SyntaxError: invalid syntax'):
ip.run_cell('1 2 3') # plain python syntax error
with tt.AssertPrints('SyntaxError: input contains "syntaxerror"'):
ip.run_cell('2345 # syntaxerror') # input transformer syntax error
with tt.AssertPrints('3456'):
ip.run_cell('3456')
MinRK
user_variables and user_expressions use rich repr API...
r10636