feat: Vimsopaka尊严映射修复 + 单元测试22项全绿 + 审计报告
## Vimsopaka 尊严等级映射 (Step 4.5) - 新增 _get_dignity_level() 函数：根据行星+星座+度数判断尊严等级 - 新增 MOOLATRIKONA 常量表（含度数范围） - 新增 PERMANENT_FRIENDS / PERMANENT_ENEMIES 关系表（BPHS标准） - Vimsopaka 集成不再默认 NEUTRAL，正确映射 Exalted/Moolatrikona/Own/Friend/Enemy/Debilitated ## Dasha 展示修正 - 修正之前的错误修复：日期计算必须用完整年数（数学等价性） - 第一个 MD 新增字段：balance_years（balance值）、full_years（完整值）、elapsed_at_birth - years 字段显示 balance 值（展示友好），full_years 显示完整值（用于链式计算） ## 单元测试 (tests/) - tests/test_dasha.py: 22 项测试全绿 - TestDignityLevel: 10 项（Exalted/Debilitated/Own/Moolatrikona/Friend/Enemy/Neutral） - TestDashaCalculation: 7 项（balance展示、日期连续性、birth+remaining验证、Dasha顺序、AD、Nakshatra映射） - TestNakshatraMapping: 3 项（Ashwini/Revati/Jyeshtha边界） - TestYogaDetection: 2 项（Raja Yoga检测、无Yoga误报） ## COVERAGE_AUDIT_REPORT.md - 推送 v3.13.1 覆盖矩阵审计报告到仓库
This commit is contained in:
732642856
@@ -0,0 +1,258 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Jyotish Engine 单元测试
|
||||
覆盖: Dasha 计算、尊严等级判定、Yoga 格局识别
|
||||
|
||||
运行方式:
|
||||
cd yinduzhanxing
|
||||
python3 -m pytest tests/ -v
|
||||
# 或
|
||||
python3 -m unittest discover tests/
|
||||
"""
|
||||
|
||||
import unittest
|
||||
import sys
|
||||
import os
|
||||
from datetime import datetime
|
||||
|
||||
# 确保可以导入 scripts/ 下的模块
|
||||
SCRIPT_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'scripts')
|
||||
sys.path.insert(0, SCRIPT_DIR)
|
||||
ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||||
sys.path.insert(0, ROOT_DIR)
|
||||
|
||||
from jyotish_engine import (
|
||||
cmd_dasha, _get_dignity_level, cmd_yoga,
|
||||
EXALTATION, DEBILITATION, SIGN_LORDS, MOOLATRIKONA,
|
||||
DASHA_ORDER, DASHA_YEARS, NAKSHATRA_LIST, SIGNS,
|
||||
)
|
||||
|
||||
|
||||
class TestDignityLevel(unittest.TestCase):
|
||||
"""测试尊严等级判定 (_get_dignity_level)"""
|
||||
|
||||
def test_exalted(self):
|
||||
"""太阳在白羊座 = Exalted"""
|
||||
self.assertEqual(_get_dignity_level('Sun', 'Aries'), 'EXALTED')
|
||||
|
||||
def test_debilitated(self):
|
||||
"""太阳在天秤座 = Debilitated"""
|
||||
self.assertEqual(_get_dignity_level('Sun', 'Libra'), 'DEBILITATED')
|
||||
|
||||
def test_own_sign(self):
|
||||
"""太阳在狮子座 = Own Sign"""
|
||||
self.assertEqual(_get_dignity_level('Sun', 'Leo'), 'OWN_SIGN')
|
||||
|
||||
def test_moolatrikona_in_range(self):
|
||||
"""火星在白羊座 0-12° = Moolatrikona"""
|
||||
self.assertEqual(_get_dignity_level('Mars', 'Aries', 5.0), 'MOOLATRIKONA')
|
||||
|
||||
def test_moolatrikona_out_of_range(self):
|
||||
"""火星在白羊座 12°+ = Own Sign (不是 Moolatrikona)"""
|
||||
self.assertEqual(_get_dignity_level('Mars', 'Aries', 15.0), 'OWN_SIGN')
|
||||
|
||||
def test_friend(self):
|
||||
"""月亮在 Aries（火星的星座，月亮是火星的朋友）"""
|
||||
# Mars 的 PERMANENT_FRIENDS 包含 Moon，所以 Moon 在 Mars 的星座是 Friend
|
||||
# 但这里测的是 Moon 在 Aries: Aries lord = Mars, Moon in PERMANENT_FRIENDS['Mars'] = yes
|
||||
self.assertEqual(_get_dignity_level('Moon', 'Aries'), 'FRIEND')
|
||||
|
||||
def test_enemy(self):
|
||||
"""金星在 Aries（火星的星座，金星是火星的敌人 → 但实际看 Aries lord(Mars) 的朋友表里没有 Venus → 不是 Friend）
|
||||
Venus 在 PERMANENT_ENEMIES['Mars'] 吗？不在。看 Sun 的: Venus in PERMANENT_ENEMIES['Sun'].
|
||||
测试：Venus 在 Cancer（Moon 的星座），PERMANENT_FRIENDS['Moon'] = ['Sun', 'Mercury']，不包含 Venus
|
||||
PERMANENT_ENEMIES['Moon'] = []，所以 Venus 在 Cancer = NEUTRAL
|
||||
"""
|
||||
# 测一个明确的 Enemy: Venus 在 Aries（Mars 守护，Venus 不在 Mars 的 friends 也不在 enemies）
|
||||
# Venus 在 Scorpio（Mars 守护）：同上
|
||||
# 更好的例子: Saturn 在 Leo（Sun 守护，Saturn 在 PERMANENT_ENEMIES['Sun']）
|
||||
self.assertEqual(_get_dignity_level('Saturn', 'Leo'), 'ENEMY')
|
||||
|
||||
def test_neutral(self):
|
||||
"""木星在 Gemini（Mercury 守护），Jupiter 不在 Mercury 的 friends 也不在 enemies"""
|
||||
# PERMANENT_FRIENDS['Mercury'] = ['Sun', 'Venus']
|
||||
# PERMANENT_ENEMIES['Mercury'] = ['Moon']
|
||||
self.assertEqual(_get_dignity_level('Jupiter', 'Gemini'), 'NEUTRAL')
|
||||
|
||||
def test_all_exaltation_signs(self):
|
||||
"""验证所有行星的入旺星座"""
|
||||
expected = {
|
||||
'Sun': 'Aries', 'Moon': 'Taurus', 'Mars': 'Capricorn',
|
||||
'Mercury': 'Virgo', 'Jupiter': 'Cancer', 'Venus': 'Pisces', 'Saturn': 'Libra'
|
||||
}
|
||||
for planet, sign in expected.items():
|
||||
result = _get_dignity_level(planet, sign)
|
||||
self.assertEqual(result, 'EXALTED', f"{planet} in {sign} should be EXALTED, got {result}")
|
||||
|
||||
def test_all_debilitation_signs(self):
|
||||
"""验证所有行星的落陷星座"""
|
||||
expected = {
|
||||
'Sun': 'Libra', 'Moon': 'Scorpio', 'Mars': 'Cancer',
|
||||
'Mercury': 'Pisces', 'Jupiter': 'Capricorn', 'Venus': 'Virgo', 'Saturn': 'Aries'
|
||||
}
|
||||
for planet, sign in expected.items():
|
||||
result = _get_dignity_level(planet, sign)
|
||||
self.assertEqual(result, 'DEBILITATED', f"{planet} in {sign} should be DEBILITATED, got {result}")
|
||||
|
||||
|
||||
class TestDashaCalculation(unittest.TestCase):
|
||||
"""测试 Dasha 时间线计算"""
|
||||
|
||||
def _make_dasha_args(self, moon_lon, birthdate, today=None):
|
||||
"""构造 cmd_dasha 参数"""
|
||||
today = today or '2026-05-04'
|
||||
return type('Args', (), {
|
||||
'moon_lon': moon_lon,
|
||||
'nakshatra': None,
|
||||
'pada': None,
|
||||
'birthdate': birthdate,
|
||||
'today': today,
|
||||
})()
|
||||
|
||||
def test_first_md_shows_balance(self):
|
||||
"""第一个 MD 的 years 字段显示 balance，full_years 显示完整年数"""
|
||||
# 月亮在 Ardra (Rahu 主宰, 18年), 月亮黄经 73° → Ardra 范围 66.67-80°
|
||||
args = self._make_dasha_args(73.0, '1990-01-15')
|
||||
result = cmd_dasha(args)
|
||||
timeline = result.get('timeline', [])
|
||||
self.assertTrue(len(timeline) > 0, "Timeline should not be empty")
|
||||
first_md = timeline[0]
|
||||
# Ardra 主宰 Rahu
|
||||
self.assertEqual(first_md['lord'], 'Rahu')
|
||||
# years (展示值) 应该 < 18（balance）
|
||||
self.assertLess(first_md['years'], 18, f"First MD display years should be < 18, got {first_md['years']}")
|
||||
# full_years 应该 = 18
|
||||
self.assertEqual(first_md['full_years'], 18)
|
||||
# 应该有 is_balance 和 balance_years
|
||||
self.assertTrue(first_md.get('is_balance', False))
|
||||
self.assertIsNotNone(first_md.get('balance_years'))
|
||||
|
||||
def test_timeline_date_chain_continuous(self):
|
||||
"""验证时间线日期链条连续：每个 MD 的 start = 上一个 MD 的 end"""
|
||||
args = self._make_dasha_args(100.0, '1990-01-15')
|
||||
result = cmd_dasha(args)
|
||||
timeline = result.get('timeline', [])
|
||||
self.assertTrue(len(timeline) >= 2, "Need at least 2 MDs")
|
||||
for i in range(1, len(timeline)):
|
||||
self.assertEqual(timeline[i]['start'], timeline[i-1]['end'],
|
||||
f"MD{i} start should equal MD{i-1} end: {timeline[i]['start']} != {timeline[i-1]['end']}")
|
||||
|
||||
def test_first_md_ends_at_birth_plus_remaining(self):
|
||||
"""验证第一个 MD 的 end 日期 = birth + remaining 年（核心正确性）
|
||||
原始代码的数学等价性：end = (birth - elapsed) + full_years = birth + remaining
|
||||
"""
|
||||
args = self._make_dasha_args(45.0, '1990-06-15', '2026-01-01')
|
||||
result = cmd_dasha(args)
|
||||
timeline = result.get('timeline', [])
|
||||
first = timeline[0]
|
||||
birth = datetime.strptime('1990-06-15', '%Y-%m-%d')
|
||||
end = datetime.strptime(first['end'], '%Y-%m-%d')
|
||||
# end - birth 应该 ≈ balance_years
|
||||
actual_years = (end - birth).days / 365.25
|
||||
balance = first.get('balance_years', first['years'])
|
||||
self.assertAlmostEqual(actual_years, balance, delta=0.1,
|
||||
msg=f"First MD end should be birth+{balance}y, got {actual_years:.2f}y")
|
||||
|
||||
def test_dasha_order_correct(self):
|
||||
"""验证 Dasha 顺序遵循 Ketu→Venus→Sun→Moon→Mars→Rahu→Jupiter→Saturn→Mercury"""
|
||||
# 用月亮在 Ashwini (Ketu 主宰) 的位置
|
||||
args = self._make_dasha_args(5.0, '1990-01-15') # Ashwini
|
||||
result = cmd_dasha(args)
|
||||
timeline = result.get('timeline', [])
|
||||
if len(timeline) >= 9:
|
||||
actual_order = [d['lord'] for d in timeline]
|
||||
self.assertEqual(actual_order, DASHA_ORDER,
|
||||
f"Dasha order should be {DASHA_ORDER}, got {actual_order}")
|
||||
|
||||
def test_nakshatra_moon_lon_mapping(self):
|
||||
"""验证月亮黄经正确映射到 Nakshatra"""
|
||||
# Revati: 346.67° - 360°
|
||||
args = self._make_dasha_args(350.0, '1990-01-15')
|
||||
result = cmd_dasha(args)
|
||||
self.assertEqual(result.get('moon_nakshatra'), 'Revati')
|
||||
|
||||
def test_antardasha_in_current_dasha(self):
|
||||
"""验证当前 Dasha 包含 Antardasha 信息"""
|
||||
args = self._make_dasha_args(100.0, '1990-01-15', '2026-05-04')
|
||||
result = cmd_dasha(args)
|
||||
current = result.get('current_dasha')
|
||||
if current:
|
||||
self.assertIn('antardasha', current, "Current dasha should have antardasha")
|
||||
ad = current['antardasha']
|
||||
self.assertTrue(len(ad) > 0, "Antardasha should not be empty")
|
||||
|
||||
def test_barack_oboma_dasha(self):
|
||||
"""已知案例: Obama 月亮在 Rohini (约 35°59' → 实际 Krittika, 用 45° 测 Rohini)"""
|
||||
# Krittika: 26.67° - 40°, Rohini: 40° - 53.33°
|
||||
# 35.98° 在 Krittika 范围，用 45° 测 Rohini
|
||||
args = self._make_dasha_args(45.0, '1961-08-04', '2024-01-01')
|
||||
result = cmd_dasha(args)
|
||||
self.assertEqual(result['moon_nakshatra'], 'Rohini')
|
||||
# Rohini 主宰 Moon，所以第一个 MD 应该是 Moon
|
||||
timeline = result.get('timeline', [])
|
||||
if timeline:
|
||||
self.assertEqual(timeline[0]['lord'], 'Moon')
|
||||
|
||||
|
||||
class TestNakshatraMapping(unittest.TestCase):
|
||||
"""测试 Nakshatra 映射"""
|
||||
|
||||
def test_ashwini_start(self):
|
||||
"""0° → Ashwini"""
|
||||
args = type('Args', (), {
|
||||
'moon_lon': 0.5, 'nakshatra': None, 'pada': None,
|
||||
'birthdate': '1990-01-01', 'today': '2026-01-01',
|
||||
})()
|
||||
result = cmd_dasha(args)
|
||||
self.assertEqual(result['moon_nakshatra'], 'Ashwini')
|
||||
|
||||
def test_revati_end(self):
|
||||
"""~359° → Revati"""
|
||||
args = type('Args', (), {
|
||||
'moon_lon': 359.0, 'nakshatra': None, 'pada': None,
|
||||
'birthdate': '1990-01-01', 'today': '2026-01-01',
|
||||
})()
|
||||
result = cmd_dasha(args)
|
||||
self.assertEqual(result['moon_nakshatra'], 'Revati')
|
||||
|
||||
def test_jyeshtha(self):
|
||||
"""~230° → Jyeshtha (226.67° - 240°)"""
|
||||
args = type('Args', (), {
|
||||
'moon_lon': 230.0, 'nakshatra': None, 'pada': None,
|
||||
'birthdate': '1990-01-01', 'today': '2026-01-01',
|
||||
})()
|
||||
result = cmd_dasha(args)
|
||||
self.assertEqual(result['moon_nakshatra'], 'Jyeshtha')
|
||||
|
||||
|
||||
class TestYogaDetection(unittest.TestCase):
|
||||
"""测试 Yoga 格局识别"""
|
||||
|
||||
def _make_yoga_args(self, ascendant, planets_str):
|
||||
return type('Args', (), {
|
||||
'ascendant': ascendant,
|
||||
'planets': planets_str,
|
||||
})()
|
||||
|
||||
def test_raja_yoga_detection(self):
|
||||
"""测试 Raja Yoga 检测：角宫主+三方主同宫"""
|
||||
# Leo Ascendant: Kendra lords = Sun(1st), Mars(4th/Scorpio), Jupiter(7th/Pisces)
|
||||
# Trikona lords = Sun(1st), Jupiter(5th/Sagittarius)
|
||||
# 如果 Jupiter 和 Sun 都在某宫 → Raja Yoga
|
||||
args = self._make_yoga_args('Leo', 'Sun:Leo:1,Jupiter:Leo:1,Moon:Taurus:10,Mars:Scorpio:4')
|
||||
result = cmd_yoga(args)
|
||||
yogas = result.get('yogas', [])
|
||||
raja = [y for y in yogas if y['name'] == 'Raja Yoga']
|
||||
self.assertTrue(len(raja) > 0, "Should detect Raja Yoga when Sun+Jupiter in 1st house")
|
||||
|
||||
def test_no_yoga_false_positive(self):
|
||||
"""测试不满足条件时不产生 Yoga"""
|
||||
# 行星分散，不满足任何 Yoga
|
||||
args = self._make_yoga_args('Aries', 'Sun:Taurus:2,Moon:Gemini:3')
|
||||
result = cmd_yoga(args)
|
||||
yogas = result.get('yogas', [])
|
||||
self.assertEqual(len(yogas), 0, "Should not detect any Yoga with scattered planets")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user