def test_rich_output_metadata():
"""test RichOutput with metadata"""
data = full_data
metadata = full_metadata
rich = capture.RichOutput(data=data, metadata=metadata)
for method, mime in _mime_map.items():
yield nt.assert_equal, getattr(rich, method)(), (data[mime], metadata[mime])
def binder(self, line, cell=''):
'''Simple example.'''
args = parse_argstring(self.binder, line)
if args.repo:
self.repo = args.repo
binderURL = binder_url(self.repo)
self.b = MyBinderCell(binderURL, message='step')
if self.b is None:
print('No connection...')
return
if cell is None:
print('Use block magic')
else:
#We can force cell numbers...
#get_ipython().execution_count = 77
#Use a different colour cell background to show we're using a Binder kernel...
_set_background()
items = self.b.execute_request(cell)
for item in items['iopub']:
if item['msg_type']=='execute_result':
return capture.RichOutput(data=item['content']['data'], metadata={})
elif item['msg_type']=='stream':
IPython.display.display({'text/plain':item['content']['text']}, raw=True)
def test_rich_output_metadata(method_mime):
"""test RichOutput with metadata"""
data = full_data
metadata = full_metadata
rich = capture.RichOutput(data=data, metadata=metadata)
method, mime = method_mime
nt.assert_equal(getattr(rich, method)(), (data[mime], metadata[mime]))
def test_capture_output():
"""capture_output works"""
rich = capture.RichOutput(data=full_data)
with capture.capture_output() as cap:
print(hello_stdout, end="")
print(hello_stderr, end="", file=sys.stderr)
rich.display()
yield nt.assert_equal, hello_stdout, cap.stdout
yield nt.assert_equal, hello_stderr, cap.stderr
def test_capture_output_no_stdout():
"""test capture_output(stdout=False)"""
rich = capture.RichOutput(data=full_data)
with capture.capture_output(stdout=False) as cap:
print(hello_stdout, end="")
print(hello_stderr, end="", file=sys.stderr)
rich.display()
nt.assert_equal("", cap.stdout)
nt.assert_equal(hello_stderr, cap.stderr)
nt.assert_equal(len(cap.outputs), 1)
def test_rich_output():
"""test RichOutput basics"""
data = basic_data
metadata = basic_metadata
rich = capture.RichOutput(data=data, metadata=metadata)
yield nt.assert_equal, rich._repr_html_(), data['text/html']
yield nt.assert_equal, rich._repr_png_(), (data['image/png'], metadata['image/png'])
yield nt.assert_equal, rich._repr_latex_(), None
yield nt.assert_equal, rich._repr_javascript_(), None
yield nt.assert_equal, rich._repr_svg_(), None
def test_rich_output():
"""test RichOutput basics"""
data = basic_data
metadata = basic_metadata
rich = capture.RichOutput(data=data, metadata=metadata)
assert rich._repr_html_() == data["text/html"]
assert rich._repr_png_() == (data["image/png"], metadata["image/png"])
assert rich._repr_latex_() is None
assert rich._repr_javascript_() is None
assert rich._repr_svg_() is None
def test_capture_output_no_display():
"""test capture_output(display=False)"""
rich = capture.RichOutput(data=full_data)
with capture.capture_output(display=False) as cap:
print(hello_stdout, end="")
print(hello_stderr, end="", file=sys.stderr)
rich.display()
assert hello_stdout == cap.stdout
assert hello_stderr == cap.stderr
assert cap.outputs == []
def test_capture_output_no_display():
"""test capture_output(display=False)"""
rich = capture.RichOutput(data=full_data)
with capture.capture_output(display=False) as cap:
print(hello_stdout, end="")
print(hello_stderr, end="", file=sys.stderr)
rich.display()
yield nt.assert_equal, hello_stdout, cap.stdout
yield nt.assert_equal, hello_stderr, cap.stderr
yield nt.assert_equal, cap.outputs, []
def test_capture_output_no_stderr():
"""test capture_output(stderr=False)"""
rich = capture.RichOutput(data=full_data)
# add nested capture_output so stderr doesn't make it to nose output
with capture.capture_output(), capture.capture_output(stderr=False) as cap:
print(hello_stdout, end="")
print(hello_stderr, end="", file=sys.stderr)
rich.display()
yield nt.assert_equal, hello_stdout, cap.stdout
yield nt.assert_equal, "", cap.stderr
yield nt.assert_equal, len(cap.outputs), 1
def test_rich_output():
"""test RichOutput basics"""
data = basic_data
metadata = basic_metadata
rich = capture.RichOutput(data=data, metadata=metadata)
nt.assert_equal(rich._repr_html_(), data["text/html"])
nt.assert_equal(rich._repr_png_(),
(data["image/png"], metadata["image/png"]))
nt.assert_equal(rich._repr_latex_(), None)
nt.assert_equal(rich._repr_javascript_(), None)
nt.assert_equal(rich._repr_svg_(), None)
def test_capture_output_no_stderr():
"""test capture_output(stderr=False)"""
rich = capture.RichOutput(data=full_data)
# add nested capture_output so stderr doesn't make it to nose output
with capture.capture_output(), capture.capture_output(stderr=False) as cap:
print(hello_stdout, end="")
print(hello_stderr, end="", file=sys.stderr)
rich.display()
assert hello_stdout == cap.stdout
assert "" == cap.stderr
assert len(cap.outputs) == 1
def test_rich_output_display():
"""test RichOutput.display
This is a bit circular, because we are actually using the capture code we are testing
to test itself.
"""
data = full_data
rich = capture.RichOutput(data=data)
with capture.capture_output() as cap:
rich.display()
yield nt.assert_equal, len(cap.outputs), 1
rich2 = cap.outputs[0]
yield nt.assert_equal, rich2.data, rich.data
yield nt.assert_equal, rich2.metadata, rich.metadata
def test_rich_output_empty(method_mime):
"""RichOutput with no args"""
rich = capture.RichOutput()
method, mime = method_mime
nt.assert_equal(getattr(rich, method)(), None)
def test_rich_output_no_metadata():
"""test RichOutput with no metadata"""
data = full_data
rich = capture.RichOutput(source="test", data=data)
for method, mime in _mime_map.items():
yield nt.assert_equal, getattr(rich, method)(), data[mime]
def test_rich_output_empty():
"""RichOutput with no args"""
rich = capture.RichOutput()
for method, mime in _mime_map.items():
yield nt.assert_equal, getattr(rich, method)(), None
def test_rich_output_no_metadata(method_mime):
"""test RichOutput with no metadata"""
data = full_data
rich = capture.RichOutput(data=data)
method, mime = method_mime
assert getattr(rich, method)() == data[mime]