aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--queries/python/highlights.scm1
-rw-r--r--tests/query/highlights/python/future_import.py4
2 files changed, 5 insertions, 0 deletions
diff --git a/queries/python/highlights.scm b/queries/python/highlights.scm
index 681b9e4de..18ed5162f 100644
--- a/queries/python/highlights.scm
+++ b/queries/python/highlights.scm
@@ -224,6 +224,7 @@
] @keyword.return
(yield "from" @keyword.return)
+(future_import_statement "from" @include "__future__" @constant.builtin)
(import_from_statement "from" @include)
"import" @include
diff --git a/tests/query/highlights/python/future_import.py b/tests/query/highlights/python/future_import.py
new file mode 100644
index 000000000..7495588a9
--- /dev/null
+++ b/tests/query/highlights/python/future_import.py
@@ -0,0 +1,4 @@
+from __future__ import print_function
+# ^ @include
+# ^ @constant.builtin
+# ^ @include