diff options
| -rw-r--r-- | setuptools/tests/doctest.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/setuptools/tests/doctest.py b/setuptools/tests/doctest.py index fcaf3a4d..bffce58f 100644 --- a/setuptools/tests/doctest.py +++ b/setuptools/tests/doctest.py @@ -1330,11 +1330,13 @@ class DocTestRunner: __LINECACHE_FILENAME_RE = re.compile(r'<doctest ' r'(?P<name>[\w\.]+)' r'\[(?P<examplenum>\d+)\]>$') - def __patched_linecache_getlines(self, filename): + def __patched_linecache_getlines(self, filename, module_globals=None): m = self.__LINECACHE_FILENAME_RE.match(filename) if m and m.group('name') == self.test.name: example = self.test.examples[int(m.group('examplenum'))] return example.source.splitlines(True) + 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) |