PageIndex/tests/test_config.py

# tests/test_config.py
import pytest
from pageindex.config import IndexConfig


def test_defaults():
    config = IndexConfig()
    assert config.model == "gpt-4o-2024-11-20"
    assert config.retrieve_model is None
    assert config.toc_check_page_num == 20


def test_overrides():
    config = IndexConfig(model="gpt-5.4", retrieve_model="claude-sonnet")
    assert config.model == "gpt-5.4"
    assert config.retrieve_model == "claude-sonnet"


def test_unknown_key_raises():
    with pytest.raises(Exception):
        IndexConfig(nonexistent_key="value")


def test_model_copy_with_update():
    config = IndexConfig(toc_check_page_num=30)
    updated = config.model_copy(update={"model": "gpt-5.4"})
    assert updated.model == "gpt-5.4"
    assert updated.toc_check_page_num == 30


def test_legacy_yes_no_strings_coerce_to_bool():
    """Legacy page_index()/run_pageindex callers pass 'yes'/'no' strings;
    pydantic must coerce them to the booleans the pipeline now branches on."""
    config = IndexConfig(if_add_node_id="yes", if_add_node_summary="no")
    assert config.if_add_node_id is True
    assert config.if_add_node_summary is False


def test_llm_params_field_defaults_to_none():
    assert IndexConfig().llm_params is None
    assert IndexConfig(llm_params={"temperature": 1}).llm_params == {"temperature": 1}