xref: /llvm-project/mlir/test/Unit/lit.cfg.py (revision f9008e6366c2496b1ca1785b891d5578174ad63e)
1# -*- Python -*-
2
3# Configuration file for the 'lit' test runner.
4
5import os
6import subprocess
7
8import lit.formats
9
10# name: The name of this test suite.
11config.name = "MLIR-Unit"
12
13# suffixes: A list of file extensions to treat as test files.
14config.suffixes = []
15
16# test_source_root: The root path where tests are located.
17# test_exec_root: The root path where tests should be run.
18config.test_exec_root = os.path.join(config.mlir_obj_root, "unittests")
19config.test_source_root = config.test_exec_root
20
21# testFormat: The test format to use to interpret tests.
22config.test_format = lit.formats.GoogleTest(config.llvm_build_mode, "Tests")
23
24# Propagate the temp directory. Windows requires this because it uses \Windows\
25# if none of these are present.
26if "TMP" in os.environ:
27    config.environment["TMP"] = os.environ["TMP"]
28if "TEMP" in os.environ:
29    config.environment["TEMP"] = os.environ["TEMP"]
30
31# Propagate HOME as it can be used to override incorrect homedir in passwd
32# that causes the tests to fail.
33if "HOME" in os.environ:
34    config.environment["HOME"] = os.environ["HOME"]
35
36# Propagate sanitizer options.
37for var in [
38    "ASAN_SYMBOLIZER_PATH",
39    "HWASAN_SYMBOLIZER_PATH",
40    "MSAN_SYMBOLIZER_PATH",
41    "TSAN_SYMBOLIZER_PATH",
42    "UBSAN_SYMBOLIZER_PATH",
43    "ASAN_OPTIONS",
44    "HWASAN_OPTIONS",
45    "MSAN_OPTIONS",
46    "TSAN_OPTIONS",
47    "UBSAN_OPTIONS",
48]:
49    if var in os.environ:
50        config.environment[var] = os.environ[var]
51