From f596a5125a50c8e2026984531e9be3fc111b3d49 Mon Sep 17 00:00:00 2001
From: Michael Foord <fuzzyman@voidspace.org.uk>
Date: Tue, 12 May 2009 10:49:13 +0000
Subject: Better fix for modules using unittest.main(). Fixes regression caused
 by commit for issue 5995. Michael Foord

---
 Lib/unittest.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Lib/unittest.py b/Lib/unittest.py
index d781354..5858d89 100644
--- 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,)
-- 
cgit v0.12