jason/mempalace
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: use tmp_path for embedding device config tests

Browse Source 
Agent-Logs-Url: https://github.com/MemPalace/mempalace/sessions/3213a67a-6871-4bb2-9ae0-23fa11001a22

Co-authored-by: igorls <4753812+igorls@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-04-24 23:08:26 +00:00
committed by GitHub
parent fbd0904799
commit 25c885ae0b
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_config.py
+6 -8
View File
@@ -25,22 +25,20 @@ def test_embedding_device_defaults_to_auto():
assert cfg.embedding_device == "auto" assert cfg.embedding_device == "auto"
def test_embedding_device_from_config_is_normalized(): def test_embedding_device_from_config_is_normalized(tmp_path):
tmpdir = tempfile.mkdtemp() with open(tmp_path / "config.json", "w") as f:
with open(os.path.join(tmpdir, "config.json"), "w") as f:
json.dump({"embedding_device": " CUDA "}, f) json.dump({"embedding_device": " CUDA "}, f)
cfg = MempalaceConfig(config_dir=tmpdir) cfg = MempalaceConfig(config_dir=str(tmp_path))
assert cfg.embedding_device == "cuda" assert cfg.embedding_device == "cuda"
def test_embedding_device_env_overrides_config(): def test_embedding_device_env_overrides_config(tmp_path):
tmpdir = tempfile.mkdtemp() with open(tmp_path / "config.json", "w") as f:
with open(os.path.join(tmpdir, "config.json"), "w") as f:
json.dump({"embedding_device": "cpu"}, f) json.dump({"embedding_device": "cpu"}, f)
os.environ["MEMPALACE_EMBEDDING_DEVICE"] = " CoreML " os.environ["MEMPALACE_EMBEDDING_DEVICE"] = " CoreML "
try: try:
cfg = MempalaceConfig(config_dir=tmpdir) cfg = MempalaceConfig(config_dir=str(tmp_path))
assert cfg.embedding_device == "coreml" assert cfg.embedding_device == "coreml"
finally: finally:
del os.environ["MEMPALACE_EMBEDDING_DEVICE"] del os.environ["MEMPALACE_EMBEDDING_DEVICE"]