| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162 |
- #!/usr/bin/env python3
- # -*- coding: utf-8 -*-
- """
- Config tests
- """
- import os
- from data_modules import config as config_module
- from data_modules.config import DataModulesConfig, get_config, set_project_root
- def test_config_paths_and_defaults(tmp_path):
- cfg = DataModulesConfig.from_project_root(tmp_path)
- assert cfg.project_root == tmp_path
- assert cfg.webnovel_dir.name == ".webnovel"
- assert cfg.state_file.name == "state.json"
- assert cfg.index_db.name == "index.db"
- assert cfg.rag_db.name == "rag.db"
- assert cfg.vector_db.name == "vectors.db"
- cfg.ensure_dirs()
- assert cfg.webnovel_dir.exists()
- def test_get_config_and_set_project_root(tmp_path):
- set_project_root(tmp_path)
- cfg = get_config()
- assert cfg.project_root == tmp_path
- def test_load_dotenv(monkeypatch, tmp_path):
- # prepare .env
- env_path = tmp_path / ".env"
- env_path.write_text("EMBED_BASE_URL=https://example.com\n", encoding="utf-8")
- monkeypatch.chdir(tmp_path)
- monkeypatch.delenv("EMBED_BASE_URL", raising=False)
- # call loader explicitly
- config_module._load_dotenv()
- assert os.environ.get("EMBED_BASE_URL") == "https://example.com"
- def test_config_default_context_template_weights_dynamic_is_available(tmp_path):
- cfg = DataModulesConfig.from_project_root(tmp_path)
- dynamic = cfg.context_template_weights_dynamic
- assert isinstance(dynamic, dict)
- assert "early" in dynamic
- assert "mid" in dynamic
- assert "late" in dynamic
- assert "plot" in dynamic["early"]
- def test_config_dynamic_template_weights_are_independent_instances(tmp_path):
- cfg1 = DataModulesConfig.from_project_root(tmp_path)
- cfg2 = DataModulesConfig.from_project_root(tmp_path)
- cfg1.context_template_weights_dynamic["early"]["plot"]["core"] = 0.77
- assert cfg2.context_template_weights_dynamic["early"]["plot"]["core"] != 0.77
|
