MAINT: add ability to specify recursionlimit

matthew-brett · matthew-brett · commit 3434eb5076c2 · 2017-01-16T15:39:59.000-08:00
We're getting a test failure on OSX and Python 3.6 where parsing reaches the recursion limit: #7799 Upping the recursionlimit resolves the error. Add command line argument to tests to up the recursion limit for the test run.
diff --git a/tests.py b/tests.py
@@ -12,6 +12,7 @@
 import os
 import sys
 import time
+import argparse
 
 import matplotlib
 matplotlib.use('agg')
@@ -50,5 +51,11 @@ def run(extra_args):
         disable_internet.turn_off_internet()
         extra_args.extend(['-a', '!network'])
         sys.argv.remove('--no-network')
-
+    # Get recursion limit
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--recursionlimit', type=int, default=0,
+                        help='Specify recursionlimit for test run')
+    found_args, sys.argv = parser.parse_known_args(sys.argv)
+    if found_args.recursionlimit:
+        sys.setrecursionlimit(found_args.recursionlimit)
     run(extra_args)
