Fix for unittest ModuleNotFoundError when discovering tests (#17369)

karthiknadig · karthiknadig · commit 7c51c8c2b407 · 2021-09-13T12:05:16.000-07:00
* Fix for ModuleNotFoundError

* Add news

* Ensure we use cwd

* linting
diff --git a/news/2 Fixes/17363.md b/news/2 Fixes/17363.md
@@ -0,0 +1 @@
+Fix for `unittest` ModuleNotFoundError when discovering tests.
diff --git a/pythonFiles/testing_tools/unittest_discovery.py b/pythonFiles/testing_tools/unittest_discovery.py
@@ -1,10 +1,12 @@
 import unittest
 import inspect
+import os
 import sys
 import traceback
 
 start_dir = sys.argv[1]
 pattern = sys.argv[2]
+sys.path.insert(0, os.getcwd())
 
 
 def get_sourceline(obj):

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	+Fix for `unittest` ModuleNotFoundError when discovering tests.