Better fix for modules using unittest.main(). Fixes regression caused by commit for...

diff --git a/Lib/unittest.py b/Lib/unittest.py
index d781354ba4201290256d27b5338066dd0e72c30b..5858d89e485a9060f3659e91d2a9fca75a40e6b1 100644 (file)
--- a/Lib/unittest.py
+++ b/Lib/unittest.py
@@ -1567,7 +1567,7 @@ Examples:
                 return
             if len(args) > 0:
                 self.testNames = args
-                if os.path.splitext(os.path.basename(__file__))[0] == 'unitest':
+                if sys.modules['unittest'] is sys.modules['__main__']:
                     self.module = None
             else:
                 self.testNames = (self.defaultTest,)