bpo-38870: Don't start generated output with newlines in ast.unparse (GH-19636)

This commit is contained in:
Batuhan Taskaya 2020-05-03 20:11:51 +03:00 committed by GitHub
parent 3dd2157feb
commit 493bf1cc31
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 11 deletions

View file

@ -128,19 +128,17 @@ class ASTTestCase(unittest.TestCase):
def check_invalid(self, node, raises=ValueError):
self.assertRaises(raises, ast.unparse, node)
def get_source(self, code1, code2=None, strip=True):
def get_source(self, code1, code2=None):
code2 = code2 or code1
code1 = ast.unparse(ast.parse(code1))
if strip:
code1 = code1.strip()
return code1, code2
def check_src_roundtrip(self, code1, code2=None, strip=True):
code1, code2 = self.get_source(code1, code2, strip)
def check_src_roundtrip(self, code1, code2=None):
code1, code2 = self.get_source(code1, code2)
self.assertEqual(code2, code1)
def check_src_dont_roundtrip(self, code1, code2=None, strip=True):
code1, code2 = self.get_source(code1, code2, strip)
def check_src_dont_roundtrip(self, code1, code2=None):
code1, code2 = self.get_source(code1, code2)
self.assertNotEqual(code2, code1)
class UnparseTestCase(ASTTestCase):