Fix Python type checking errors and format code

- Added NotRequired to imports for optional TypedDict fields
- Made lineno an optional field in TestNode TypedDict
- Formatted Python files with ruff

Co-authored-by: eleanorjboyd <26030610+eleanorjboyd@users.noreply.github.com>

Author: Copilot

python_files/tests/pytestadapter/expected_discovery_test_output.py

@@ -100,9 +100,7 @@
                         "unittest_pytest_same_file.py::TestExample",
                         unit_pytest_same_file_path,
                     ),
-                    "lineno": find_class_line_number(
-                        "TestExample", unit_pytest_same_file_path
-                    ),
+                    "lineno": find_class_line_number("TestExample", unit_pytest_same_file_path),
                 },
                 {
                     "name": "test_true_pytest",
@@ -215,9 +213,7 @@
                                 "unittest_folder/test_add.py::TestAddFunction",
                                 test_add_path,
                             ),
-                            "lineno": find_class_line_number(
-                                "TestAddFunction", test_add_path
-                            ),
+                            "lineno": find_class_line_number("TestAddFunction", test_add_path),
                         },
                         {
                             "name": "TestDuplicateFunction",
@@ -573,9 +569,7 @@
                         "parametrize_tests.py::TestClass",
                         parameterize_tests_path,
                     ),
-                    "lineno": find_class_line_number(
-                        "TestClass", parameterize_tests_path
-                    ),
+                    "lineno": find_class_line_number("TestClass", parameterize_tests_path),
                     "children": [
                         {
                             "name": "test_adding",
@@ -952,9 +946,7 @@
                         "test_multi_class_nest.py::TestFirstClass",
                         TEST_MULTI_CLASS_NEST_PATH,
                     ),
-                    "lineno": find_class_line_number(
-                        "TestFirstClass", TEST_MULTI_CLASS_NEST_PATH
-                    ),
+                    "lineno": find_class_line_number("TestFirstClass", TEST_MULTI_CLASS_NEST_PATH),
                     "children": [
                         {
                             "name": "TestSecondClass",
@@ -1547,9 +1539,7 @@
                                 "pytest_describe_plugin/describe_only.py::describe_A",
                                 describe_only_path,
                             ),
-                            "lineno": find_class_line_number(
-                                "describe_A", describe_only_path
-                            ),
+                            "lineno": find_class_line_number("describe_A", describe_only_path),
                         }
                     ],
                 }
@@ -1673,9 +1663,7 @@
                                 "pytest_describe_plugin/nested_describe.py::describe_list",
                                 nested_describe_path,
                             ),
-                            "lineno": find_class_line_number(
-                                "describe_list", nested_describe_path
-                            ),
+                            "lineno": find_class_line_number("describe_list", nested_describe_path),
                         }
                     ],
                 }

python_files/tests/pytestadapter/helpers.py

@@ -382,9 +382,11 @@ def find_class_line_number(class_name: str, test_file_path) -> str:
         for i, line in enumerate(f):
             # Match "class ClassName" or "class ClassName(" or "class ClassName:"
             # Also match "def ClassName(" for pytest-describe blocks
-            if line.strip().startswith(f"class {class_name}") or line.strip().startswith(
-                f"class {class_name}("
-            ) or line.strip().startswith(f"def {class_name}("):
+            if (
+                line.strip().startswith(f"class {class_name}")
+                or line.strip().startswith(f"class {class_name}(")
+                or line.strip().startswith(f"def {class_name}(")
+            ):
                 return str(i + 1)
     error_str: str = f"Class {class_name!r} not found on any line in {test_file_path}"
     raise ValueError(error_str)

python_files/vscode_pytest/__init__.py

@@ -10,7 +10,17 @@
 import pathlib
 import sys
 import traceback
-from typing import TYPE_CHECKING, Any, Dict, Generator, Literal, Protocol, TypedDict, cast
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Dict,
+    Generator,
+    Literal,
+    NotRequired,
+    Protocol,
+    TypedDict,
+    cast,
+)
 
 import pytest
 
@@ -52,6 +62,7 @@ class TestNode(TestData):
     """A general class that handles all test data which contains children."""
 
     children: list[TestNode | TestItem | None]
+    lineno: NotRequired[str]  # Optional field for class/function nodes
 
 
 class VSCodePytestError(Exception):