Add typing_extensions for NotRequired support and update imports

Author: eleanorjboyd

build/test-requirements.txt

@@ -28,6 +28,9 @@ namedpipe; platform_system == "Windows"
 # typing for Django files
 django-stubs
 
+# Required for NotRequired import in adapter modules
+typing_extensions
+
 coverage
 pytest-cov
 pytest-json

python_files/unittestadapter/django_test_runner.py

@@ -1,6 +1,8 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 
+from __future__ import annotations
+
 import os
 import pathlib
 import sys

python_files/unittestadapter/pvsc_utils.py

@@ -1,6 +1,8 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 
+from __future__ import annotations
+
 import argparse
 import atexit
 import doctest

python_files/vscode_pytest/__init__.py

@@ -16,12 +16,18 @@
     Dict,
     Generator,
     Literal,
-    NotRequired,
     Protocol,
     TypedDict,
     cast,
 )
 
+# NotRequired is only available in typing from Python 3.11+
+# For earlier versions, use typing_extensions
+try:
+    from typing import NotRequired
+except ImportError:
+    from typing_extensions import NotRequired
+
 import pytest
 
 if TYPE_CHECKING: