import pytest
from scripts.shadbala import _ayana_bala


@pytest.mark.parametrize("planet", ["Sun", "Moon", "Mars", "Mercury", "Jupiter", "Venus", "Saturn"])
def test_ayana_uses_wrapped_tropical_hemisphere(planet):
    assert _ayana_bala(planet, 350.0, 22.0) == _ayana_bala(planet, -10.0, 22.0)


def test_venus_just_after_tropical_aries_uses_north_addition():
    bala, declination = _ayana_bala("Venus", 350.0, 22.0)
    assert declination > 0
    assert bala > 30