def test_printing(self):
self.uut = ConsolePrinter(print_colored=True)
with retrieve_stdout() as stdout:
self.uut.print("\ntest", "message", color="green")
self.assertRegex(stdout.getvalue(), "\033.*\ntest message.*")
with retrieve_stdout() as stdout:
self.uut.print("\ntest", "message", color="greeeeen")
self.assertEqual(stdout.getvalue(), "\ntest message\n")
with retrieve_stdout() as stdout:
self.uut.print("\ntest", "message")
self.assertEqual(stdout.getvalue(), "\ntest message\n")
def test_retrieve_stdout(self): # Tests replace_stdout implicitly
with retrieve_stdout() as sio:
print("test")
self.assertEqual(sio.getvalue(), "test\n")