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")