test_interactiveshell.py
148 lines
| 5.5 KiB
| text/x-python
|
PythonLexer
Fernando Perez
|
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 | ||||
import unittest | ||||
MinRK
|
r4794 | from StringIO import StringIO | ||
MinRK
|
r3822 | |||
Paul Ivanov
|
r3503 | from IPython.testing import decorators as dec | ||
MinRK
|
r3822 | from IPython.utils import io | ||
Fernando Perez
|
r3300 | |||
#----------------------------------------------------------------------------- | ||||
# Tests | ||||
#----------------------------------------------------------------------------- | ||||
class InteractiveShellTestCase(unittest.TestCase): | ||||
def test_naked_string_cells(self): | ||||
"""Test that cells with only naked strings are fully executed""" | ||||
ip = get_ipython() | ||||
# First, single-line inputs | ||||
ip.run_cell('"a"\n') | ||||
self.assertEquals(ip.user_ns['_'], 'a') | ||||
# And also multi-line cells | ||||
ip.run_cell('"""a\nb"""\n') | ||||
self.assertEquals(ip.user_ns['_'], 'a\nb') | ||||
Paul Ivanov
|
r3499 | |||
Thomas Kluyver
|
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.""" | ||||
ip = get_ipython() | ||||
Thomas Kluyver
|
r3706 | old_xc = ip.execution_count | ||
Thomas Kluyver
|
r3441 | ip.run_cell('') | ||
Thomas Kluyver
|
r3706 | self.assertEquals(ip.execution_count, old_xc) | ||
Fernando Perez
|
r3467 | |||
Paul Ivanov
|
r3499 | def test_run_cell_multiline(self): | ||
Fernando Perez
|
r3467 | """Multi-block, multi-line cells must execute correctly. | ||
""" | ||||
ip = get_ipython() | ||||
src = '\n'.join(["x=1", | ||||
"y=2", | ||||
"if 1:", | ||||
" x += 1", | ||||
" y += 1",]) | ||||
ip.run_cell(src) | ||||
self.assertEquals(ip.user_ns['x'], 2) | ||||
self.assertEquals(ip.user_ns['y'], 3) | ||||
Paul Ivanov
|
r3499 | |||
def test_multiline_string_cells(self): | ||||
Paul Ivanov
|
r3500 | "Code sprinkled with multiline strings should execute (GH-306)" | ||
Paul Ivanov
|
r3499 | ip = get_ipython() | ||
ip.run_cell('tmp=0') | ||||
self.assertEquals(ip.user_ns['tmp'], 0) | ||||
ip.run_cell('tmp=1;"""a\nb"""\n') | ||||
self.assertEquals(ip.user_ns['tmp'], 1) | ||||
Paul Ivanov
|
r3500 | |||
def test_dont_cache_with_semicolon(self): | ||||
"Ending a line with semicolon should not cache the returned object (GH-307)" | ||||
ip = get_ipython() | ||||
oldlen = len(ip.user_ns['Out']) | ||||
a = ip.run_cell('1;') | ||||
newlen = len(ip.user_ns['Out']) | ||||
self.assertEquals(oldlen, newlen) | ||||
#also test the default caching behavior | ||||
Paul Ivanov
|
r3501 | ip.run_cell('1') | ||
Paul Ivanov
|
r3500 | newlen = len(ip.user_ns['Out']) | ||
self.assertEquals(oldlen+1, newlen) | ||||
Paul Ivanov
|
r3503 | |||
Paul Ivanov
|
r3501 | def test_In_variable(self): | ||
"Verify that In variable grows with user input (GH-284)" | ||||
ip = get_ipython() | ||||
oldlen = len(ip.user_ns['In']) | ||||
ip.run_cell('1;') | ||||
newlen = len(ip.user_ns['In']) | ||||
self.assertEquals(oldlen+1, newlen) | ||||
self.assertEquals(ip.user_ns['In'][-1],'1;') | ||||
Thomas Kluyver
|
r3707 | |||
def test_magic_names_in_string(self): | ||||
ip = get_ipython() | ||||
Thomas Kluyver
|
r3709 | ip.run_cell('a = """\n%exit\n"""') | ||
self.assertEquals(ip.user_ns['a'], '\n%exit\n') | ||||
MinRK
|
r3821 | |||
def test_alias_crash(self): | ||||
"""Errors in prefilter can't crash IPython""" | ||||
ip = get_ipython() | ||||
ip.run_cell('%alias parts echo first %s second %s') | ||||
MinRK
|
r3822 | # capture stderr: | ||
save_err = io.stderr | ||||
io.stderr = StringIO() | ||||
MinRK
|
r3821 | ip.run_cell('parts 1') | ||
MinRK
|
r3822 | err = io.stderr.getvalue() | ||
io.stderr = save_err | ||||
self.assertEquals(err.split(':')[0], 'ERROR') | ||||
MinRK
|
r3908 | |||
def test_trailing_newline(self): | ||||
"""test that running !(command) does not raise a SyntaxError""" | ||||
ip = get_ipython() | ||||
ip.run_cell('!(true)\n', False) | ||||
ip.run_cell('!(true)\n\n\n', False) | ||||
Julian Taylor
|
r4281 | |||
def test_gh_597(self): | ||||
Thomas Kluyver
|
r4375 | """Pretty-printing lists of objects with non-ascii reprs may cause | ||
problems.""" | ||||
Julian Taylor
|
r4281 | class Spam(object): | ||
def __repr__(self): | ||||
return "\xe9"*50 | ||||
import IPython.core.formatters | ||||
f = IPython.core.formatters.PlainTextFormatter() | ||||
f([Spam(),Spam()]) | ||||
Thomas Kluyver
|
r4795 | |||
def test_future_flags(self): | ||||
"""Check that future flags are used for parsing code (gh-777)""" | ||||
ip = get_ipython() | ||||
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
|
r4811 | |||
def test_future_unicode(self): | ||||
"""Check that unicode_literals is imported from __future__ (gh #786)""" | ||||
ip = get_ipython() | ||||
try: | ||||
Olivier Verdier
|
r4820 | ip.run_cell(u'byte_str = "a"') | ||
Olivier Verdier
|
r4821 | assert isinstance(ip.user_ns['byte_str'], str) # string literals are byte strings by default | ||
Olivier Verdier
|
r4811 | ip.run_cell('from __future__ import unicode_literals') | ||
Olivier Verdier
|
r4820 | ip.run_cell(u'unicode_str = "a"') | ||
Olivier Verdier
|
r4821 | assert isinstance(ip.user_ns['unicode_str'], unicode) # strings literals are now unicode | ||
Olivier Verdier
|
r4811 | finally: | ||
# Reset compiler flags so we don't mess up other tests. | ||||
ip.compile.reset_compiler_flags() | ||||