moxiaonian
d9da4301cd
副本编号 LY-20261007-AC63（水印指纹已嵌入） 含 README、.gitignore；排除 .DS_Store / __pycache__ / .pytest_cache
70 lines
2.7 KiB
Python
70 lines
2.7 KiB
Python
"""子初换日（设计文档 §4.5 / §6.1 B 组）。
|
||
|
||
- 纯函数边界：真太阳时 22:59→当日 False，23:00→次日 True，
|
||
23:59→次日 True，00:00→当日 False
|
||
- 黄金样本：golden/time_boundary.yaml 中带 expect.is_late_zi 的条目，
|
||
走全链条 L0-a ~ L0-e（日界判定必须用真太阳时，不能用钟表时间）
|
||
"""
|
||
import sys
|
||
import unittest
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
|
||
import yaml
|
||
|
||
SKILL_ROOT = Path(__file__).resolve().parents[1]
|
||
SCRIPTS_DIR = SKILL_ROOT / "scripts"
|
||
GOLDEN_DIR = Path(__file__).resolve().parent / "golden"
|
||
if str(SCRIPTS_DIR) not in sys.path:
|
||
sys.path.insert(0, str(SCRIPTS_DIR))
|
||
|
||
from astro.day_boundary import resolve_calendar_day # noqa: E402
|
||
from astro.solar_time import to_true_solar_time # noqa: E402
|
||
from astro.timezone_resolve import to_utc # noqa: E402
|
||
from geo.cities import resolve_location # noqa: E402
|
||
|
||
# (真太阳时, 期望日历日, 期望 is_late_zi)
|
||
BOUNDARY_CASES = [
|
||
("1990-06-15 22:59", "1990-06-15", False),
|
||
("1990-06-15 23:00", "1990-06-16", True),
|
||
("1990-06-15 23:59", "1990-06-16", True),
|
||
("1990-06-16 00:00", "1990-06-16", False),
|
||
]
|
||
|
||
|
||
class DayBoundaryUnitTest(unittest.TestCase):
|
||
def test_boundary_cases(self):
|
||
for tst_str, day_str, expect_late in BOUNDARY_CASES:
|
||
with self.subTest(tst=tst_str):
|
||
tst = datetime.strptime(tst_str, "%Y-%m-%d %H:%M")
|
||
day, flag = resolve_calendar_day(tst)
|
||
self.assertEqual(day, datetime.strptime(day_str, "%Y-%m-%d"))
|
||
self.assertIs(flag["is_late_zi"], expect_late)
|
||
|
||
def test_day_is_midnight(self):
|
||
# 返回的日历日必须是 00:00:00
|
||
day, _flag = resolve_calendar_day(datetime(1990, 6, 15, 23, 30))
|
||
self.assertEqual((day.hour, day.minute, day.second), (0, 0, 0))
|
||
|
||
|
||
class LateZiGoldenTest(unittest.TestCase):
|
||
def test_late_zi_samples(self):
|
||
with open(GOLDEN_DIR / "time_boundary.yaml", encoding="utf-8") as f:
|
||
samples = [
|
||
s for s in yaml.safe_load(f)
|
||
if "is_late_zi" in s.get("expect", {})
|
||
]
|
||
self.assertGreaterEqual(len(samples), 2, "time_boundary.yaml B 组应有 2 条样本")
|
||
for s in samples:
|
||
with self.subTest(sample=s["name"], place=s["place"]):
|
||
loc = resolve_location(s["place"])
|
||
clock = datetime.strptime(s["birth"], "%Y-%m-%d %H:%M")
|
||
utc, _tz_audit = to_utc(clock, loc)
|
||
tst, _st_audit = to_true_solar_time(utc, loc.lon)
|
||
_day, flag = resolve_calendar_day(tst)
|
||
self.assertIs(flag["is_late_zi"], s["expect"]["is_late_zi"])
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|