Merge pull request #2 from bswck/lazy-import-pyrepl-highlight

Highlight lazy imports in the new REPL

Author: pablogsal

Lib/_pyrepl/utils.py

@@ -270,6 +270,8 @@ def is_soft_keyword_used(*tokens: TI | None) -> bool:
             TI(T.NAME, string=s)
         ):
             return not keyword.iskeyword(s)
+        case (None | TI(T.NEWLINE) | TI(T.INDENT) | TI(T.DEDENT), TI(string="lazy"), TI(string="import")):
+            return True
         case _:
             return False
 

Lib/test/test_pyrepl/test_utils.py

@@ -68,10 +68,13 @@ def test_gen_colors_keyword_highlighting(self):
             ("obj.list", [(".", "op")]),
             ("obj.match", [(".", "op")]),
             ("b. \\\n format", [(".", "op")]),
+            ("lazy", []),
+            ("lazy()", [('(', 'op'), (')', 'op')]),
             # highlights
             ("set", [("set", "builtin")]),
             ("list", [("list", "builtin")]),
             ("    \n dict", [("dict", "builtin")]),
+            ("    lazy import", [("lazy", "soft_keyword"), ("import", "keyword")]),
         ]
         for code, expected_highlights in cases:
             with self.subTest(code=code):