diff --git a/test_autoflake.py b/test_autoflake.py index df4e8f9..859c5ee 100755 --- a/test_autoflake.py +++ b/test_autoflake.py @@ -1362,7 +1362,7 @@ def test_check_with_empty_file(self): autoflake._main(argv=['my_fake_program', '--check', filename], standard_out=output_file, standard_error=None) - self.assertEqual('No issues detected!', output_file.getvalue()) + self.assertEqual('No issues detected!\n', output_file.getvalue()) def test_check_correct_file(self): with temporary_file("""\ @@ -1374,7 +1374,7 @@ def test_check_correct_file(self): autoflake._main(argv=['my_fake_program', '--check', filename], standard_out=output_file, standard_error=None) - self.assertEqual('No issues detected!', output_file.getvalue()) + self.assertEqual('No issues detected!\n', output_file.getvalue()) def test_check_useless_pass(self): with temporary_file("""\