iptest.py
437 lines
| 15.7 KiB
| text/x-python
|
PythonLexer
Fernando Perez
|
r1574 | # -*- coding: utf-8 -*- | ||
"""IPython Test Suite Runner. | ||||
Fernando Perez
|
r1851 | |||
Brian Granger
|
r1972 | This module provides a main entry point to a user script to test IPython | ||
itself from the command line. There are two ways of running this script: | ||||
1. With the syntax `iptest all`. This runs our entire test suite by | ||||
Brian Granger
|
r3486 | calling this script (with different arguments) recursively. This | ||
Brian Granger
|
r1972 | causes modules and package to be tested in different processes, using nose | ||
or trial where appropriate. | ||||
2. With the regular nose syntax, like `iptest -vvs IPython`. In this form | ||||
the script simply calls nose, but with special command line flags and | ||||
plugins loaded. | ||||
Fernando Perez
|
r1574 | """ | ||
Fernando Perez
|
r1851 | #----------------------------------------------------------------------------- | ||
Brian Granger
|
r2498 | # Copyright (C) 2009 The IPython Development Team | ||
# | ||||
# Distributed under the terms of the BSD License. The full license is in | ||||
# the file COPYING, distributed as part of this software. | ||||
#----------------------------------------------------------------------------- | ||||
#----------------------------------------------------------------------------- | ||||
# Imports | ||||
Fernando Perez
|
r1851 | #----------------------------------------------------------------------------- | ||
Fernando Perez
|
r2442 | # Stdlib | ||
Brian Granger
|
r1972 | import os | ||
import os.path as path | ||||
Fernando Perez
|
r2399 | import signal | ||
Fernando Perez
|
r1574 | import sys | ||
Brian Granger
|
r1972 | import subprocess | ||
Fernando Perez
|
r2111 | import tempfile | ||
Brian Granger
|
r1972 | import time | ||
Fernando Perez
|
r1574 | import warnings | ||
Fernando Perez
|
r2442 | # Note: monkeypatch! | ||
# We need to monkeypatch a small problem in nose itself first, before importing | ||||
# it for actual use. This should get into nose upstream, but its release cycle | ||||
# is slow and we need it for our parametric tests to work correctly. | ||||
Fernando Perez
|
r2480 | from IPython.testing import nosepatch | ||
Fernando Perez
|
r2442 | # Now, proceed to import nose itself | ||
Fernando Perez
|
r1574 | import nose.plugins.builtin | ||
Fernando Perez
|
r1851 | from nose.core import TestProgram | ||
Fernando Perez
|
r1574 | |||
Fernando Perez
|
r2442 | # Our own imports | ||
Brian Granger
|
r2498 | from IPython.utils.path import get_ipython_module_path | ||
from IPython.utils.process import find_cmd, pycmd2argv | ||||
from IPython.utils.sysinfo import sys_info | ||||
Fernando Perez
|
r2480 | from IPython.testing import globalipapp | ||
from IPython.testing.plugin.ipdoctest import IPythonDoctest | ||||
Paul Ivanov
|
r3511 | from IPython.external.decorators import KnownFailure | ||
Fernando Perez
|
r1574 | |||
Brian Granger
|
r1979 | pjoin = path.join | ||
Fernando Perez
|
r2494 | |||
#----------------------------------------------------------------------------- | ||||
# Globals | ||||
#----------------------------------------------------------------------------- | ||||
Fernando Perez
|
r1851 | #----------------------------------------------------------------------------- | ||
Fernando Perez
|
r2398 | # Warnings control | ||
#----------------------------------------------------------------------------- | ||||
Brian Granger
|
r2512 | |||
Fernando Perez
|
r2398 | # Twisted generates annoying warnings with Python 2.6, as will do other code | ||
# that imports 'sets' as of today | ||||
warnings.filterwarnings('ignore', 'the sets module is deprecated', | ||||
DeprecationWarning ) | ||||
Fernando Perez
|
r2461 | # This one also comes from Twisted | ||
warnings.filterwarnings('ignore', 'the sha module is deprecated', | ||||
DeprecationWarning) | ||||
Fernando Perez
|
r2483 | # Wx on Fedora11 spits these out | ||
warnings.filterwarnings('ignore', 'wxPython/wxWidgets release number mismatch', | ||||
UserWarning) | ||||
Fernando Perez
|
r2398 | #----------------------------------------------------------------------------- | ||
Administrator
|
r1981 | # Logic for skipping doctests | ||
Fernando Perez
|
r1851 | #----------------------------------------------------------------------------- | ||
MinRK
|
r3640 | def test_for(mod, min_version=None): | ||
Administrator
|
r1981 | """Test to see if mod is importable.""" | ||
try: | ||||
__import__(mod) | ||||
Fernando Perez
|
r2488 | except (ImportError, RuntimeError): | ||
# GTK reports Runtime error if it can't be initialized even if it's | ||||
# importable. | ||||
Administrator
|
r1981 | return False | ||
else: | ||||
MinRK
|
r3640 | if min_version: | ||
return sys.modules[mod].__version__ >= min_version | ||||
else: | ||||
return True | ||||
Administrator
|
r1981 | |||
Fernando Perez
|
r2496 | # Global dict where we can store information on what we have and what we don't | ||
# have available at test run time | ||||
have = {} | ||||
have['curses'] = test_for('_curses') | ||||
Fernando Perez
|
r3731 | have['matplotlib'] = test_for('matplotlib') | ||
have['pexpect'] = test_for('pexpect') | ||||
have['pymongo'] = test_for('pymongo') | ||||
Fernando Perez
|
r2496 | have['wx'] = test_for('wx') | ||
have['wx.aui'] = test_for('wx.aui') | ||||
MinRK
|
r3779 | if os.name == 'nt': | ||
MinRK
|
r3858 | have['zmq'] = test_for('zmq', '2.1.7') | ||
MinRK
|
r3779 | else: | ||
have['zmq'] = test_for('zmq', '2.1.4') | ||||
MinRK
|
r3761 | have['qt'] = test_for('IPython.external.qt') | ||
Administrator
|
r1981 | |||
Fernando Perez
|
r2494 | #----------------------------------------------------------------------------- | ||
# Functions and classes | ||||
#----------------------------------------------------------------------------- | ||||
Administrator
|
r1980 | |||
Fernando Perez
|
r2496 | def report(): | ||
"""Return a string with a summary report of test-related variables.""" | ||||
Fernando Perez
|
r3204 | out = [ sys_info(), '\n'] | ||
Fernando Perez
|
r2496 | |||
avail = [] | ||||
not_avail = [] | ||||
for k, is_avail in have.items(): | ||||
if is_avail: | ||||
avail.append(k) | ||||
else: | ||||
not_avail.append(k) | ||||
if avail: | ||||
out.append('\nTools and libraries available at test time:\n') | ||||
avail.sort() | ||||
out.append(' ' + ' '.join(avail)+'\n') | ||||
if not_avail: | ||||
out.append('\nTools and libraries NOT available at test time:\n') | ||||
not_avail.sort() | ||||
out.append(' ' + ' '.join(not_avail)+'\n') | ||||
return ''.join(out) | ||||
Brian Granger
|
r2079 | def make_exclude(): | ||
Fernando Perez
|
r2454 | """Make patterns of modules and packages to exclude from testing. | ||
Brian Granger
|
r2512 | |||
Fernando Perez
|
r2454 | For the IPythonDoctest plugin, we need to exclude certain patterns that | ||
cause testing problems. We should strive to minimize the number of | ||||
Brian Granger
|
r2512 | skipped modules, since this means untested code. | ||
Fernando Perez
|
r2454 | These modules and packages will NOT get scanned by nose at all for tests. | ||
""" | ||||
# Simple utility to make IPython paths more readably, we need a lot of | ||||
# these below | ||||
ipjoin = lambda *paths: pjoin('IPython', *paths) | ||||
exclusions = [ipjoin('external'), | ||||
Fernando Perez
|
r2417 | pjoin('IPython_doctest_plugin'), | ||
Fernando Perez
|
r2454 | ipjoin('quarantine'), | ||
ipjoin('deathrow'), | ||||
ipjoin('testing', 'attic'), | ||||
Fernando Perez
|
r2461 | # This guy is probably attic material | ||
Fernando Perez
|
r2454 | ipjoin('testing', 'mkdoctests'), | ||
Fernando Perez
|
r2461 | # Testing inputhook will need a lot of thought, to figure out | ||
# how to have tests that don't lock up with the gui event | ||||
# loops in the picture | ||||
Fernando Perez
|
r2454 | ipjoin('lib', 'inputhook'), | ||
Fernando Perez
|
r2417 | # Config files aren't really importable stand-alone | ||
Fernando Perez
|
r2454 | ipjoin('config', 'default'), | ||
ipjoin('config', 'profile'), | ||||
Fernando Perez
|
r2417 | ] | ||
Brian Granger
|
r2079 | |||
Fernando Perez
|
r2496 | if not have['wx']: | ||
Fernando Perez
|
r2454 | exclusions.append(ipjoin('lib', 'inputhookwx')) | ||
Thomas Kluyver
|
r3458 | |||
# We do this unconditionally, so that the test suite doesn't import | ||||
# gtk, changing the default encoding and masking some unicode bugs. | ||||
exclusions.append(ipjoin('lib', 'inputhookgtk')) | ||||
Brian Granger
|
r2083 | |||
Brian Granger
|
r2079 | # These have to be skipped on win32 because the use echo, rm, cd, etc. | ||
Thomas Kluyver
|
r3917 | # See ticket https://github.com/ipython/ipython/issues/87 | ||
Brian Granger
|
r2079 | if sys.platform == 'win32': | ||
Fernando Perez
|
r2454 | exclusions.append(ipjoin('testing', 'plugin', 'test_exampleip')) | ||
exclusions.append(ipjoin('testing', 'plugin', 'dtexample')) | ||||
Brian Granger
|
r2079 | |||
Fernando Perez
|
r2496 | if not have['pexpect']: | ||
Fernando Perez
|
r2455 | exclusions.extend([ipjoin('scripts', 'irunner'), | ||
MinRK
|
r3761 | ipjoin('lib', 'irunner'), | ||
ipjoin('lib', 'tests', 'test_irunner')]) | ||||
Brian Granger
|
r2079 | |||
MinRK
|
r3640 | if not have['zmq']: | ||
exclusions.append(ipjoin('zmq')) | ||||
MinRK
|
r3761 | exclusions.append(ipjoin('frontend', 'qt')) | ||
MinRK
|
r3672 | exclusions.append(ipjoin('parallel')) | ||
MinRK
|
r3761 | elif not have['qt']: | ||
exclusions.append(ipjoin('frontend', 'qt')) | ||||
MinRK
|
r3674 | |||
if not have['pymongo']: | ||||
exclusions.append(ipjoin('parallel', 'controller', 'mongodb')) | ||||
MinRK
|
r3875 | exclusions.append(ipjoin('parallel', 'tests', 'test_mongodb')) | ||
MinRK
|
r3640 | |||
Fernando Perez
|
r3731 | if not have['matplotlib']: | ||
exclusions.extend([ipjoin('lib', 'pylabtools'), | ||||
MinRK
|
r3764 | ipjoin('lib', 'tests', 'test_pylabtools')]) | ||
Fernando Perez
|
r3731 | |||
Brian Granger
|
r2079 | # This is needed for the reg-exp to match on win32 in the ipdoctest plugin. | ||
if sys.platform == 'win32': | ||||
Fernando Perez
|
r2414 | exclusions = [s.replace('\\','\\\\') for s in exclusions] | ||
Brian Granger
|
r2079 | |||
Fernando Perez
|
r2414 | return exclusions | ||
Administrator
|
r1980 | |||
Fernando Perez
|
r2398 | class IPTester(object): | ||
"""Call that calls iptest or trial in a subprocess. | ||||
""" | ||||
Fernando Perez
|
r2399 | #: string, name of test runner that will be called | ||
runner = None | ||||
#: list, parameters for test runner | ||||
params = None | ||||
#: list, arguments of system call to be made to call test runner | ||||
call_args = None | ||||
#: list, process ids of subprocesses we start (for cleanup) | ||||
pids = None | ||||
Fernando Perez
|
r2480 | def __init__(self, runner='iptest', params=None): | ||
Fernando Perez
|
r2399 | """Create new test runner.""" | ||
Fernando Perez
|
r2494 | p = os.path | ||
Fernando Perez
|
r2398 | if runner == 'iptest': | ||
Brian Granger
|
r2507 | iptest_app = get_ipython_module_path('IPython.testing.iptest') | ||
self.runner = pycmd2argv(iptest_app) + sys.argv[1:] | ||||
Brian Granger
|
r2512 | else: | ||
raise Exception('Not a valid test runner: %s' % repr(runner)) | ||||
Fernando Perez
|
r2398 | if params is None: | ||
params = [] | ||||
Fernando Perez
|
r2480 | if isinstance(params, str): | ||
Fernando Perez
|
r2398 | params = [params] | ||
self.params = params | ||||
# Assemble call | ||||
self.call_args = self.runner+self.params | ||||
Fernando Perez
|
r2399 | # Store pids of anything we start to clean up on deletion, if possible | ||
# (on posix only, since win32 has no os.kill) | ||||
self.pids = [] | ||||
Fernando Perez
|
r2398 | if sys.platform == 'win32': | ||
def _run_cmd(self): | ||||
# On Windows, use os.system instead of subprocess.call, because I | ||||
# was having problems with subprocess and I just don't know enough | ||||
# about win32 to debug this reliably. Os.system may be the 'old | ||||
# fashioned' way to do it, but it works just fine. If someone | ||||
# later can clean this up that's fine, as long as the tests run | ||||
# reliably in win32. | ||||
Brian Granger
|
r2512 | # What types of problems are you having. They may be related to | ||
# running Python in unboffered mode. BG. | ||||
Fernando Perez
|
r2398 | return os.system(' '.join(self.call_args)) | ||
else: | ||||
def _run_cmd(self): | ||||
Brian Granger
|
r2781 | # print >> sys.stderr, '*** CMD:', ' '.join(self.call_args) # dbg | ||
Fernando Perez
|
r2399 | subp = subprocess.Popen(self.call_args) | ||
self.pids.append(subp.pid) | ||||
# If this fails, the pid will be left in self.pids and cleaned up | ||||
# later, but if the wait call succeeds, then we can clear the | ||||
# stored pid. | ||||
retcode = subp.wait() | ||||
self.pids.pop() | ||||
return retcode | ||||
Fernando Perez
|
r2398 | |||
def run(self): | ||||
"""Run the stored commands""" | ||||
try: | ||||
return self._run_cmd() | ||||
except: | ||||
import traceback | ||||
traceback.print_exc() | ||||
return 1 # signal failure | ||||
Fernando Perez
|
r2399 | def __del__(self): | ||
"""Cleanup on exit by killing any leftover processes.""" | ||||
if not hasattr(os, 'kill'): | ||||
return | ||||
for pid in self.pids: | ||||
try: | ||||
print 'Cleaning stale PID:', pid | ||||
os.kill(pid, signal.SIGKILL) | ||||
except OSError: | ||||
# This is just a best effort, if we fail or the process was | ||||
# really gone, ignore it. | ||||
Fernando Perez
|
r2400 | pass | ||
Fernando Perez
|
r2398 | |||
def make_runners(): | ||||
"""Define the top-level packages that need to be tested. | ||||
""" | ||||
Fernando Perez
|
r2483 | # Packages to be tested via nose, that only depend on the stdlib | ||
nose_pkg_names = ['config', 'core', 'extensions', 'frontend', 'lib', | ||||
Brian E. Granger
|
r4402 | 'scripts', 'testing', 'utils', 'nbformat' ] | ||
MinRK
|
r3673 | |||
if have['zmq']: | ||||
nose_pkg_names.append('parallel') | ||||
Fernando Perez
|
r2483 | # For debugging this code, only load quick stuff | ||
Fernando Perez
|
r2494 | #nose_pkg_names = ['core', 'extensions'] # dbg | ||
Fernando Perez
|
r2483 | |||
# Make fully qualified package names prepending 'IPython.' to our name lists | ||||
nose_packages = ['IPython.%s' % m for m in nose_pkg_names ] | ||||
# Make runners | ||||
runners = [ (v, IPTester('iptest', params=v)) for v in nose_packages ] | ||||
Fernando Perez
|
r2398 | return runners | ||
Brian Granger
|
r1972 | def run_iptest(): | ||
"""Run the IPython test suite using nose. | ||||
This function is called when this script is **not** called with the form | ||||
`iptest all`. It simply calls nose with appropriate command line flags | ||||
and accepts all of the standard nose arguments. | ||||
Fernando Perez
|
r1574 | """ | ||
warnings.filterwarnings('ignore', | ||||
'This will be removed soon. Use IPython.testing.util instead') | ||||
Fernando Perez
|
r2487 | argv = sys.argv + [ '--detailed-errors', # extra info in tracebacks | ||
Fernando Perez
|
r2494 | |||
Fernando Perez
|
r2414 | # Loading ipdoctest causes problems with Twisted, but | ||
# our test suite runner now separates things and runs | ||||
# all Twisted tests with trial. | ||||
'--with-ipdoctest', | ||||
'--ipdoctest-tests','--ipdoctest-extension=txt', | ||||
Fernando Perez
|
r1761 | # We add --exe because of setuptools' imbecility (it | ||
# blindly does chmod +x on ALL files). Nose does the | ||||
# right thing and it tries to avoid executables, | ||||
# setuptools unfortunately forces our hand here. This | ||||
# has been discussed on the distutils list and the | ||||
# setuptools devs refuse to fix this problem! | ||||
'--exe', | ||||
] | ||||
Fernando Perez
|
r1574 | |||
Fernando Perez
|
r2494 | if nose.__version__ >= '0.11': | ||
# I don't fully understand why we need this one, but depending on what | ||||
# directory the test suite is run from, if we don't give it, 0 tests | ||||
# get run. Specifically, if the test suite is run from the source dir | ||||
# with an argument (like 'iptest.py IPython.core', 0 tests are run, | ||||
# even if the same call done in this directory works fine). It appears | ||||
# that if the requested package is in the current dir, nose bails early | ||||
# by default. Since it's otherwise harmless, leave it in by default | ||||
# for nose >= 0.11, though unfortunately nose 0.10 doesn't support it. | ||||
argv.append('--traverse-namespace') | ||||
Fernando Perez
|
r1574 | |||
Matthew Brett
|
r4567 | # use our plugin for doctesting. It will remove the standard doctest plugin | ||
# if it finds it enabled | ||||
Paul Ivanov
|
r3511 | plugins = [IPythonDoctest(make_exclude()), KnownFailure()] | ||
Fernando Perez
|
r2414 | # We need a global ipython running in this process | ||
globalipapp.start_ipython() | ||||
# Now nose can run | ||||
Matthew Brett
|
r4567 | TestProgram(argv=argv, addplugins=plugins) | ||
Brian Granger
|
r1972 | |||
def run_iptestall(): | ||||
"""Run the entire IPython test suite by calling nose and trial. | ||||
This function constructs :class:`IPTester` instances for all IPython | ||||
modules and package and then runs each of them. This causes the modules | ||||
and packages of IPython to be tested each in their own subprocess using | ||||
nose or twisted.trial appropriately. | ||||
""" | ||||
Brian Granger
|
r2079 | |||
Brian Granger
|
r1972 | runners = make_runners() | ||
Brian Granger
|
r2079 | |||
Fernando Perez
|
r2398 | # Run the test runners in a temporary dir so we can nuke it when finished | ||
# to clean up any junk files left over by accident. This also makes it | ||||
# robust against being run in non-writeable directories by mistake, as the | ||||
# temp dir will always be user-writeable. | ||||
Jörgen Stenarson
|
r4208 | curdir = os.getcwdu() | ||
Fernando Perez
|
r2398 | testdir = tempfile.gettempdir() | ||
os.chdir(testdir) | ||||
Brian Granger
|
r1972 | # Run all test runners, tracking execution time | ||
Fernando Perez
|
r2480 | failed = [] | ||
Brian Granger
|
r1972 | t_start = time.time() | ||
Fernando Perez
|
r2398 | try: | ||
Fernando Perez
|
r2480 | for (name, runner) in runners: | ||
print '*'*70 | ||||
Fernando Perez
|
r2398 | print 'IPython test group:',name | ||
res = runner.run() | ||||
if res: | ||||
Fernando Perez
|
r2480 | failed.append( (name, runner) ) | ||
Fernando Perez
|
r2398 | finally: | ||
os.chdir(curdir) | ||||
Brian Granger
|
r1972 | t_end = time.time() | ||
t_tests = t_end - t_start | ||||
nrunners = len(runners) | ||||
nfail = len(failed) | ||||
# summarize results | ||||
Fernando Perez
|
r2480 | print '*'*70 | ||
Fernando Perez
|
r2496 | print 'Test suite completed for system with the following information:' | ||
print report() | ||||
Fernando Perez
|
r2091 | print 'Ran %s test groups in %.3fs' % (nrunners, t_tests) | ||
Brian Granger
|
r1972 | |||
Fernando Perez
|
r2496 | print 'Status:' | ||
Brian Granger
|
r1972 | if not failed: | ||
print 'OK' | ||||
else: | ||||
# If anything went wrong, point out what command to rerun manually to | ||||
# see the actual errors and individual summary | ||||
Fernando Perez
|
r2091 | print 'ERROR - %s out of %s test groups failed.' % (nfail, nrunners) | ||
Fernando Perez
|
r2480 | for name, failed_runner in failed: | ||
Brian Granger
|
r1972 | print '-'*40 | ||
print 'Runner failed:',name | ||||
print 'You may wish to rerun this one individually, with:' | ||||
print ' '.join(failed_runner.call_args) | ||||
Thomas Kluyver
|
r3895 | # Ensure that our exit code indicates failure | ||
sys.exit(1) | ||||
Brian Granger
|
r1972 | |||
def main(): | ||||
Fernando Perez
|
r2480 | for arg in sys.argv[1:]: | ||
if arg.startswith('IPython'): | ||||
Fernando Perez
|
r2481 | # This is in-process | ||
Brian Granger
|
r1990 | run_iptest() | ||
Fernando Perez
|
r2480 | else: | ||
Fernando Perez
|
r2481 | # This starts subprocesses | ||
Fernando Perez
|
r2480 | run_iptestall() | ||
Brian Granger
|
r1972 | |||
if __name__ == '__main__': | ||||
Fernando Perez
|
r2091 | main() | ||