add describe type to class creation function

eleanorjboyd · eleanorjboyd · commit be9dfebb5055 · 2024-11-07T14:30:37.000-08:00
diff --git a/python_files/vscode_pytest/__init__.py b/python_files/vscode_pytest/__init__.py
@@ -703,7 +703,7 @@ def create_session_node(session: pytest.Session) -> TestNode:
     }
 
 
-def create_class_node(class_module: pytest.Class) -> TestNode:
+def create_class_node(class_module: pytest.Class | DescribeBlock) -> TestNode:
     """Creates a class node from a pytest class object.
 
     Keyword arguments:

Original file line number	Diff line number	Diff line change
`@@ -703,7 +703,7 @@ def create_session_node(session: pytest.Session) -> TestNode:`
`703`	`703`	`}`
`704`	`704`
`705`	`705`
`706`		`-def create_class_node(class_module: pytest.Class) -> TestNode:`
	`706`	`+def create_class_node(class_module: pytest.Class \| DescribeBlock) -> TestNode:`
`707`	`707`	`"""Creates a class node from a pytest class object.`
`708`	`708`
`709`	`709`	`Keyword arguments:`