| # -*- Python -*- |
| import os |
| import sys |
| |
| import lit.formats |
| |
| config.name = "per_test_timeout" |
| |
| shellType = lit_config.params.get("external", "1") |
| |
| if shellType == "0": |
| lit_config.note("Using internal shell") |
| externalShell = False |
| else: |
| lit_config.note("Using external shell") |
| externalShell = True |
| |
| configSetTimeout = lit_config.params.get("set_timeout", "0") |
| |
| if configSetTimeout != "0": |
| # Try setting the max individual test time in the configuration |
| lit_config.maxIndividualTestTime = int(configSetTimeout) |
| |
| config.test_format = lit.formats.ShTest(execute_external=externalShell) |
| config.suffixes = [".py"] |
| |
| config.test_source_root = os.path.dirname(__file__) |
| config.test_exec_root = config.test_source_root |
| config.target_triple = "(unused)" |
| src_root = os.path.join(config.test_source_root, "..") |
| |
| pythonpath_list = [src_root] |
| # Ensure the user's PYTHONPATH is included. |
| if "PYTHONPATH" in os.environ: |
| pythonpath_list.append(os.environ["PYTHONPATH"]) |
| if "PYTHONPATH" in config.environment: |
| pythonpath_list.append(config.environment["PYTHONPATH"]) |
| config.environment["PYTHONPATH"] = os.pathsep.join(pythonpath_list) |
| |
| config.substitutions.append(("%{python}", '"%s"' % (sys.executable))) |