; 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) ) ))
; 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)
)
)
)