Skip to content

Commit

Permalink
Refactored tests.py with argparse
Browse files Browse the repository at this point in the history
  • Loading branch information
bcongdon committed Nov 8, 2016
1 parent ca8dcee commit 5706876
Showing 1 changed file with 14 additions and 7 deletions.
21 changes: 14 additions & 7 deletions tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,32 +10,39 @@
# these options.

import sys
import argparse


if __name__ == '__main__':
from matplotlib import default_test_modules, test

extra_args = []

if '--no-pep8' in sys.argv:
parser = argparse.ArgumentParser()
parser.add_argument('--no-pep8', action="store_true")
parser.add_argument('--pep8', action="store_true")
parser.add_argument('--no-network', action="store_true")
parser.add_argument('-j', type=int)
args = parser.parse_args()

if args.no_pep8:
default_test_modules.remove('matplotlib.tests.test_coding_standards')
sys.argv.remove('--no-pep8')
elif '--pep8' in sys.argv:
elif args.pep8:
default_test_modules[:] = ['matplotlib.tests.test_coding_standards']
sys.argv.remove('--pep8')
if '--no-network' in sys.argv:
if args.no_network:
from matplotlib.testing import disable_internet
disable_internet.turn_off_internet()
extra_args.extend(['-a', '!network'])
sys.argv.remove('--no-network')
if '-j' in sys.argv:
nproc = sys.argv[sys.argv.index('-j') + 1]
if args.j:
extra_args.extend([
'--processes={}'.format(int(nproc)),
'--processes={}'.format(args.j),
'--process-timeout=300'
])
sys.argv.pop(sys.argv.index('-j') + 1)
sys.argv.remove('-j')
sys.argv.remove(nproc)

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

Expand Down

0 comments on commit 5706876

Please sign in to comment.