def worker(inputs, result_container):
def mock_stdin_provider():
if not inputs:
return None
val = inputs.pop(0)
if val == 'interrupt':
raise KeyboardInterrupt
return val
mock_stdin_widget, echo_updater_calls = create_mock_stdin_widget()
with \
mock.patch.object(
_message,
'_read_stdin_message',
side_effect=mock_stdin_provider,
autospec=True), \
mock.patch.object(
_system_commands,
'_display_stdin_widget',
mock_stdin_widget):
_system_commands._register_magics(self.ip)
with io.capture_output() as captured:
with mock.patch.object(
captured._stdout,
'flush',
wraps=captured._stdout.flush) as stdout_flushes:
self.ip.run_cell(cell_contents)
run_cell_result = RunCellResult(captured, echo_updater_calls,
stdout_flushes.call_count)
result_container['run_cell_result'] = run_cell_result
def worker(inputs, result_container):
def mock_stdin_provider():
if not inputs:
return None
val = inputs.pop(0)
if val == 'interrupt':
raise KeyboardInterrupt
return val
mock_stdin_widget, echo_updater_calls = create_mock_stdin_widget()
with \
mock.patch.object(
_message,
'_read_stdin_message',
side_effect=mock_stdin_provider,
autospec=True), \
mock.patch.object(
_system_commands,
'_display_stdin_widget',
mock_stdin_widget):
_system_commands._register_magics(self.ip)
with io.capture_output() as captured:
self.ip.run_cell(cell_contents)
result_container.update({
'output':
captured,
'echo_updater_calls':
echo_updater_calls,
})
def load_ipython_extension(ipython): """Called by IPython when this module is loaded as an IPython extension.""" _shell_customizations.initialize() _system_commands._register_magics(ipython) # pylint:disable=protected-access _installation_commands._register_magics(ipython) # pylint:disable=protected-access _import_hooks._register_hooks() # pylint:disable=protected-access _tensorflow_magics._register_magics(ipython) # pylint:disable=protected-access