fac5118f10
Follow-up to #22325, where I missed a couple points in the review that were actually quite relevant. Closes #ISSUE Release Notes: - N/A
84 lines
2.3 KiB
Scheme
84 lines
2.3 KiB
Scheme
; subclasses of unittest.TestCase or TestCase
|
|
(
|
|
(class_definition
|
|
name: (identifier) @run @_unittest_class_name
|
|
superclasses: (argument_list
|
|
[(identifier) @_superclass
|
|
(attribute (identifier) @_superclass)]
|
|
)
|
|
(#eq? @_superclass "TestCase")
|
|
) @_python-unittest-class
|
|
(#set! tag python-unittest-class)
|
|
)
|
|
|
|
; test methods whose names start with `test` in a TestCase
|
|
(
|
|
(class_definition
|
|
name: (identifier) @_unittest_class_name
|
|
superclasses: (argument_list
|
|
[(identifier) @_superclass
|
|
(attribute (identifier) @_superclass)]
|
|
)
|
|
(#eq? @_superclass "TestCase")
|
|
body: (block
|
|
(function_definition
|
|
name: (identifier) @run @_unittest_method_name
|
|
(#match? @_unittest_method_name "^test.*")
|
|
) @_python-unittest-method
|
|
(#set! tag python-unittest-method)
|
|
)
|
|
)
|
|
)
|
|
|
|
; pytest functions
|
|
(
|
|
(module
|
|
(function_definition
|
|
name: (identifier) @run @_pytest_method_name
|
|
(#match? @_pytest_method_name "^test_")
|
|
) @_python-pytest-method
|
|
)
|
|
(#set! tag python-pytest-method)
|
|
)
|
|
|
|
; decorated pytest functions
|
|
(
|
|
(module
|
|
(decorated_definition
|
|
(decorator)+ @_decorator
|
|
definition: (function_definition
|
|
name: (identifier) @run @_pytest_method_name
|
|
(#match? @_pytest_method_name "^test_")
|
|
)
|
|
) @_python-pytest-method
|
|
)
|
|
(#set! tag python-pytest-method)
|
|
)
|
|
|
|
; pytest classes
|
|
(
|
|
(module
|
|
(class_definition
|
|
name: (identifier) @run @_pytest_class_name
|
|
(#match? @_pytest_class_name "^Test")
|
|
)
|
|
(#set! tag python-pytest-class)
|
|
)
|
|
)
|
|
|
|
; pytest class methods
|
|
(
|
|
(module
|
|
(class_definition
|
|
name: (identifier) @_pytest_class_name
|
|
(#match? @_pytest_class_name "^Test")
|
|
body: (block
|
|
(function_definition
|
|
name: (identifier) @run @_pytest_method_name
|
|
(#match? @_pytest_method_name "^test")
|
|
) @_python-pytest-method
|
|
(#set! tag python-pytest-method)
|
|
)
|
|
)
|
|
)
|
|
)
|