732642856
79 lines
2.0 KiB
Python
79 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
def test_vimshottari_subperiod_timeline_expands_ad_to_pd() -> None:
|
|
completed = subprocess.run(
|
|
[
|
|
sys.executable,
|
|
"scripts/vimshottari_subperiod_timeline.py",
|
|
"--year",
|
|
"2000",
|
|
"--month",
|
|
"1",
|
|
"--day",
|
|
"1",
|
|
"--hour",
|
|
"12",
|
|
"--minute",
|
|
"0",
|
|
"--lat",
|
|
"0.0",
|
|
"--lon",
|
|
"0.0",
|
|
"--tz",
|
|
"0",
|
|
"--years",
|
|
"30",
|
|
],
|
|
cwd=ROOT,
|
|
text=True,
|
|
capture_output=True,
|
|
timeout=180,
|
|
check=False,
|
|
)
|
|
assert completed.returncode == 0, completed.stderr or completed.stdout
|
|
report = json.loads(completed.stdout)
|
|
first_ad = report["timeline"][0]["antardasha"][0]
|
|
assert len(first_ad["pratyantar"]) == 9
|
|
assert first_ad["pratyantar"][0]["start"] == first_ad["start"]
|
|
assert first_ad["pratyantar"][-1]["end"] == first_ad["end"]
|
|
|
|
|
|
def test_daily_transit_window_scan_runs_one_day_json() -> None:
|
|
completed = subprocess.run(
|
|
[
|
|
sys.executable,
|
|
"scripts/daily_transit_window_scan.py",
|
|
"--start",
|
|
"2027-03-01",
|
|
"--end",
|
|
"2027-03-01",
|
|
"--planets",
|
|
"Jupiter,Saturn",
|
|
"--tz",
|
|
"0",
|
|
],
|
|
cwd=ROOT,
|
|
text=True,
|
|
capture_output=True,
|
|
timeout=120,
|
|
check=False,
|
|
)
|
|
assert completed.returncode == 0, completed.stderr or completed.stdout
|
|
report = json.loads(completed.stdout)
|
|
assert report["days"] == 1
|
|
assert report["rows"][0]["date"] == "2027-03-01"
|
|
planets = report["rows"][0]["transit"]["planets"]
|
|
if isinstance(planets, dict):
|
|
assert "Jupiter" in planets
|
|
else:
|
|
assert any(item.get("planet") == "Jupiter" for item in planets)
|