From 5e844763252820546b62bea0c89f42759ff4eaa4 Mon Sep 17 00:00:00 2001 From: Guido van Rossum Date: Tue, 31 Oct 2023 16:41:49 -0700 Subject: [PATCH] Fix test_generated_cases.py by stripping preprocessor prefix/suffix --- Lib/test/test_generated_cases.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/Lib/test/test_generated_cases.py b/Lib/test/test_generated_cases.py index 475d7492040b7b..ea0c11621a2ca9 100644 --- a/Lib/test/test_generated_cases.py +++ b/Lib/test/test_generated_cases.py @@ -118,17 +118,19 @@ def run_cases_test(self, input: str, expected: str): with open(self.temp_output_filename) as temp_output: lines = temp_output.readlines() - while lines and lines[0].startswith("// "): + while lines and lines[0].startswith(("// ", "#", " #", "\n")): lines.pop(0) + while lines and lines[-1].startswith(("#", "\n")): + lines.pop(-1) actual = "".join(lines) - # if actual.rstrip() != expected.rstrip(): + # if actual.strip() != expected.strip(): # print("Actual:") # print(actual) # print("Expected:") # print(expected) # print("End") - self.assertEqual(actual.rstrip(), expected.rstrip()) + self.assertEqual(actual.strip(), expected.strip()) def test_inst_no_args(self): input = """