From 1a314a58d6d7695d226b56f6b61b1596993d5ac7 Mon Sep 17 00:00:00 2001
From: Salomon Popp <hi@salomonpopp.me>
Date: Wed, 12 Mar 2025 09:22:33 +0100
Subject: fix(python): highlight function calls correctly (#7728)

---
 tests/query/highlights/python/decorators.py | 22 ++++++++++++++++++++++
 tests/query/highlights/python/functions.py  | 13 +++++++++++++
 2 files changed, 35 insertions(+)
 create mode 100644 tests/query/highlights/python/decorators.py
 create mode 100644 tests/query/highlights/python/functions.py

(limited to 'tests/query/highlights/python')

diff --git a/tests/query/highlights/python/decorators.py b/tests/query/highlights/python/decorators.py
new file mode 100644
index 000000000..278e4e311
--- /dev/null
+++ b/tests/query/highlights/python/decorators.py
@@ -0,0 +1,22 @@
+from dataclasses import dataclass
+
+
+@dataclass
+#^^^^^^^^^ @attribute
+class Data:
+    _foo: str
+
+    @property
+#   ^ @attribute
+#    ^^^^^^^^ @attribute.builtin
+    def foo(self) -> str:
+        return self._foo
+
+
+@pytest.mark.filterwarnings("ignore::DeprecationWarning")
+#^^^^^^ @variable
+#       ^^^^ @variable.member
+#            ^^^^^^^^^^^^^^ @attribute
+def test_func():
+    pass
+
diff --git a/tests/query/highlights/python/functions.py b/tests/query/highlights/python/functions.py
new file mode 100644
index 000000000..6d3028433
--- /dev/null
+++ b/tests/query/highlights/python/functions.py
@@ -0,0 +1,13 @@
+def func() -> None: ...
+
+_ = func()
+#   ^^^^ @function.call
+
+"{}".format(1)
+#    ^^^^^^ @function.method.call
+
+class Foo:
+    def method(self) -> None: ...
+
+Foo().method()
+#     ^^^^^^ @function.method.call
-- 
cgit v1.2.3-70-g09d2