732642856
341 lines
14 KiB
Python
341 lines
14 KiB
Python
#!/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,
|
||
)
|
||
from kalachakra_dasha import calculate_kalachakra_dasha
|
||
|
||
|
||
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):
|
||
"""太阳在 Cancer（Sun 对 Moon 为友，且不触发更高优先级尊严）"""
|
||
self.assertEqual(_get_dignity_level('Sun', 'Cancer'), '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):
|
||
"""月亮在 Aries（Moon 对 Mars 既非友也非敌）"""
|
||
self.assertEqual(_get_dignity_level('Moon', 'Aries'), '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}")
|
||
|
||
def test_temporary_relationship_depends_on_passed_chart_context(self):
|
||
"""同一落位若误用 D1 planets_data，会改变分盘 dignity 结果。"""
|
||
d1_like_context = {
|
||
'Jupiter': {'sign': 'Virgo'},
|
||
'Mercury': {'sign': 'Libra'},
|
||
}
|
||
d9_like_context = {
|
||
'Jupiter': {'sign': 'Virgo'},
|
||
'Mercury': {'sign': 'Capricorn'},
|
||
}
|
||
d1_result = _get_dignity_level('Jupiter', 'Virgo', 10.0, d1_like_context)
|
||
d9_result = _get_dignity_level('Jupiter', 'Virgo', 10.0, d9_like_context)
|
||
|
||
self.assertEqual(d1_result, 'NEUTRAL')
|
||
self.assertEqual(d9_result, 'GREAT_ENEMY')
|
||
|
||
def test_neecha_bhanga_depends_on_passed_chart_context(self):
|
||
"""落陷取消必须依赖当前分盘上下文，而不是借用 D1 planets_data。"""
|
||
with_cancelling_dispositor = {
|
||
'Mercury': {'sign': 'Pisces'},
|
||
'Jupiter': {'sign': 'Cancer'},
|
||
'Venus': {'sign': 'Aries'},
|
||
}
|
||
without_cancelling_dispositor = {
|
||
'Mercury': {'sign': 'Pisces'},
|
||
'Jupiter': {'sign': 'Capricorn'},
|
||
'Venus': {'sign': 'Aries'},
|
||
}
|
||
self.assertEqual(
|
||
_get_dignity_level('Mercury', 'Pisces', 12.0, with_cancelling_dispositor),
|
||
'NEECHA_BHANGA',
|
||
)
|
||
self.assertEqual(
|
||
_get_dignity_level('Mercury', 'Pisces', 12.0, without_cancelling_dispositor),
|
||
'DEBILITATED',
|
||
)
|
||
|
||
def test_vimsopaka_calculator_supports_advanced_dignity_levels(self):
|
||
from vimsopaka_calculator import DignityLevel, VargaType, VimsopakaBalaCalculator
|
||
|
||
calculator = VimsopakaBalaCalculator(mode="shodasavarga")
|
||
results = calculator.calculate_vimsopaka_bala({
|
||
'Jupiter': {
|
||
VargaType.D9: DignityLevel.NEECHA_BHANGA,
|
||
VargaType.D1: DignityLevel.GREAT_FRIEND,
|
||
VargaType.D10: DignityLevel.GREAT_ENEMY,
|
||
}
|
||
})
|
||
|
||
jupiter = results['Jupiter']
|
||
assert jupiter['varga_scores']['Navamsa']['dignity'] == 'Neecha Bhanga'
|
||
assert jupiter['varga_scores']['Navamsa']['virupas'] == 9.0
|
||
assert jupiter['varga_scores']['Rashi']['dignity'] == 'Great Friend'
|
||
assert jupiter['varga_scores']['Rashi']['virupas'] == 12.5
|
||
assert jupiter['varga_scores']['Dasamsa']['dignity'] == 'Great Enemy'
|
||
assert jupiter['varga_scores']['Dasamsa']['virupas'] == 3.5
|
||
|
||
|
||
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 TestKalachakraDasha(unittest.TestCase):
|
||
"""测试 Kalachakra Dasha 简化集成状态"""
|
||
|
||
def test_kalachakra_reports_savya_mode(self):
|
||
result = calculate_kalachakra_dasha({
|
||
'moon_nakshatra_index': 0,
|
||
'moon_pada': 1,
|
||
'birth_datetime': datetime(1990, 6, 15, 10, 30),
|
||
})
|
||
self.assertEqual(result['mode'], 'savya')
|
||
self.assertEqual(result['starting_lord'], 'Ketu')
|
||
self.assertEqual(result['starting_rashi'], 'Capricorn')
|
||
self.assertTrue(result['current'])
|
||
|
||
def test_kalachakra_reports_apasavya_mode(self):
|
||
result = calculate_kalachakra_dasha({
|
||
'moon_nakshatra_index': 3,
|
||
'moon_pada': 2,
|
||
'birth_datetime': datetime(1985, 3, 20, 8, 0),
|
||
})
|
||
self.assertEqual(result['mode'], 'apasavya')
|
||
self.assertEqual(result['starting_lord'], 'Ketu')
|
||
self.assertTrue(result['current'])
|
||
self.assertGreater(result['total_cycle'], 0)
|
||
|
||
|
||
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_scattered_planets_do_not_trigger_raja_yoga(self):
|
||
"""分散行星不应触发 Raja Yoga。"""
|
||
# v6.0.32 起规则库扩大到数百条，极简星盘可能触发其他低阶/专项 Yoga；
|
||
# 此回归测试只约束原始目标：不应误报 Raja Yoga。
|
||
args = self._make_yoga_args('Aries', 'Sun:Taurus:2,Mars:Leo:5')
|
||
result = cmd_yoga(args)
|
||
yogas = result.get('yogas', [])
|
||
raja = [y for y in yogas if y['name'] == 'Raja Yoga']
|
||
self.assertEqual(len(raja), 0, "Should not detect Raja Yoga with scattered planets")
|
||
|
||
|
||
if __name__ == '__main__':
|
||
unittest.main()
|