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