1# -*- Python -*-
2
3import os
4import sys
5
6import lit.formats
7
8# Configuration file for the 'lit' test runner.
9
10# name: The name of this test suite.
11config.name = 'lit'
12
13# testFormat: The test format to use to interpret tests.
14config.test_format = lit.formats.ShTest(execute_external=False)
15
16# suffixes: A list of file extensions to treat as test files.
17config.suffixes = ['.py']
18
19# excludes: A list of individual files to exclude.
20config.excludes = ['Inputs']
21
22# test_source_root: The root path where tests are located.
23config.test_source_root = os.path.dirname(__file__)
24config.test_exec_root = config.test_source_root
25
26config.target_triple = '(unused)'
27
28src_root = os.path.join(config.test_source_root, '..')
29llvm_src_root = getattr(config, 'llvm_src_root', None)
30if llvm_src_root != None:
31  # ``src_root`` may be in LLVM's binary build directory which does not contain
32  # ``lit.py``, so use `llvm_src_root` instead.
33  lit_path = os.path.join(llvm_src_root, 'utils', 'lit')
34else:
35  lit_path = src_root
36
37config.environment['PYTHONPATH'] = lit_path # Required because some tests import the lit module
38config.substitutions.append(('%{src_root}', src_root))
39config.substitutions.append(('%{inputs}', os.path.join(
40            src_root, 'tests', 'Inputs')))
41config.substitutions.append(('%{lit}', "%%{python} %s" % (
42            os.path.join(lit_path, 'lit.py'),)))
43config.substitutions.append(('%{python}', '"%s"' % (sys.executable)))
44
45
46# Enable coverage.py reporting, assuming the coverage module has been installed
47# and sitecustomize.py in the virtualenv has been modified appropriately.
48if lit_config.params.get('check-coverage', None):
49    config.environment['COVERAGE_PROCESS_START'] = os.path.join(
50        os.path.dirname(__file__), ".coveragerc")
51
52# Add a feature to detect the Python version.
53config.available_features.add("python%d.%d" % (sys.version_info[0],
54                                                  sys.version_info[1]))
55
56# Add a feature to detect if psutil is available
57try:
58    import psutil
59    lit_config.note('Found python psutil module')
60    config.available_features.add("python-psutil")
61except ImportError:
62    lit_config.warning('Could not import psutil. Some tests will be skipped and'
63                       ' the --timeout command line argument will not work.')
64
65if sys.platform.startswith('win') or sys.platform.startswith('cygwin'):
66    config.available_features.add('windows')
67
68# Add llvm and lit tools directories if this config is being loaded indirectly.
69path = config.environment['PATH']
70for attribute in ('llvm_tools_dir', 'lit_tools_dir'):
71    directory = getattr(config, attribute, None)
72    if directory:
73        path = os.path.pathsep.join((directory, path))
74config.environment['PATH'] = path
75