diff --git a/tests.py b/tests.py index 94b0b1bb8d70..722c50a2c7cd 100755 --- 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)