diff --git a/test.py b/test.py index 395d0b8..0c25ea9 100755 --- a/test.py +++ b/test.py @@ -118,8 +118,8 @@ class T(unittest.TestCase): except SystemExit: pass output = output.getvalue() - assert len(output) > 0 - assert '\n' in output + self.assertGreater(len(output), 0) + self.assertIn('\n', output) def test_correct_output(self): """