Retain 2.3/2.4 compatibility as well...

git-svn-id: http://svn.python.org/projects/sandbox/trunk/setuptools@51931 6015fed2-1504-0410-9fe1-9d1591cc4771

1 files changed, 3 insertions, 1 deletions

diff --git a/doctest.py b/doctest.py
index 70c355a..7fe7fa0 100644
--- a/doctest.py
+++ b/doctest.py
@@ -1333,8 +1333,10 @@ class DocTestRunner:
         if m and m.group('name') == self.test.name:
             example = self.test.examples[int(m.group('examplenum'))]
             return example.source.splitlines(True)
-        else:
+        elif self.save_linecache_getlines.func_code.co_argcount>1:
             return self.save_linecache_getlines(filename, module_globals)
+        else:
+            return self.save_linecache_getlines(filename)
 
     def run(self, test, compileflags=None, out=None, clear_globs=True):
         """