"""
简单测试五行颜色自动推导功能
"""

import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from app.core.data_parse.calendar import CalendarInfo

def test_simple_wuxing_color():
    """简单测试五行颜色推导"""
    print("=== 简单测试五行颜色推导 ===")
    
    # 测试没有预设颜色，只根据wuxing推导
    test_data = {
        'yangli': '2025-01-20',
        'yinli': '腊月二十一',
        'wuxing': '金'  # 只有五行元素，没有预设颜色
    }
    
    calendar_info = CalendarInfo.from_dict(test_data)
    print(f"输入wuxing: {test_data['wuxing']}")
    print(f"推导出的color: {calendar_info.color}")
    print(f"期望color: 白")
    
    if calendar_info.color == '白':
        print("✅ 五行颜色推导测试通过！")
        return True
    else:
        print("❌ 五行颜色推导测试失败！")
        return False

if __name__ == "__main__":
    test_simple_wuxing_color()

