add tests

Author: Carreau

ipykernel/inprocess/tests/test_kernel.py

@@ -99,6 +99,27 @@ def test_stdout(self):
         out, err = assemble_output(kc.iopub_channel)
         assert out == 'bar\n'
 
+    @pytest.mark.skipif(
+        sys.platform == 'win32',
+        reason="not ment to work on windows"
+    )
+    def test_capfd(self):
+        """ Does correctly capture fd
+        """
+        kernel = InProcessKernel()
+
+        with capture_output() as io:
+            kernel.shell.run_cell('print("foo")')
+        assert io.stdout == 'foo\n'
+
+        kc = BlockingInProcessKernelClient(kernel=kernel, session=kernel.session)
+        kernel.frontends.append(kc)
+        kc.execute('import os')
+        kc.execute('os.system("echo capfd")')
+        out, err = assemble_output(kc.iopub_channel)
+        assert out == 'capfd\n'
+
+
     def test_getpass_stream(self):
         "Tests that kernel getpass accept the stream parameter"
         kernel = InProcessKernel()

ipykernel/tests/test_kernel.py

@@ -28,7 +28,7 @@
 def _check_master(kc, expected=True, stream="stdout"):
     execute(kc=kc, code="import sys")
     flush_channels(kc)
-    msg_id, content = execute(kc=kc, code="print (sys.%s._is_master_process())" % stream)
+    msg_id, content = execute(kc=kc, code="print(sys.%s._is_master_process())" % stream)
     stdout, stderr = assemble_output(kc.iopub_channel)
     assert stdout.strip() == repr(expected)
 
@@ -51,6 +51,18 @@ def test_simple_print():
         assert stderr == ''
         _check_master(kc, expected=True)
 
+@pytest.mark.skipif(sys.platform=='win32', reason='Not meant to work on windows')
+def test_capture_fd():
+    """simple print statement in kernel"""
+    with kernel() as kc:
+        iopub = kc.iopub_channel
+        msg_id, content = execute(kc=kc, code="import os; os.system('echo capsys')")
+        stdout, stderr = assemble_output(iopub)
+        assert stdout == 'capsys\n'
+        assert stderr == ''
+        _check_master(kc, expected=True)
+
+
 
 def test_sys_path():
     """test that sys.path doesn't get messed up by default"""