diff options
| -rw-r--r-- | queries/python/highlights.scm | 5 | ||||
| -rw-r--r-- | tests/query/highlights/python/yield_from.py | 7 |
2 files changed, 11 insertions, 1 deletions
diff --git a/queries/python/highlights.scm b/queries/python/highlights.scm index ccbb2e79d..e7b59caf6 100644 --- a/queries/python/highlights.scm +++ b/queries/python/highlights.scm @@ -226,8 +226,11 @@ "return" "yield" ] @keyword.return +(yield "from" @keyword.return) + +(import_from_statement "from" @include) +"import" @include -["from" "import"] @include (aliased_import "as" @include) ["if" "elif" "else" "match" "case"] @conditional diff --git a/tests/query/highlights/python/yield_from.py b/tests/query/highlights/python/yield_from.py new file mode 100644 index 000000000..2f3238379 --- /dev/null +++ b/tests/query/highlights/python/yield_from.py @@ -0,0 +1,7 @@ +from foo import bar +# ^ @include +# ^ @include +def generator(): + yield from bar(42) + # ^ @keyword.return + # ^ @keyword.return |