#!/usr/bin/env python3
"""
main_nom, red_nom 기능 테스트 스크립트
"""

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

from app.services.fitting_classifier import classify_fitting
from app.services.flange_classifier import classify_flange

def test_main_red_nom():
    """main_nom과 red_nom 분류 테스트"""
    
    print("🔧 main_nom/red_nom 분류 테스트 시작!")
    print("=" * 60)
    
    test_cases = [
        {
            "name": "일반 TEE (동일 사이즈)",
            "description": "TEE, SCH 40, ASTM A234 GR WPB",
            "main_nom": "4\"",
            "red_nom": None,
            "expected": "EQUAL TEE"
        },
        {
            "name": "리듀싱 TEE (다른 사이즈)",
            "description": "TEE RED, SCH 40 x SCH 40, ASTM A234 GR WPB",
            "main_nom": "4\"",
            "red_nom": "2\"", 
            "expected": "REDUCING TEE"
        },
        {
            "name": "동심 리듀서",
            "description": "RED CONC, SCH 40 x SCH 40, ASTM A234 GR WPB",
            "main_nom": "6\"",
            "red_nom": "4\"",
            "expected": "CONCENTRIC REDUCER"
        },
        {
            "name": "리듀싱 플랜지",
            "description": "FLG REDUCING, 300LB, ASTM A105",
            "main_nom": "6\"",
            "red_nom": "4\"",
            "expected": "REDUCING FLANGE"
        }
    ]
    
    for i, test in enumerate(test_cases, 1):
        print(f"\n{i}. {test['name']}")
        print(f"   설명: {test['description']}")
        print(f"   MAIN_NOM: {test['main_nom']}")
        print(f"   RED_NOM: {test['red_nom']}")
        
        # 피팅 분류 테스트
        fitting_result = classify_fitting(
            "", 
            test['description'], 
            test['main_nom'], 
            test['red_nom']
        )
        
        print(f"   🔧 FITTING 분류 결과:")
        print(f"      카테고리: {fitting_result.get('category')}")
        print(f"      타입: {fitting_result.get('fitting_type', {}).get('type')}")
        print(f"      서브타입: {fitting_result.get('fitting_type', {}).get('subtype')}")
        print(f"      신뢰도: {fitting_result.get('overall_confidence', 0):.2f}")
        
        # 사이즈 정보 확인
        size_info = fitting_result.get('size_info', {})
        print(f"      메인 사이즈: {size_info.get('main_size')}")
        print(f"      축소 사이즈: {size_info.get('reduced_size')}")
        print(f"      사이즈 설명: {size_info.get('size_description')}")
        
        # RED_NOM이 있는 경우 REDUCING 분류 확인
        if test['red_nom']:
            fitting_type = fitting_result.get('fitting_type', {})
            if 'REDUCING' in fitting_type.get('subtype', '').upper():
                print(f"      ✅ REDUCING 타입 정상 인식!")
            else:
                print(f"      ❌ REDUCING 타입 인식 실패")
        
        print("-" * 50)
    
    print("\n🎯 테스트 완료!")

if __name__ == "__main__":
    test_main_red_nom()