def test_wrapper_arguments(self): """Make sure wrapper with arguments implementation of IOCage works.""" _original_stdout = sys.stdout captured = IOCage() def foo(my_string, my_kwarg="default"): self.assertIsNot(sys.stdout, _original_stdout) print(my_string + my_kwarg) captured.capture(foo, args=("test ", ), kwargs={"my_kwarg": "string"}) self.assertIs(sys.stdout, _original_stdout) self.assertEqual(captured.stdout, "test string\n")
def test_wrapper(self): """Make sure wrapper implementation of IOCage works.""" _original_stdout = sys.stdout captured = IOCage() def foo(): self.assertIsNot(sys.stdout, _original_stdout) print("test string") captured.capture(foo) self.assertIs(sys.stdout, _original_stdout) self.assertEqual(captured.stdout, "test string\n")