##// END OF EJS Templates
Backport PR #2170: Fix tab completion with IPython.embed_kernel()....
Backport PR #2170: Fix tab completion with IPython.embed_kernel(). Closes #2162.

File last commit:

r7018:bb6dab94
r7998:f4be28f0
Show More
ipdoctest.py
811 lines | 31.9 KiB | text/x-python | PythonLexer
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 """Nose Plugin that supports IPython doctests.
Limitations:
- When generating examples for use as doctests, make sure that you have
Bernardo B. Marques
remove all trailling spaces
r4872 pretty-printing OFF. This can be done either by setting the
``PlainTextFormatter.pprint`` option in your configuration file to False, or
Erik Tollerud
updated references to configuration file options to specify the class as well as the option itself
r4468 by interactively disabling it with %Pprint. This is required so that IPython
Bernardo B. Marques
remove all trailling spaces
r4872 output matches that of normal Python, which is used by doctest for internal
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 execution.
- Do not rely on specific prompt numbers for results (such as using
'_34==True', for example). For IPython tests run via an external process the
prompt numbers may be different, and IPython tests run as normal python code
won't even have these special _NN variables set at all.
"""
#-----------------------------------------------------------------------------
# Module imports
# From the standard library
Thomas Kluyver
For doctests, globals must now be a real dictionary.
r7018 import __builtin__ as builtin_mod
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 import commands
import doctest
import inspect
import logging
import os
import re
import sys
Fernando Perez
Improvements to namespace handling with %run....
r1425 import traceback
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 import unittest
from inspect import getmodule
Fernando Perez
Improvements to namespace handling with %run....
r1425 from StringIO import StringIO
# We are overriding the default doctest runner, so we need to import a few
# things from doctest directly
from doctest import (REPORTING_FLAGS, REPORT_ONLY_FIRST_FAILURE,
_unittest_reportflags, DocTestRunner,
_extract_future_flags, pdb, _OutputRedirectingPdb,
_exception_traceback,
linecache)
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334
# Third-party modules
import nose.core
from nose.plugins import doctests, Plugin
from nose.util import anyp, getpackage, test_address, resolve_name, tolist
#-----------------------------------------------------------------------------
# Module globals and other constants
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #-----------------------------------------------------------------------------
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334
log = logging.getLogger(__name__)
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #-----------------------------------------------------------------------------
Fernando Perez
Minor cleanups.
r1377 # Classes and functions
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #-----------------------------------------------------------------------------
Fernando Perez
Minor cleanups.
r1377
def is_extension_module(filename):
"""Return whether the given filename is an extension module.
This simply checks that the extension is either .so or .pyd.
"""
return os.path.splitext(filename)[1].lower() in ('.so','.pyd')
Fernando Perez
Slight improvements to doctest skipping....
r1764 class DocTestSkip(object):
"""Object wrapper for doctests to be skipped."""
Bernardo B. Marques
remove all trailling spaces
r4872
Fernando Perez
Slight improvements to doctest skipping....
r1764 ds_skip = """Doctest to skip.
>>> 1 #doctest: +SKIP
"""
Fernando Perez
Complete first pass on testing system. All tests pass on my box. Whew....
r1435 def __init__(self,obj):
self.obj = obj
def __getattribute__(self,key):
if key == '__doc__':
Fernando Perez
Slight improvements to doctest skipping....
r1764 return DocTestSkip.ds_skip
Fernando Perez
Complete first pass on testing system. All tests pass on my box. Whew....
r1435 else:
return getattr(object.__getattribute__(self,'obj'),key)
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 # Modified version of the one in the stdlib, that fixes a python bug (doctests
# not found in extension modules, http://bugs.python.org/issue3158)
class DocTestFinder(doctest.DocTestFinder):
def _from_module(self, module, object):
"""
Return true if the given object is defined in the given
module.
"""
if module is None:
return True
elif inspect.isfunction(object):
return module.__dict__ is object.func_globals
elif inspect.isbuiltin(object):
return module.__name__ == object.__module__
elif inspect.isclass(object):
return module.__name__ == object.__module__
elif inspect.ismethod(object):
# This one may be a bug in cython that fails to correctly set the
# __module__ attribute of methods, but since the same error is easy
# to make by extension code writers, having this safety in place
# isn't such a bad idea
return module.__name__ == object.im_class.__module__
elif inspect.getmodule(object) is not None:
return module is inspect.getmodule(object)
elif hasattr(object, '__module__'):
return module.__name__ == object.__module__
elif isinstance(object, property):
return True # [XX] no way not be sure.
else:
raise ValueError("object must be a class or function")
def _find(self, tests, obj, name, module, source_lines, globs, seen):
"""
Find tests for the given object and any contained objects, and
add them to `tests`.
"""
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #print '_find for:', obj, name, module # dbg
Fernando Perez
Complete first pass on testing system. All tests pass on my box. Whew....
r1435 if hasattr(obj,"skip_doctest"):
#print 'SKIPPING DOCTEST FOR:',obj # dbg
Fernando Perez
Slight improvements to doctest skipping....
r1764 obj = DocTestSkip(obj)
Fernando Perez
Complete first pass on testing system. All tests pass on my box. Whew....
r1435
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 doctest.DocTestFinder._find(self,tests, obj, name, module,
source_lines, globs, seen)
# Below we re-run pieces of the above method with manual modifications,
# because the original code is buggy and fails to correctly identify
# doctests in extension modules.
# Local shorthands
from inspect import isroutine, isclass, ismodule
# Look for tests in a module's contained objects.
if inspect.ismodule(obj) and self._recurse:
for valname, val in obj.__dict__.items():
valname1 = '%s.%s' % (name, valname)
if ( (isroutine(val) or isclass(val))
and self._from_module(module, val) ):
self._find(tests, val, valname1, module, source_lines,
globs, seen)
# Look for tests in a class's contained objects.
if inspect.isclass(obj) and self._recurse:
#print 'RECURSE into class:',obj # dbg
for valname, val in obj.__dict__.items():
# Special handling for staticmethod/classmethod.
if isinstance(val, staticmethod):
val = getattr(obj, valname)
if isinstance(val, classmethod):
val = getattr(obj, valname).im_func
# Recurse to methods, properties, and nested classes.
if ((inspect.isfunction(val) or inspect.isclass(val) or
inspect.ismethod(val) or
isinstance(val, property)) and
self._from_module(module, val)):
valname = '%s.%s' % (name, valname)
self._find(tests, val, valname, module, source_lines,
globs, seen)
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 class IPDoctestOutputChecker(doctest.OutputChecker):
Fernando Perez
Cleanups, document, working on support for full random tests.
r1429 """Second-chance checker with support for random tests.
Bernardo B. Marques
remove all trailling spaces
r4872
Fernando Perez
Cleanups, document, working on support for full random tests.
r1429 If the default comparison doesn't pass, this checker looks in the expected
output string for flags that tell us to ignore the output.
"""
Fernando Perez
Adjust regexps for random tests
r1453 random_re = re.compile(r'#\s*random\s+')
Bernardo B. Marques
remove all trailling spaces
r4872
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 def check_output(self, want, got, optionflags):
Fernando Perez
Cleanups, document, working on support for full random tests.
r1429 """Check output, accepting special markers embedded in the output.
Fernando Perez
Checkpoint with more tests working....
r1420
Fernando Perez
Cleanups, document, working on support for full random tests.
r1429 If the output didn't pass the default validation but the special string
'#random' is included, we accept it."""
# Let the original tester verify first, in case people have valid tests
# that happen to have a comment saying '#random' embedded in.
Fernando Perez
Checkpoint with more tests working....
r1420 ret = doctest.OutputChecker.check_output(self, want, got,
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 optionflags)
Fernando Perez
Cleanups, document, working on support for full random tests.
r1429 if not ret and self.random_re.search(want):
#print >> sys.stderr, 'RANDOM OK:',want # dbg
return True
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403
return ret
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 class DocTestCase(doctests.DocTestCase):
"""Proxy for DocTestCase: provides an address() method that
returns the correct address for the doctest case. Otherwise
acts as a proxy to the test case. To provide hints for address(),
an obj may also be passed -- this will be used as the test object
for purposes of determining the test address, if it is provided.
"""
Fernando Perez
Checkpoint where tests are recognized. Random tests not working yet.
r1378 # Note: this method was taken from numpy's nosetester module.
Fernando Perez
Checkpoint with more tests working....
r1420
# Subclass nose.plugins.doctests.DocTestCase to work around a bug in
Fernando Perez
Checkpoint where tests are recognized. Random tests not working yet.
r1378 # its constructor that blocks non-default arguments from being passed
# down into doctest.DocTestCase
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403
def __init__(self, test, optionflags=0, setUp=None, tearDown=None,
checker=None, obj=None, result_var='_'):
self._result_var = result_var
Fernando Perez
Checkpoint with more tests working....
r1420 doctests.DocTestCase.__init__(self, test,
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 optionflags=optionflags,
Fernando Perez
Checkpoint with more tests working....
r1420 setUp=setUp, tearDown=tearDown,
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 checker=checker)
# Now we must actually copy the original constructor from the stdlib
# doctest class, because we can't call it directly and a bug in nose
# means it never gets passed the right arguments.
Fernando Perez
Checkpoint with more tests working....
r1420
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 self._dt_optionflags = optionflags
self._dt_checker = checker
self._dt_test = test
Fernando Perez
Fix subtle bug in doctests (ipython namespace was being destroyed)....
r2445 self._dt_test_globs_ori = test.globs
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 self._dt_setUp = setUp
self._dt_tearDown = tearDown
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509 # XXX - store this runner once in the object!
runner = IPDocTestRunner(optionflags=optionflags,
checker=checker, verbose=False)
self._dt_runner = runner
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 # Each doctest should remember the directory it was loaded from, so
# things like %run work without too many contortions
self._ori_dir = os.path.dirname(test.filename)
Fernando Perez
Fix directory handling bug in tests, other small cleanups.
r1428
Fernando Perez
Improvements to namespace handling with %run....
r1425 # Modified runTest from the default stdlib
def runTest(self):
test = self._dt_test
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509 runner = self._dt_runner
Bernardo B. Marques
remove all trailling spaces
r4872
Fernando Perez
Improvements to namespace handling with %run....
r1425 old = sys.stdout
new = StringIO()
optionflags = self._dt_optionflags
if not (optionflags & REPORTING_FLAGS):
# The option flags don't include any reporting flags,
# so add the default reporting flags
optionflags |= _unittest_reportflags
try:
Fernando Perez
Fix directory handling bug in tests, other small cleanups.
r1428 # Save our current directory and switch out to the one where the
# test was originally created, in case another doctest did a
# directory change. We'll restore this in the finally clause.
Jörgen Stenarson
Search of getcwd and replace with getcwdu. Ignoring core/prompts.py
r4208 curdir = os.getcwdu()
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #print 'runTest in dir:', self._ori_dir # dbg
Fernando Perez
Fix directory handling bug in tests, other small cleanups.
r1428 os.chdir(self._ori_dir)
Fernando Perez
Improvements to namespace handling with %run....
r1425 runner.DIVIDER = "-"*70
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509 failures, tries = runner.run(test,out=new.write,
clear_globs=False)
Fernando Perez
Improvements to namespace handling with %run....
r1425 finally:
sys.stdout = old
Fernando Perez
Fix directory handling bug in tests, other small cleanups.
r1428 os.chdir(curdir)
Fernando Perez
Improvements to namespace handling with %run....
r1425
if failures:
raise self.failureException(self.format_failure(new.getvalue()))
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509 def setUp(self):
"""Modified test setup that syncs with ipython namespace"""
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #print "setUp test", self._dt_test.examples # dbg
Thomas Kluyver
For doctests, globals must now be a real dictionary.
r7018 if isinstance(self._dt_test.examples[0], IPExample):
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509 # for IPython examples *only*, we swap the globals with the ipython
# namespace, after updating it with the globals (which doctest
# fills with the necessary info from the module being tested).
Thomas Kluyver
Use user_ns as global namespace if it is passed without user_module, and add test for pickling interactively defined objects....
r5456 self.user_ns_orig = {}
self.user_ns_orig.update(_ip.user_ns)
Brian Granger
Continuing a massive refactor of everything.
r2205 _ip.user_ns.update(self._dt_test.globs)
Thomas Kluyver
For doctests, globals must now be a real dictionary.
r7018 # We must remove the _ key in the namespace, so that Python's
# doctest code sets it naturally
_ip.user_ns.pop('_', None)
_ip.user_ns['__builtins__'] = builtin_mod
Brian Granger
Continuing a massive refactor of everything.
r2205 self._dt_test.globs = _ip.user_ns
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509
Fernando Perez
Add a workaround for a nose problem so the test suite runs with nose 0.11.x
r2134 super(DocTestCase, self).setUp()
def tearDown(self):
Fernando Perez
Fix subtle bug in doctests (ipython namespace was being destroyed)....
r2445
# Undo the test.globs reassignment we made, so that the parent class
# teardown doesn't destroy the ipython namespace
Thomas Kluyver
For doctests, globals must now be a real dictionary.
r7018 if isinstance(self._dt_test.examples[0], IPExample):
Fernando Perez
Fix subtle bug in doctests (ipython namespace was being destroyed)....
r2445 self._dt_test.globs = self._dt_test_globs_ori
Thomas Kluyver
Use user_ns as global namespace if it is passed without user_module, and add test for pickling interactively defined objects....
r5456 _ip.user_ns.clear()
_ip.user_ns.update(self.user_ns_orig)
Bernardo B. Marques
remove all trailling spaces
r4872
Fernando Perez
Add a workaround for a nose problem so the test suite runs with nose 0.11.x
r2134 # XXX - fperez: I am not sure if this is truly a bug in nose 0.11, but
# it does look like one to me: its tearDown method tries to run
#
# delattr(__builtin__, self._result_var)
#
# without checking that the attribute really is there; it implicitly
# assumes it should have been set via displayhook. But if the
# displayhook was never called, this doesn't necessarily happen. I
# haven't been able to find a little self-contained example outside of
# ipython that would show the problem so I can report it to the nose
# team, but it does happen a lot in our code.
#
# So here, we just protect as narrowly as possible by trapping an
# attribute error whose message would be the name of self._result_var,
# and letting any other error propagate.
try:
super(DocTestCase, self).tearDown()
except AttributeError, exc:
Fernando Perez
Update nose fix to avoid a python 2.6 deprecation warning on exc.message
r2135 if exc.args[0] != self._result_var:
Fernando Perez
Add a workaround for a nose problem so the test suite runs with nose 0.11.x
r2134 raise
Fernando Perez
Checkpoint before merging with upstream
r1482
Fernando Perez
Checkpoint where tests are recognized. Random tests not working yet.
r1378
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 # A simple subclassing of the original with a different class name, so we can
# distinguish and treat differently IPython examples from pure python ones.
class IPExample(doctest.Example): pass
Fernando Perez
Minor cleanups.
r1377
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 class IPExternalExample(doctest.Example):
"""Doctest examples to be run in an external process."""
def __init__(self, source, want, exc_msg=None, lineno=0, indent=0,
options=None):
# Parent constructor
doctest.Example.__init__(self,source,want,exc_msg,lineno,indent,options)
# An EXTRA newline is needed to prevent pexpect hangs
self.source += '\n'
Fernando Perez
Minor cleanups.
r1377
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 class IPDocTestParser(doctest.DocTestParser):
"""
A class used to parse strings containing doctest examples.
Note: This is a version modified to properly recognize IPython input and
convert any IPython examples into valid Python ones.
"""
# This regular expression is used to find doctest examples in a
# string. It defines three groups: `source` is the source code
# (including leading indentation and prompts); `indent` is the
# indentation of the first (PS1) line of the source code; and
# `want` is the expected output (including leading indentation).
# Classic Python prompts or default IPython ones
_PS1_PY = r'>>>'
_PS2_PY = r'\.\.\.'
_PS1_IP = r'In\ \[\d+\]:'
_PS2_IP = r'\ \ \ \.\.\.+:'
_RE_TPL = r'''
# Source consists of a PS1 line followed by zero or more PS2 lines.
(?P<source>
(?:^(?P<indent> [ ]*) (?P<ps1> %s) .*) # PS1 line
(?:\n [ ]* (?P<ps2> %s) .*)*) # PS2 lines
\n? # a newline
# Want consists of any non-blank lines that do not start with PS1.
(?P<want> (?:(?![ ]*$) # Not a blank line
(?![ ]*%s) # Not a line starting with PS1
(?![ ]*%s) # Not a line starting with PS2
.*$\n? # But any other line
)*)
'''
_EXAMPLE_RE_PY = re.compile( _RE_TPL % (_PS1_PY,_PS2_PY,_PS1_PY,_PS2_PY),
re.MULTILINE | re.VERBOSE)
_EXAMPLE_RE_IP = re.compile( _RE_TPL % (_PS1_IP,_PS2_IP,_PS1_IP,_PS2_IP),
re.MULTILINE | re.VERBOSE)
Fernando Perez
Implement support for all random tests. Other minor cleanups.
r1430 # Mark a test as being fully random. In this case, we simply append the
# random marker ('#random') to each individual example's output. This way
# we don't need to modify any other code.
Fernando Perez
Adjust regexps for random tests
r1453 _RANDOM_TEST = re.compile(r'#\s*all-random\s+')
Fernando Perez
Implement support for all random tests. Other minor cleanups.
r1430
# Mark tests to be executed in an external process - currently unsupported.
Fernando Perez
Cleanups, document, working on support for full random tests.
r1429 _EXTERNAL_IP = re.compile(r'#\s*ipdoctest:\s*EXTERNAL')
Bernardo B. Marques
remove all trailling spaces
r4872
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 def ip2py(self,source):
"""Convert input IPython source into valid Python."""
out = []
newline = out.append
Fernando Perez
Fix problems with multiline doctests and add docs about testing....
r1868 #print 'IPSRC:\n',source,'\n###' # dbg
# The input source must be first stripped of all bracketing whitespace
# and turned into lines, so it looks to the parser like regular user
# input
for lnum,line in enumerate(source.strip().splitlines()):
Brian Granger
Continuing a massive refactor of everything.
r2205 newline(_ip.prefilter(line,lnum>0))
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 newline('') # ensure a closing newline, needed by doctest
Fernando Perez
Checkpoint with more tests working....
r1420 #print "PYSRC:", '\n'.join(out) # dbg
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 return '\n'.join(out)
def parse(self, string, name='<string>'):
"""
Divide the given string into examples and intervening text,
and return them as a list of alternating Examples and strings.
Line numbers for the Examples are 0-based. The optional
argument `name` is a name identifying this string, and is only
used for error messages.
"""
Fernando Perez
Checkpoint with more tests working....
r1420
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 #print 'Parse string:\n',string # dbg
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334
string = string.expandtabs()
# If all lines begin with the same indentation, then strip it.
min_indent = self._min_indent(string)
if min_indent > 0:
string = '\n'.join([l[min_indent:] for l in string.split('\n')])
output = []
charno, lineno = 0, 0
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509 # We make 'all random' tests by adding the '# random' mark to every
# block of output in the test.
Fernando Perez
Implement support for all random tests. Other minor cleanups.
r1430 if self._RANDOM_TEST.search(string):
random_marker = '\n# random'
else:
random_marker = ''
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 # Whether to convert the input from ipython to python syntax
ip2py = False
# Find all doctest examples in the string. First, try them as Python
# examples, then as IPython ones
terms = list(self._EXAMPLE_RE_PY.finditer(string))
if terms:
# Normal Python example
#print '-'*70 # dbg
#print 'PyExample, Source:\n',string # dbg
#print '-'*70 # dbg
Example = doctest.Example
else:
# It's an ipython example. Note that IPExamples are run
# in-process, so their syntax must be turned into valid python.
# IPExternalExamples are run out-of-process (via pexpect) so they
# don't need any filtering (a real ipython will be executing them).
terms = list(self._EXAMPLE_RE_IP.finditer(string))
Fernando Perez
Cleanups, document, working on support for full random tests.
r1429 if self._EXTERNAL_IP.search(string):
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 #print '-'*70 # dbg
#print 'IPExternalExample, Source:\n',string # dbg
#print '-'*70 # dbg
Example = IPExternalExample
else:
#print '-'*70 # dbg
#print 'IPExample, Source:\n',string # dbg
#print '-'*70 # dbg
Example = IPExample
ip2py = True
for m in terms:
# Add the pre-example text to `output`.
output.append(string[charno:m.start()])
# Update lineno (lines before this example)
lineno += string.count('\n', charno, m.start())
# Extract info from the regexp match.
(source, options, want, exc_msg) = \
self._parse_example(m, name, lineno,ip2py)
Fernando Perez
Implement support for all random tests. Other minor cleanups.
r1430
# Append the random-output marker (it defaults to empty in most
# cases, it's only non-empty for 'all-random' tests):
want += random_marker
Bernardo B. Marques
remove all trailling spaces
r4872
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 if Example is IPExternalExample:
options[doctest.NORMALIZE_WHITESPACE] = True
want += '\n'
Fernando Perez
Implement support for all random tests. Other minor cleanups.
r1430
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 # Create an Example, and add it to the list.
if not self._IS_BLANK_OR_COMMENT(source):
output.append(Example(source, want, exc_msg,
lineno=lineno,
indent=min_indent+len(m.group('indent')),
options=options))
# Update lineno (lines inside this example)
lineno += string.count('\n', m.start(), m.end())
# Update charno.
charno = m.end()
# Add any remaining post-example text to `output`.
output.append(string[charno:])
return output
def _parse_example(self, m, name, lineno,ip2py=False):
"""
Given a regular expression match from `_EXAMPLE_RE` (`m`),
return a pair `(source, want)`, where `source` is the matched
example's source code (with prompts and indentation stripped);
and `want` is the example's expected output (with indentation
stripped).
`name` is the string's name, and `lineno` is the line number
where the example starts; both are used for error messages.
Optional:
`ip2py`: if true, filter the input via IPython to convert the syntax
into valid python.
"""
# Get the example's indentation level.
indent = len(m.group('indent'))
# Divide source into lines; check that they're properly
# indented; and then strip their indentation & prompts.
source_lines = m.group('source').split('\n')
# We're using variable-length input prompts
ps1 = m.group('ps1')
ps2 = m.group('ps2')
ps1_len = len(ps1)
self._check_prompt_blank(source_lines, indent, name, lineno,ps1_len)
if ps2:
self._check_prefix(source_lines[1:], ' '*indent + ps2, name, lineno)
source = '\n'.join([sl[indent+ps1_len+1:] for sl in source_lines])
if ip2py:
# Convert source input from IPython into valid Python syntax
source = self.ip2py(source)
# Divide want into lines; check that it's properly indented; and
# then strip the indentation. Spaces before the last newline should
# be preserved, so plain rstrip() isn't good enough.
want = m.group('want')
want_lines = want.split('\n')
if len(want_lines) > 1 and re.match(r' *$', want_lines[-1]):
del want_lines[-1] # forget final newline & spaces after it
self._check_prefix(want_lines, ' '*indent, name,
lineno + len(source_lines))
# Remove ipython output prompt that might be present in the first line
want_lines[0] = re.sub(r'Out\[\d+\]: \s*?\n?','',want_lines[0])
want = '\n'.join([wl[indent:] for wl in want_lines])
# If `want` contains a traceback message, then extract it.
m = self._EXCEPTION_RE.match(want)
if m:
exc_msg = m.group('msg')
else:
exc_msg = None
# Extract options from the source.
options = self._find_options(source, name, lineno)
return source, options, want, exc_msg
def _check_prompt_blank(self, lines, indent, name, lineno, ps1_len):
"""
Given the lines of a source string (including prompts and
leading indentation), check to make sure that every prompt is
followed by a space character. If any line is not followed by
a space character, then raise ValueError.
Note: IPython-modified version which takes the input prompt length as a
parameter, so that prompts of variable length can be dealt with.
"""
space_idx = indent+ps1_len
min_len = space_idx+1
for i, line in enumerate(lines):
if len(line) >= min_len and line[space_idx] != ' ':
raise ValueError('line %r of the docstring for %s '
'lacks blank after %s: %r' %
(lineno+i+1, name,
line[indent:space_idx], line))
Fernando Perez
Improvements to namespace handling with %run....
r1425
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 SKIP = doctest.register_optionflag('SKIP')
Fernando Perez
Vastly cleaner solution for the namespace problem!
r1427 class IPDocTestRunner(doctest.DocTestRunner,object):
"""Test runner that synchronizes the IPython namespace with test globals.
"""
Bernardo B. Marques
remove all trailling spaces
r4872
Fernando Perez
Checkpoint with more tests working....
r1420 def run(self, test, compileflags=None, out=None, clear_globs=True):
Fernando Perez
Vastly cleaner solution for the namespace problem!
r1427 # Hack: ipython needs access to the execution context of the example,
# so that it can propagate user variables loaded by %run into
# test.globs. We put them here into our modified %run as a function
# attribute. Our new %run will then only make the namespace update
# when called (rather than unconconditionally updating test.globs here
# for all examples, most of which won't be calling %run anyway).
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #_ip._ipdoctest_test_globs = test.globs
#_ip._ipdoctest_test_filename = test.filename
test.globs.update(_ip.user_ns)
Bernardo B. Marques
remove all trailling spaces
r4872
Fernando Perez
Vastly cleaner solution for the namespace problem!
r1427 return super(IPDocTestRunner,self).run(test,
compileflags,out,clear_globs)
Fernando Perez
Checkpoint with more tests working....
r1420
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 class DocFileCase(doctest.DocFileCase):
"""Overrides to provide filename
"""
def address(self):
return (self._dt_test.filename, None, None)
class ExtensionDoctest(doctests.Doctest):
"""Nose Plugin that supports doctests in extension modules.
"""
name = 'extdoctest' # call nosetests with --with-extdoctest
enabled = True
Fernando Perez
Cleanup testing machinery.
r1851 def __init__(self,exclude_patterns=None):
"""Create a new ExtensionDoctest plugin.
Parameters
----------
exclude_patterns : sequence of strings, optional
These patterns are compiled as regular expressions, subsequently used
to exclude any filename which matches them from inclusion in the test
suite (using pattern.search(), NOT pattern.match() ).
"""
Fernando Perez
- Make ipdoctest a little cleaner by giving it separate option names....
r1910
Fernando Perez
Cleanup testing machinery.
r1851 if exclude_patterns is None:
exclude_patterns = []
self.exclude_patterns = map(re.compile,exclude_patterns)
doctests.Doctest.__init__(self)
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 def options(self, parser, env=os.environ):
Plugin.options(self, parser, env)
Fernando Perez
Checkpoint before merging with upstream
r1568 parser.add_option('--doctest-tests', action='store_true',
dest='doctest_tests',
default=env.get('NOSE_DOCTEST_TESTS',True),
help="Also look for doctests in test modules. "
"Note that classes, methods and functions should "
"have either doctests or non-doctest tests, "
"not both. [NOSE_DOCTEST_TESTS]")
parser.add_option('--doctest-extension', action="append",
dest="doctestExtension",
help="Also look for doctests in files with "
"this extension [NOSE_DOCTEST_EXTENSION]")
# Set the default as a list, if given in env; otherwise
# an additional value set on the command line will cause
# an error.
env_setting = env.get('NOSE_DOCTEST_EXTENSION')
if env_setting is not None:
parser.set_defaults(doctestExtension=tolist(env_setting))
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334
def configure(self, options, config):
Plugin.configure(self, options, config)
Matthew Brett
BF - allow nose with-doctest setting in environment...
r4567 # Pull standard doctest plugin out of config; we will do doctesting
config.plugins.plugins = [p for p in config.plugins.plugins
if p.name != 'doctest']
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 self.doctest_tests = options.doctest_tests
self.extension = tolist(options.doctestExtension)
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 self.parser = doctest.DocTestParser()
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509 self.finder = DocTestFinder()
self.checker = IPDoctestOutputChecker()
Fernando Perez
Checkpoint with more tests working....
r1420 self.globs = None
self.extraglobs = None
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334
Fernando Perez
Fixes to testing system: ipdocetst plugin wasn't being properly loaded.
r1761
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 def loadTestsFromExtensionModule(self,filename):
bpath,mod = os.path.split(filename)
modname = os.path.splitext(mod)[0]
try:
sys.path.append(bpath)
module = __import__(modname)
tests = list(self.loadTestsFromModule(module))
finally:
sys.path.pop()
return tests
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 # NOTE: the method below is almost a copy of the original one in nose, with
# a few modifications to control output checking.
Fernando Perez
Checkpoint with more tests working....
r1420
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 def loadTestsFromModule(self, module):
Fernando Perez
Fixes to testing system: ipdocetst plugin wasn't being properly loaded.
r1761 #print '*** ipdoctest - lTM',module # dbg
Bernardo B. Marques
remove all trailling spaces
r4872
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 if not self.matches(module.__name__):
log.debug("Doctest doesn't want module %s", module)
return
Fernando Perez
Checkpoint with more tests working....
r1420
tests = self.finder.find(module,globs=self.globs,
extraglobs=self.extraglobs)
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 if not tests:
return
Fernando Perez
Fix directory handling bug in tests, other small cleanups.
r1428
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509 # always use whitespace and ellipsis options
optionflags = doctest.NORMALIZE_WHITESPACE | doctest.ELLIPSIS
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 tests.sort()
module_file = module.__file__
if module_file[-4:] in ('.pyc', '.pyo'):
module_file = module_file[:-1]
for test in tests:
if not test.examples:
continue
if not test.filename:
test.filename = module_file
Bernardo B. Marques
remove all trailling spaces
r4872
Fernando Perez
Checkpoint with more tests working....
r1420 yield DocTestCase(test,
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403 optionflags=optionflags,
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509 checker=self.checker)
Fernando Perez
Local checkpoint of changes to testing machinery. Work in progress.
r1403
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 def loadTestsFromFile(self, filename):
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #print "ipdoctest - from file", filename # dbg
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 if is_extension_module(filename):
for t in self.loadTestsFromExtensionModule(filename):
yield t
else:
Fernando Perez
Checkpoint with more tests working....
r1420 if self.extension and anyp(filename.endswith, self.extension):
name = os.path.basename(filename)
dh = open(filename)
try:
doc = dh.read()
finally:
dh.close()
test = self.parser.get_doctest(
doc, globs={'__file__': filename}, name=name,
filename=filename, lineno=0)
if test.examples:
#print 'FileCase:',test.examples # dbg
yield DocFileCase(test)
else:
yield False # no tests to load
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334
def wantFile(self,filename):
"""Return whether the given filename should be scanned for tests.
Modified version that accepts extension modules as valid containers for
doctests.
"""
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #print '*** ipdoctest- wantFile:',filename # dbg
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334
Fernando Perez
Cleanup testing machinery.
r1851 for pat in self.exclude_patterns:
if pat.search(filename):
Administrator
Fixing bugs with the testing system.
r1981 # print '###>>> SKIP:',filename # dbg
Fernando Perez
Fixes to testing....
r1376 return False
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 if is_extension_module(filename):
return True
else:
return doctests.Doctest.wantFile(self,filename)
MinRK
add wantDirectory to ipdoctest, so that directories will be checked for exclusion...
r5291 def wantDirectory(self, directory):
"""Return whether the given directory should be scanned for tests.
Modified version that supports exclusions.
"""
for pat in self.exclude_patterns:
if pat.search(directory):
return False
return True
Fernando Perez
Minor cleanups.
r1377
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 class IPythonDoctest(ExtensionDoctest):
"""Nose Plugin that supports doctests in extension modules.
"""
name = 'ipdoctest' # call nosetests with --with-ipdoctest
enabled = True
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414
Fernando Perez
Override makeTest to correctly support path/to/test.py:test_function syntax....
r1763 def makeTest(self, obj, parent):
"""Look for doctests in the given object, which will be a
function, method or class.
"""
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #print 'Plugin analyzing:', obj, parent # dbg
Fernando Perez
Override makeTest to correctly support path/to/test.py:test_function syntax....
r1763 # always use whitespace and ellipsis options
optionflags = doctest.NORMALIZE_WHITESPACE | doctest.ELLIPSIS
doctests = self.finder.find(obj, module=getmodule(parent))
if doctests:
for test in doctests:
if len(test.examples) == 0:
continue
yield DocTestCase(test, obj=obj,
optionflags=optionflags,
checker=self.checker)
Fernando Perez
- Make ipdoctest a little cleaner by giving it separate option names....
r1910 def options(self, parser, env=os.environ):
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #print "Options for nose plugin:", self.name # dbg
Fernando Perez
- Make ipdoctest a little cleaner by giving it separate option names....
r1910 Plugin.options(self, parser, env)
parser.add_option('--ipdoctest-tests', action='store_true',
dest='ipdoctest_tests',
default=env.get('NOSE_IPDOCTEST_TESTS',True),
help="Also look for doctests in test modules. "
"Note that classes, methods and functions should "
"have either doctests or non-doctest tests, "
"not both. [NOSE_IPDOCTEST_TESTS]")
parser.add_option('--ipdoctest-extension', action="append",
Fernando Perez
Fix naming convention to PEP 8.
r1914 dest="ipdoctest_extension",
Fernando Perez
- Make ipdoctest a little cleaner by giving it separate option names....
r1910 help="Also look for doctests in files with "
"this extension [NOSE_IPDOCTEST_EXTENSION]")
# Set the default as a list, if given in env; otherwise
# an additional value set on the command line will cause
# an error.
env_setting = env.get('NOSE_IPDOCTEST_EXTENSION')
if env_setting is not None:
Fernando Perez
Fix naming convention to PEP 8.
r1914 parser.set_defaults(ipdoctest_extension=tolist(env_setting))
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334
Fernando Perez
- Make ipdoctest a little cleaner by giving it separate option names....
r1910 def configure(self, options, config):
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #print "Configuring nose plugin:", self.name # dbg
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 Plugin.configure(self, options, config)
Matthew Brett
BF - allow nose with-doctest setting in environment...
r4567 # Pull standard doctest plugin out of config; we will do doctesting
config.plugins.plugins = [p for p in config.plugins.plugins
if p.name != 'doctest']
Fernando Perez
- Make ipdoctest a little cleaner by giving it separate option names....
r1910 self.doctest_tests = options.ipdoctest_tests
Fernando Perez
Fix naming convention to PEP 8.
r1914 self.extension = tolist(options.ipdoctest_extension)
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509
Fernando Perez
Added Nose support for IPython doctests and extension modules.
r1334 self.parser = IPDocTestParser()
self.finder = DocTestFinder(parser=self.parser)
Fernando Perez
Correctly implement namespace handling. Minor cleanups....
r1509 self.checker = IPDoctestOutputChecker()
Fernando Perez
Checkpoint with more tests working....
r1420 self.globs = None
self.extraglobs = None