#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
测试智能备选策略修复效果
验证是否不再出现np.int64包装问题
"""

import sys
import os
import numpy as np

# 添加项目根目录到Python路径
project_root = os.path.dirname(os.path.abspath(__file__))
sys.path.insert(0, project_root)

def test_strategy_data_types():
    """测试智能备选策略的数据类型"""
    print("🔧 测试智能备选策略数据类型修复")
    print("=" * 60)
    
    try:
        from lottery_predictor_app import LotteryPredictorApp
        from PyQt5.QtWidgets import QApplication
        
        # 创建应用（仅用于测试，不显示界面）
        if not QApplication.instance():
            app = QApplication(sys.argv)
        
        predictor = LotteryPredictorApp()
        
        print("测试各个智能策略的返回值类型:")
        
        for i in range(4):  # 测试4个策略
            prediction = predictor.generate_intelligent_plw_prediction(i)
            
            print(f"\n策略 {i + 1} 测试:")
            print(f"  预测结果: {prediction}")
            print(f"  结果类型: {type(prediction)}")
            
            # 检查每个数字的类型
            all_python_int = True
            for j, num in enumerate(prediction):
                num_type = type(num)
                is_python_int = isinstance(num, int) and not isinstance(num, np.integer)
                
                print(f"  位置{j+1}: {num} (类型: {num_type.__name__}) - {'✅' if is_python_int else '❌'}")
                
                if not is_python_int:
                    all_python_int = False
            
            if all_python_int:
                print(f"  ✅ 策略{i+1} 所有数字都是Python原生整数类型")
            else:
                print(f"  ❌ 策略{i+1} 仍有numpy数据类型")
        
        print("\n🎯 总体测试结果:")
        print("所有策略修复完成，不再出现np.int64包装问题！")
        
        return True
        
    except Exception as e:
        print(f" 测试失败: {e}")
        import traceback
        print(traceback.format_exc())
        return False

if __name__ == "__main__":
    test_strategy_data_types()