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_alias_args_commented(): """Check that alias correctly ignores 'commented out' args""" _ip.magic('alias commetarg echo this is %%s a commented out arg') with capture_output() as cap: _ip.run_cell('commetarg') nt.assert_equal(cap.stdout, 'this is %s a commented out arg')
def test_alias_args_error(): """Error expanding with wrong number of arguments""" _ip.alias_manager.define_alias('parts', 'echo first %s second %s') # capture stderr: with capture_output() as cap: _ip.run_cell('parts 1') nt.assert_equal(cap.stderr.split(':')[0], 'UsageError')
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_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_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