Merge pull request #24 from StanFromIreland/lazy-color

Author: pablogsal

Lib/_pyrepl/utils.py

@@ -276,7 +276,7 @@ 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")):
+        case (None | TI(T.NEWLINE) | TI(T.INDENT) | TI(T.DEDENT), TI(string="lazy"), TI(string="import") | TI(string="from")):
             return True
         case _:
             return False

Lib/test/test_pyrepl/test_utils.py

@@ -96,6 +96,7 @@ def test_gen_colors_keyword_highlighting(self):
             ("list", [("list", "builtin")]),
             ("    \n dict", [("dict", "builtin")]),
             ("    lazy import", [("lazy", "soft_keyword"), ("import", "keyword")]),
+            ("lazy from cool_people import pablo", [('lazy', 'soft_keyword'), ('from', 'keyword'), ('import', 'keyword')])
         ]
         for code, expected_highlights in cases:
             with self.subTest(code=code):