from datetime import datetime

from scripts.kp_monthly_vimshottari import build_monthly_vimshottari_snapshot


def test_monthly_vimshottari_snapshot_returns_five_levels() -> None:
    snapshot = build_monthly_vimshottari_snapshot(
        birth_dt=datetime(1990, 1, 1, 12, 0),
        moon_lon=248.0,
        anchor_dt=datetime(2026, 1, 1, 12, 0),
    )

    assert set(snapshot["levels"].keys()) == {
        "mahadasha",
        "antardasha",
        "pratyantardasha",
        "sookshma",
        "prana",
    }
    assert snapshot["levels"]["mahadasha"]["lord"]
    assert snapshot["status"] == "parameter_sensitive"