732642856
132 lines
4.3 KiB
Python
132 lines
4.3 KiB
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
SCRIPTS = Path(__file__).resolve().parents[1] / "scripts"
|
|
if str(SCRIPTS) not in sys.path:
|
|
sys.path.insert(0, str(SCRIPTS))
|
|
|
|
import domain_calculation_service as calculation_service # noqa: E402
|
|
import jyotish_api_server # noqa: E402
|
|
from jyotish_api_server import JyotishAPIHandler # noqa: E402
|
|
from jyotish_engine import _compute_chart_from_args # noqa: E402
|
|
|
|
BIRTH = {
|
|
"year": 1990,
|
|
"month": 1,
|
|
"day": 1,
|
|
"hour": 12,
|
|
"minute": 0,
|
|
"second": 0,
|
|
"lat": 28.6139,
|
|
"lon": 77.2090,
|
|
"tz": 5.5,
|
|
"ayanamsa": "lahiri",
|
|
}
|
|
|
|
|
|
def test_true_node_changes_effective_rahu_and_contract() -> None:
|
|
mean = calculation_service.compute_chart({**BIRTH, "node_mode": "mean"})
|
|
true = calculation_service.compute_chart({**BIRTH, "node_mode": "true"})
|
|
|
|
assert mean["planets"]["Rahu"]["lon"] != pytest.approx(
|
|
true["planets"]["Rahu"]["lon"], abs=1e-8
|
|
)
|
|
assert mean["calculation_contract"]["effective"]["node_mode"] == "mean"
|
|
assert true["calculation_contract"]["effective"]["node_mode"] == "true"
|
|
assert mean["result_hash"] != true["result_hash"]
|
|
|
|
|
|
def test_vimshottari_uses_birth_balance_as_canonical_timeline() -> None:
|
|
birth_dt = datetime(1990, 1, 1, 12, 0)
|
|
result = calculation_service.compute_vimshottari_timeline(
|
|
birth_dt=birth_dt,
|
|
moon_lon=100.0,
|
|
current_date=birth_dt,
|
|
)
|
|
|
|
first = result["periods"][0]
|
|
assert first["lord"] == "Saturn"
|
|
assert first["start"] == "1980-07-02"
|
|
assert first["end"] == "1999-07-02"
|
|
assert result["birth_balance"]["remaining_years"] == pytest.approx(9.5)
|
|
assert result["calculation_contract"]["algorithm"] == "vimshottari_birth_balance"
|
|
|
|
|
|
def test_sade_sati_uses_real_saturn_transit_for_reference_date() -> None:
|
|
result = calculation_service.compute_sade_sati(
|
|
moon_degree=300.0,
|
|
asc_degree=330.0,
|
|
reference_date="2026-07-11",
|
|
tz=5.5,
|
|
ayanamsa="lahiri",
|
|
)
|
|
oracle = calculation_service.compute_transit_longitude(
|
|
planet="Saturn",
|
|
reference_date="2026-07-11",
|
|
tz=5.5,
|
|
ayanamsa="lahiri",
|
|
)
|
|
|
|
assert result["transit_saturn_lon"] == pytest.approx(oracle["longitude"], abs=1e-8)
|
|
assert result["provenance"]["data_layer"] == "true_transit_positions"
|
|
assert result["provenance"]["reference_date"] == "2026-07-11"
|
|
|
|
|
|
def test_jupiter_transit_uses_the_canonical_transit_contract() -> None:
|
|
result = calculation_service.compute_transit_longitude(
|
|
planet="Jupiter",
|
|
reference_date="2026-07-11",
|
|
tz=5.5,
|
|
ayanamsa="lahiri",
|
|
)
|
|
|
|
assert result["planet"] == "Jupiter"
|
|
assert 0 <= result["longitude"] < 360
|
|
assert result["data_layer"] == "true_transit_positions"
|
|
assert result["ayanamsa"] == "lahiri"
|
|
|
|
|
|
def test_timezone_inference_fails_closed(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(
|
|
calculation_service,
|
|
"_lookup_timezone_name",
|
|
lambda _lat, _lon: None,
|
|
)
|
|
|
|
with pytest.raises(calculation_service.TimezoneInferenceError, match="timezone inference"):
|
|
calculation_service.infer_timezone_offset(
|
|
lat=0.0,
|
|
lon=0.0,
|
|
local_datetime=datetime(1990, 1, 1, 12, 0),
|
|
)
|
|
|
|
|
|
def test_chart_hash_matches_domain_cli_and_rest(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
monkeypatch.setenv("JYOTISH_API_CHART_CACHE_TTL_SECONDS", "0")
|
|
monkeypatch.setenv("VEDASTRO_ENABLE_NETWORK", "0")
|
|
monkeypatch.setattr(
|
|
jyotish_api_server,
|
|
"_attach_vedastro_main_entry_overview",
|
|
lambda result, _birth: result,
|
|
)
|
|
expected = calculation_service.compute_chart({**BIRTH, "node_mode": "true"})
|
|
cli, _asc_idx, _jd, _ayanamsa = _compute_chart_from_args(
|
|
SimpleNamespace(**BIRTH, node_mode="true")
|
|
)
|
|
rest = JyotishAPIHandler.__new__(JyotishAPIHandler)._compute_chart_sync(
|
|
{**BIRTH, "node_mode": "true", "transit_date": "2026-07-11"}
|
|
)
|
|
|
|
assert cli["result_hash"] == expected["result_hash"]
|
|
assert rest["result_hash"] == expected["result_hash"]
|
|
assert rest["birth"]["node_mode"] == "true"
|
|
assert rest["calculation_contract"]["effective"]["node_mode"] == "true"
|