1# -*- Python -*- 2import os 3import sys 4 5import lit.formats 6 7config.name = "per_test_timeout" 8 9shellType = lit_config.params.get("external", "1") 10 11if shellType == "0": 12 lit_config.note("Using internal shell") 13 externalShell = False 14else: 15 lit_config.note("Using external shell") 16 externalShell = True 17 18configSetTimeout = lit_config.params.get("set_timeout", "0") 19 20if configSetTimeout != "0": 21 # Try setting the max individual test time in the configuration 22 lit_config.maxIndividualTestTime = int(configSetTimeout) 23 24config.test_format = lit.formats.ShTest(execute_external=externalShell) 25config.suffixes = [".py"] 26 27config.test_source_root = os.path.dirname(__file__) 28config.test_exec_root = config.test_source_root 29config.target_triple = "(unused)" 30src_root = os.path.join(config.test_source_root, "..") 31 32pythonpath_list = [src_root] 33# Ensure the user's PYTHONPATH is included. 34if "PYTHONPATH" in os.environ: 35 pythonpath_list.append(os.environ["PYTHONPATH"]) 36if "PYTHONPATH" in config.environment: 37 pythonpath_list.append(config.environment["PYTHONPATH"]) 38config.environment["PYTHONPATH"] = os.pathsep.join(pythonpath_list) 39 40config.substitutions.append(("%{python}", '"%s"' % (sys.executable))) 41