Refactored tests.py with argparse

bcongdon · bcongdon · commit 57068760c9e8 · 2016-11-08T10:09:48.000-06:00
diff --git a/tests.py b/tests.py
@@ -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)
 
