test_interactivshell.py
153 lines
| 4.8 KiB
| text/x-python
|
PythonLexer
Julian Taylor
|
r5177 | # -*- coding: utf-8 -*- | ||
Thomas Kluyver
|
r22436 | """Tests for the TerminalInteractiveShell and related pieces.""" | ||
Min RK
|
r23307 | # Copyright (c) IPython Development Team. | ||
| # Distributed under the terms of the Modified BSD License. | ||||
|
Julian Taylor
|
r5177 | |||
|
MinRK
|
r7683 | import sys | ||
|
Julian Taylor
|
r5177 | import unittest | ||
Thomas Kluyver
|
r13528 | from IPython.core.inputtransformer import InputTransformer | ||
mr.Shu
|
r8974 | from IPython.testing import tools as tt | ||
|
Min RK
|
r23307 | from IPython.utils.capture import capture_output | ||
|
Julian Taylor
|
r5177 | |||
|
Thomas Kluyver
|
r13528 | # Decorator for interaction loop tests ----------------------------------------- | ||
| class mock_input_helper(object): | ||||
| """Machinery for tests of the main interact loop. | ||||
| Used by the mock_input decorator. | ||||
| """ | ||||
| def __init__(self, testgen): | ||||
| self.testgen = testgen | ||||
| self.exception = None | ||||
| self.ip = get_ipython() | ||||
| def __enter__(self): | ||||
|
Thomas Kluyver
|
r22436 | self.orig_prompt_for_code = self.ip.prompt_for_code | ||
| self.ip.prompt_for_code = self.fake_input | ||||
|
Thomas Kluyver
|
r13528 | return self | ||
| def __exit__(self, etype, value, tb): | ||||
|
Thomas Kluyver
|
r22436 | self.ip.prompt_for_code = self.orig_prompt_for_code | ||
|
Thomas Kluyver
|
r13528 | |||
|
Thomas Kluyver
|
r22436 | def fake_input(self): | ||
|
Thomas Kluyver
|
r13528 | try: | ||
| return next(self.testgen) | ||||
| except StopIteration: | ||||
|
Thomas Kluyver
|
r22436 | self.ip.keep_running = False | ||
|
Thomas Kluyver
|
r13528 | return u'' | ||
| except: | ||||
| self.exception = sys.exc_info() | ||||
|
Thomas Kluyver
|
r22436 | self.ip.keep_running = False | ||
|
Thomas Kluyver
|
r13528 | return u'' | ||
| def mock_input(testfunc): | ||||
| """Decorator for tests of the main interact loop. | ||||
| Write the test as a generator, yield-ing the input strings, which IPython | ||||
| will see as if they were typed in at the prompt. | ||||
| """ | ||||
| def test_method(self): | ||||
| testgen = testfunc(self) | ||||
| with mock_input_helper(testgen) as mih: | ||||
|
Thomas Kluyver
|
r22436 | mih.ip.interact() | ||
|
Thomas Kluyver
|
r13528 | |||
| if mih.exception is not None: | ||||
| # Re-raise captured exception | ||||
| etype, value, tb = mih.exception | ||||
| import traceback | ||||
| traceback.print_tb(tb, file=sys.stdout) | ||||
| del tb # Avoid reference loop | ||||
| raise value | ||||
| return test_method | ||||
| # Test classes ----------------------------------------------------------------- | ||||
|
Julian Taylor
|
r5177 | class InteractiveShellTestCase(unittest.TestCase): | ||
| def rl_hist_entries(self, rl, n): | ||||
| """Get last n readline history entries as a list""" | ||||
| return [rl.get_history_item(rl.get_current_history_length() - x) | ||||
| for x in range(n - 1, -1, -1)] | ||||
|
mr.Shu
|
r8974 | |||
|
Thomas Kluyver
|
r13528 | @mock_input | ||
| def test_inputtransformer_syntaxerror(self): | ||||
| ip = get_ipython() | ||||
| transformer = SyntaxErrorTransformer() | ||||
| ip.input_splitter.python_line_transforms.append(transformer) | ||||
| ip.input_transformer_manager.python_line_transforms.append(transformer) | ||||
| try: | ||||
| #raise Exception | ||||
| with tt.AssertPrints('4', suppress=False): | ||||
| yield u'print(2*2)' | ||||
| with tt.AssertPrints('SyntaxError: input contains', suppress=False): | ||||
| yield u'print(2345) # syntaxerror' | ||||
| with tt.AssertPrints('16', suppress=False): | ||||
| yield u'print(4*4)' | ||||
| finally: | ||||
| ip.input_splitter.python_line_transforms.remove(transformer) | ||||
| ip.input_transformer_manager.python_line_transforms.remove(transformer) | ||||
|
Min RK
|
r22530 | def test_plain_text_only(self): | ||
| ip = get_ipython() | ||||
| formatter = ip.display_formatter | ||||
| assert formatter.active_types == ['text/plain'] | ||||
|
Min RK
|
r23307 | assert not formatter.ipython_display_formatter.enabled | ||
| class Test(object): | ||||
| def __repr__(self): | ||||
| return "<Test %i>" % id(self) | ||||
| def _repr_html_(self): | ||||
| return '<html>' | ||||
| # verify that HTML repr isn't computed | ||||
| obj = Test() | ||||
| data, _ = formatter.format(obj) | ||||
| self.assertEqual(data, {'text/plain': repr(obj)}) | ||||
| class Test2(Test): | ||||
| def _ipython_display_(self): | ||||
| from IPython.display import display | ||||
| display('<custom>') | ||||
| # verify that _ipython_display_ shortcut isn't called | ||||
| obj = Test2() | ||||
| with capture_output() as captured: | ||||
| data, _ = formatter.format(obj) | ||||
| self.assertEqual(data, {'text/plain': repr(obj)}) | ||||
| assert captured.stdout == '' | ||||
|
Min RK
|
r22530 | |||
|
Thomas Kluyver
|
r13528 | |||
| 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 | ||||
|
Thomas Kluyver
|
r10752 | class TerminalMagicsTestCase(unittest.TestCase): | ||
| def test_paste_magics_blankline(self): | ||||
| """Test that code with a blank line doesn't get split (gh-3246).""" | ||||
| ip = get_ipython() | ||||
| s = ('def pasted_func(a):\n' | ||||
| ' b = a+1\n' | ||||
| '\n' | ||||
| ' return b') | ||||
| tm = ip.magics_manager.registry['TerminalMagics'] | ||||
| tm.store_or_execute(s, name=None) | ||||
| self.assertEqual(ip.user_ns['pasted_func'](54), 55) | ||||