from datetime import datetime

from scripts.yogini_dasha import calculate_yogini_dasha


def test_yogini_uses_moon_nakshatra_plus_three_seed_when_available() -> None:
    result = calculate_yogini_dasha({
        'lagna_rashi_index': 0,
        'moon_nakshatra_index': 0,
        'birth_datetime': datetime(1990, 1, 1),
    })
    assert result['starting_yogini'] == 'Rati'
    assert result['starting_planet'] == 'Mercury'
    assert result['seed_method'] == 'moon_nakshatra_plus_3_mod_8'


def test_yogini_keeps_legacy_seed_only_without_moon_nakshatra() -> None:
    result = calculate_yogini_dasha({
        'lagna_rashi_index': 0,
        'birth_datetime': datetime(1990, 1, 1),
    })
    assert result['starting_yogini'] == 'Maya'
    assert result['seed_method'] == 'legacy_lagna_rashi_compatibility'