def do_execute(self, code, silent, store_history=True, user_expressions=None, allow_stdin=False): def f(self): if self._fatal_error is not None: return self._repeat_fatal_error() return self._do_execute(code, silent, store_history, user_expressions, allow_stdin) return wrap_unexpected_exceptions(f, self._complete_cell)(self)
def test_wrap_unexpected_exceptions(): mock_method = MagicMock() mock_method.__name__ = 'tos' decorated = wrap_unexpected_exceptions(mock_method) assert_equals(decorated.__name__, mock_method.__name__) result = decorated(self, 0.0) assert_equals(result, mock_method.return_value) assert_equals(ipython_display.send_error.call_count, 0) mock_method.assert_called_once_with(self, 0.0)
def test_wrap_unexpected_exceptions_handle(): mock_method = MagicMock(side_effect=ValueError('~~~~~~')) mock_method.__name__ = 'tos' decorated = wrap_unexpected_exceptions(mock_method) assert_equals(decorated.__name__, mock_method.__name__) result = decorated(self, 'FOOBAR', FOOBAR='FOOBAR') assert_is(result, None) assert_equals(ipython_display.send_error.call_count, 1) mock_method.assert_called_once_with(self, 'FOOBAR', FOOBAR='FOOBAR')
def do_execute(self, code, silent, store_history=True, user_expressions=None, allow_stdin=False): def f(self): if self._fatal_error is not None: return self._repeat_fatal_error() return self._do_execute(code, silent, store_history, user_expressions, allow_stdin) return wrap_unexpected_exceptions(f, self._complete_cell)(self)