Skip to content

Commit

Permalink
pythongh-117389: Fix test_compileall.EncodingTest (pythonGH-117390)
Browse files Browse the repository at this point in the history
(cherry picked from commit 44f6791)

Co-authored-by: Nikita Sobolev <[email protected]>
  • Loading branch information
sobolevn authored and miss-islington committed May 5, 2024
1 parent ba43157 commit 9ae7c43
Showing 1 changed file with 14 additions and 8 deletions.
22 changes: 14 additions & 8 deletions Lib/test/test_compileall.py
Original file line number Diff line number Diff line change
Expand Up @@ -477,19 +477,25 @@ def setUp(self):
self.directory = tempfile.mkdtemp()
self.source_path = os.path.join(self.directory, '_test.py')
with open(self.source_path, 'w', encoding='utf-8') as file:
file.write('# -*- coding: utf-8 -*-\n')
file.write('print u"\u20ac"\n')
# Intentional syntax error: bytes can only contain
# ASCII literal characters.
file.write('b"\u20ac"')

def tearDown(self):
shutil.rmtree(self.directory)

def test_error(self):
try:
orig_stdout = sys.stdout
sys.stdout = io.TextIOWrapper(io.BytesIO(),encoding='ascii')
compileall.compile_dir(self.directory)
finally:
sys.stdout = orig_stdout
buffer = io.TextIOWrapper(io.BytesIO(), encoding='ascii')
with contextlib.redirect_stdout(buffer):
compiled = compileall.compile_dir(self.directory)
self.assertFalse(compiled) # should not be successful
buffer.seek(0)
res = buffer.read()
self.assertIn(
'SyntaxError: bytes can only contain ASCII literal characters',
res,
)
self.assertNotIn('UnicodeEncodeError', res)


class CommandLineTestsBase:
Expand Down

0 comments on commit 9ae7c43

Please sign in to comment.