test(tkeg): 분류기 테스트 수정 — 변경된 반환 키 대응 (8/8 통과)

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Hyungi Ahn
2026-03-27 07:08:17 +09:00
parent ce47865890
commit ea6f7c3013
4 changed files with 90 additions and 87 deletions

View File

@@ -99,36 +99,27 @@ def test_fitting_classification():
print(f"📋 입력:")
print(f" DAT_FILE: {test['dat_file']}")
print(f" DESCRIPTION: {test['description']}")
print(f" SIZE: {result['size_info']['size_description']}")
print(f" SIZE: {test.get('main_nom', '')}")
print(f"\n🔧 분류 결과:")
print(f" 재질: {result['material']['standard']} | {result['material']['grade']}")
print(f" 피팅타입: {result['fitting_type']['type']} - {result['fitting_type']['subtype']}")
print(f" 크기정보: {result['size_info']['size_description']}") # ← 추가!
if result['size_info']['reduced_size']:
print(f" 주사이즈: {result['size_info']['main_size']}")
print(f" 축소사이즈: {result['size_info']['reduced_size']}")
print(f" 연결방식: {result['connection_method']['method']}")
print(f" 압력등급: {result['pressure_rating']['rating']} ({result['pressure_rating']['common_use']})")
print(f" 제작방법: {result['manufacturing']['method']} ({result['manufacturing']['characteristics']})")
print(f" 압력등급: {result['pressure_rating']['rating']}")
print(f" 제작방법: {result['manufacturing']}")
print(f"\n📊 신뢰도:")
print(f" 전체신뢰도: {result['overall_confidence']}")
print(f" 재질: {result['material']['confidence']}")
print(f" 피팅타입: {result['fitting_type']['confidence']}")
print(f" 연결방식: {result['connection_method']['confidence']}")
print(f" 압력등급: {result['pressure_rating']['confidence']}")
print(f"\n🛒 구매 정보:")
print(f" 공급업체: {purchase_info['supplier_type']}")
print(f" 예상납기: {purchase_info['lead_time_estimate']}")
print(f" 구매카테고리: {purchase_info['purchase_category']}")
print(f" 공급업체: {purchase_info.get('supplier_type', 'N/A')}")
print(f" 구매카테고리: {purchase_info.get('purchase_category', 'N/A')}")
# 크기 정보 저장 확인
print(f"\n💾 저장될 데이터:")
print(f" MAIN_NOM: {result['size_info']['main_size']}")
print(f" RED_NOM: {result['size_info']['reduced_size'] or 'NULL'}")
print(f" SIZE_DESCRIPTION: {result['size_info']['size_description']}")
# 분류 요약
print(f"\n💾 분류 요약:")
print(f" 카테고리: {result.get('category', 'N/A')}")
print(f" 신뢰도: {result.get('overall_confidence', 'N/A')}")
if i < len(test_cases):
print("\n" + "=" * 80)
@@ -174,10 +165,11 @@ def test_fitting_edge_cases():
test["red_nom"]
)
print(f"결과: {result['fitting_type']['type']} - {result['fitting_type']['subtype']}")
print(f"크기: {result['size_info']['size_description']}") # ← 추가!
print(f"신뢰도: {result['overall_confidence']}")
print(f"증거: {result['fitting_type']['evidence']}")
ft = result.get('fitting_type', {})
print(f"결과: {ft.get('type', 'N/A')} - {ft.get('subtype', 'N/A')}")
print(f"카테고리: {result.get('category', 'N/A')}")
print(f"신뢰도: {result.get('overall_confidence', 'N/A')}")
print(f"증거: {ft.get('evidence', [])}")
if __name__ == "__main__":
test_fitting_classification()