#!/usr/bin/env python

#

# This allows running the matplotlib tests from the command line: e.g.

#

# $ python tests.py -v -d

#

# The arguments are identical to the arguments accepted by nosetests.

#

# See https://nose.readthedocs.org/ for a detailed description of

# these options.

import sys

import argparse

if __name__ == '__main__':

try:

import setuptools

except ImportError:

pass

# The warnings need to be before any of matplotlib imports, but after

# setuptools (if present) which has syntax error with the warnings enabled.

# Filtering by module does not work as this will be raised by Python itself.

# so `module=matplotlib.*` is out of questions.

import warnings

# Python 3.6 deprecate invalid character-pairs \A, \* ... in non raw-strings

# and other things. Let's not re-introduce them

warnings.filterwarnings('error', '.*invalid escape sequence.*',

category=DeprecationWarning)

warnings.filterwarnings(

'default',

'.*inspect.getargspec\(\) is deprecated.*',

category=DeprecationWarning)

from matplotlib import test

parser = argparse.ArgumentParser(add_help=False)

parser.add_argument('--no-pep8', action='store_true',

help='Run all tests except PEP8 testing')

parser.add_argument('--pep8', action='store_true',

help='Run only PEP8 testing')

parser.add_argument('--no-network', action='store_true',

help='Run tests without network connection')

parser.add_argument('-j', type=int,

help='Shortcut for specifying number of test processes')

parser.add_argument('--recursionlimit', type=int, default=0,

help='Specify recursionlimit for test run')

args, extra_args = parser.parse_known_args()

if args.no_network:

from matplotlib.testing import disable_internet

disable_internet.turn_off_internet()

extra_args.extend(['-a', '!network'])

if args.j:

extra_args.extend([

'--processes={}'.format(args.j),

'--process-timeout=300'

])

print('Python byte-compilation optimization level: %d' % sys.flags.optimize)

success = test(argv=sys.argv[0:1] + extra_args, switch_backend_warn=False,

recursionlimit=args.recursionlimit)

sys.exit(not success)