summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Lib/doctest.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/Lib/doctest.py b/Lib/doctest.py
index 64a0375..e2b595e 100644
--- a/Lib/doctest.py
+++ b/Lib/doctest.py
@@ -1360,7 +1360,10 @@ class DocTestRunner:
linecache.getlines = self.save_linecache_getlines
if clear_globs:
test.globs.clear()
- __builtins__['_'] = None
+ if __name__ == "__main__":
+ __builtins__._ = None
+ else:
+ __builtins__['_'] = None
#/////////////////////////////////////////////////////////////////
# Summarization