mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-09-26 18:29:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

gh-111201: fix auto-indent in pyrepl for muliple pound comments (#123196)

Browse source
This commit is contained in:
Arnon Yaari 2024-09-06 10:33:40 +03:00 • committed by GitHub
parent 67957ea77d
commit d683f49a7b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 19 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
Lib/_pyrepl/readline.py
View file

@ -249,7 +249,7 @@ def _should_auto_indent(buffer: list[str], pos: int) -> bool:
while pos > 0: while pos > 0:
pos -= 1 pos -= 1
if last_char is None: if last_char is None:
if buffer[pos] not in " \t\n": # ignore whitespaces if buffer[pos] not in " \t\n#": # ignore whitespaces and comments
last_char = buffer[pos] last_char = buffer[pos]
else: else:
# even if we found a non-whitespace character before # even if we found a non-whitespace character before

18
Lib/test/test_pyrepl/test_pyrepl.py
View file

@ -466,6 +466,24 @@ class TestPyReplAutoindent(TestCase):
output = multiline_input(reader) output = multiline_input(reader)
self.assertEqual(output, output_code) self.assertEqual(output, output_code)
def test_auto_indent_with_multicomment(self):
# fmt: off
events = code_to_events(
"def f(): ## foo\n"
"pass\n\n"
)
output_code = (
"def f(): ## foo\n"
" pass\n"
" "
)
# fmt: on
reader = self.prepare_reader(events)
output = multiline_input(reader)
self.assertEqual(output, output_code)
def test_auto_indent_ignore_comments(self): def test_auto_indent_ignore_comments(self):
# fmt: off # fmt: off
events = code_to_events( events = code_to_events(