mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-22 06:50:25 +08:00
62 lines
2.4 KiB
Python
62 lines
2.4 KiB
Python
import os
|
|
|
|
import tilelang
|
|
|
|
|
|
def _env_var_descriptor(name):
|
|
return type(tilelang.env).__dict__[name]
|
|
|
|
|
|
def _restore_forced_value(name, value):
|
|
_env_var_descriptor(name)._forced_value = value
|
|
|
|
|
|
def test_env_var(monkeypatch):
|
|
desc = _env_var_descriptor("TILELANG_PRINT_ON_COMPILATION")
|
|
original_forced_value = desc._forced_value
|
|
desc._forced_value = None
|
|
|
|
# test default value
|
|
try:
|
|
monkeypatch.delenv("TILELANG_PRINT_ON_COMPILATION", raising=False)
|
|
assert tilelang.env.TILELANG_PRINT_ON_COMPILATION == "1"
|
|
|
|
# test environment value
|
|
monkeypatch.setenv("TILELANG_PRINT_ON_COMPILATION", "0")
|
|
assert tilelang.env.TILELANG_PRINT_ON_COMPILATION == "0"
|
|
|
|
# test forced value with class method
|
|
tilelang.env.TILELANG_PRINT_ON_COMPILATION = "1"
|
|
assert tilelang.env.TILELANG_PRINT_ON_COMPILATION == "1"
|
|
finally:
|
|
_restore_forced_value("TILELANG_PRINT_ON_COMPILATION", original_forced_value)
|
|
|
|
|
|
def test_tilelang_tmp_dir_default_tracks_cache_dir(monkeypatch, tmp_path):
|
|
original_cache_forced_value = _env_var_descriptor("TILELANG_CACHE_DIR")._forced_value
|
|
original_tmp_forced_value = _env_var_descriptor("TILELANG_TMP_DIR")._forced_value
|
|
_restore_forced_value("TILELANG_CACHE_DIR", None)
|
|
_restore_forced_value("TILELANG_TMP_DIR", None)
|
|
|
|
try:
|
|
monkeypatch.delenv("TILELANG_CACHE_DIR", raising=False)
|
|
monkeypatch.delenv("TILELANG_TMP_DIR", raising=False)
|
|
|
|
monkeypatch.setenv("TILELANG_CACHE_DIR", str(tmp_path / "env_cache_a"))
|
|
assert os.path.join(str(tmp_path / "env_cache_a"), "tmp") == tilelang.env.TILELANG_TMP_DIR
|
|
|
|
monkeypatch.setenv("TILELANG_CACHE_DIR", str(tmp_path / "env_cache_b"))
|
|
assert os.path.join(str(tmp_path / "env_cache_b"), "tmp") == tilelang.env.TILELANG_TMP_DIR
|
|
|
|
tilelang.env.TILELANG_CACHE_DIR = str(tmp_path / "cache_a")
|
|
assert os.path.join(str(tmp_path / "cache_a"), "tmp") == tilelang.env.TILELANG_TMP_DIR
|
|
|
|
tilelang.env.TILELANG_CACHE_DIR = str(tmp_path / "cache_b")
|
|
assert os.path.join(str(tmp_path / "cache_b"), "tmp") == tilelang.env.TILELANG_TMP_DIR
|
|
|
|
monkeypatch.setenv("TILELANG_TMP_DIR", str(tmp_path / "explicit_tmp"))
|
|
assert str(tmp_path / "explicit_tmp") == tilelang.env.TILELANG_TMP_DIR
|
|
finally:
|
|
_restore_forced_value("TILELANG_CACHE_DIR", original_cache_forced_value)
|
|
_restore_forced_value("TILELANG_TMP_DIR", original_tmp_forced_value)
|