tests/test_config.py

from pathlib import Path

import pytest

from vascx_models.config import load_app_config


def test_load_app_config_accepts_aliases_and_colours(tmp_path: Path) -> None:
    config_path = tmp_path / "config.yaml"
    config_path.write_text(
        "\n".join(
            [
                "overlay:",
                "  enabled: false",
                "  layers:",
                "    artery: false",
                "  colours:",
                "    veins: '#112233'",
                "  circles:",
                "    - name: inner",
                "      diameter: 1.5",
                "    - name: outer",
                "      diameter: 2.5",
                "      colour: [4, 5, 6]",
            ]
        ),
        encoding="utf-8",
    )

    app_config = load_app_config(config_path)

    assert app_config.source_path == config_path
    assert app_config.overlay.enabled is False
    assert app_config.overlay.layers.arteries is False
    assert app_config.overlay.layers.vessel_widths is True
    assert app_config.overlay.colors.vein == (17, 34, 51)
    assert app_config.overlay.colors.vessel == (255, 255, 255)
    assert app_config.overlay.colors.vessel_width == (0, 0, 0)
    assert [circle.name for circle in app_config.overlay.circles] == ["inner", "outer"]
    assert [circle.diameter for circle in app_config.overlay.circles] == [1.5, 2.5]
    assert app_config.overlay.circles[0].color == (0, 0, 0)
    assert app_config.overlay.circles[1].color == (4, 5, 6)
    assert app_config.vessel_widths.samples_per_connection == 5


def test_load_app_config_rejects_unknown_layer(tmp_path: Path) -> None:
    config_path = tmp_path / "config.yaml"
    config_path.write_text(
        "\n".join(
            [
                "overlay:",
                "  layers:",
                "    unknown_layer: true",
            ]
        ),
        encoding="utf-8",
    )

    with pytest.raises(ValueError, match="Unsupported overlay layer"):
        load_app_config(config_path)


def test_load_app_config_rejects_duplicate_circle_names(tmp_path: Path) -> None:
    config_path = tmp_path / "config.yaml"
    config_path.write_text(
        "\n".join(
            [
                "overlay:",
                "  circles:",
                "    - name: repeated",
                "      diameter: 2.0",
                "    - name: repeated",
                "      diameter: 3.0",
            ]
        ),
        encoding="utf-8",
    )

    with pytest.raises(ValueError, match="Duplicate circle name"):
        load_app_config(config_path)


def test_load_app_config_accepts_vessel_width_sampling_options(tmp_path: Path) -> None:
    config_path = tmp_path / "config.yaml"
    config_path.write_text(
        "\n".join(
            [
                "overlay:",
                "  layers:",
                "    vessel_widths: false",
                "  colours:",
                "    vessel: '#0A0B0C'",
                "    vessel_widths: [1, 2, 3]",
                "vessel_widths:",
                "  inner_circle: 2r",
                "  outer_circle: 3r",
                "  samples_per_connection: 4",
            ]
        ),
        encoding="utf-8",
    )

    app_config = load_app_config(config_path)

    assert app_config.overlay.layers.vessel_widths is False
    assert app_config.overlay.colors.vessel == (10, 11, 12)
    assert app_config.overlay.colors.vessel_width == (1, 2, 3)
    assert app_config.vessel_widths.inner_circle == "2r"
    assert app_config.vessel_widths.outer_circle == "3r"
    assert app_config.vessel_widths.samples_per_connection == 4


def test_load_app_config_rejects_invalid_vessel_width_samples_per_connection(tmp_path: Path) -> None:
    config_path = tmp_path / "config.yaml"
    config_path.write_text(
        "\n".join(
            [
                "vessel_widths:",
                "  samples_per_connection: 0",
            ]
        ),
        encoding="utf-8",
    )

    with pytest.raises(ValueError, match="vessel_widths.samples_per_connection"):
        load_app_config(config_path)