Add parent directory to sys.path for unittest discovery and execution (…

…microsoft#23712) fixes microsoft#23392
eleanorjboyd · Jun 27, 2024 · bd6a924 · bd6a924
1 parent 62ded1f
commit bd6a924
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/python_files/unittestadapter/discovery.py b/python_files/unittestadapter/discovery.py
@@ -57,6 +57,11 @@ def discover_tests(
     }
     """
     cwd = os.path.abspath(start_dir)
+    if "/" in start_dir:  #  is a subdir
+        parent_dir = os.path.dirname(start_dir)
+        sys.path.insert(0, parent_dir)
+    else:
+        sys.path.insert(0, cwd)
     payload: DiscoveryPayloadDict = {"cwd": cwd, "status": "success", "tests": None}
     tests = None
     error: List[str] = []

diff --git a/python_files/unittestadapter/execution.py b/python_files/unittestadapter/execution.py
@@ -195,6 +195,11 @@ def run_tests(
     locals: Optional[bool] = None,
 ) -> ExecutionPayloadDict:
     cwd = os.path.abspath(start_dir)
+    if "/" in start_dir:  #  is a subdir
+        parent_dir = os.path.dirname(start_dir)
+        sys.path.insert(0, parent_dir)
+    else:
+        sys.path.insert(0, cwd)
     status = TestExecutionStatus.error
     error = None
     payload: ExecutionPayloadDict = {"cwd": cwd, "status": status, "result": None}