From: Nick Coghlan <ncoghlan@gmail.com>
Date: Fri, 3 Dec 2010 07:44:33 +0000 (+0000)
Subject: Partially revert r78719 - it removed a check that is still needed in some cases ... 
X-Git-Tag: v3.2b1~107
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=4c4c0f2fe6e6d9058911c9383bac57e4deefddb8;p=python

Partially revert r78719 - it removed a check that is still needed in some cases (i.e. this will allow Michael to add the test.__main__ support that broke the buildbots previously)
---

diff --git a/Lib/test/regrtest.py b/Lib/test/regrtest.py
index 9eecd1dc1a..9d82b71c55 100755
--- a/Lib/test/regrtest.py
+++ b/Lib/test/regrtest.py
@@ -1468,7 +1468,20 @@ class _ExpectedSkips:
         assert self.isvalid()
         return self.expected
 
+
 if __name__ == '__main__':
+    # Remove regrtest.py's own directory from the module search path. Despite
+    # the elimination of implicit relative imports, this is still needed to
+    # ensure that submodules of the test package do not inappropriately appear
+    # as top-level modules even when people (or buildbots!) invoke regrtest.py
+    # directly instead of using the -m switch
+    mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
+    i = len(sys.path)
+    while i >= 0:
+        i -= 1
+        if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
+            del sys.path[i]
+
     # findtestdir() gets the dirname out of __file__, so we have to make it
     # absolute before changing the working directory.
     # For example __file__ may be relative when running trace or profile.
diff --git a/Misc/NEWS b/Misc/NEWS
index 4d40b70812..abd37b952c 100644
--- a/Misc/NEWS
+++ b/Misc/NEWS
@@ -191,6 +191,9 @@ C-API
 Tests
 -----
 
+- regrtest.py once again ensures the test directory is removed from
+  sys.path when it is invoked directly as the __main__ module
+
 - Issue #9424: Deprecate the `unittest.TestCase` methods `assertEquals`,
   `assertNotEquals`, `assertAlmostEquals`, `assertNotAlmostEquals` and `assert_`
   and replace them with the correct methods in the Python test suite.