54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
"""
|
|
재질 분류 테스트
|
|
"""
|
|
|
|
from .material_classifier import classify_material, get_manufacturing_method_from_material
|
|
|
|
def test_material_classification():
|
|
"""재질 분류 테스트"""
|
|
|
|
test_cases = [
|
|
{
|
|
"description": "PIPE, SMLS, SCH 80, ASTM A106 GR B BOE-POE",
|
|
"expected_standard": "ASTM A106"
|
|
},
|
|
{
|
|
"description": "TEE, SW, 3000LB, ASTM A182 F304",
|
|
"expected_standard": "ASTM A182"
|
|
},
|
|
{
|
|
"description": "90 LR ELL, SCH 40, ASTM A234 GR WPB, SMLS",
|
|
"expected_standard": "ASTM A234"
|
|
},
|
|
{
|
|
"description": "FLG WELD NECK RF SCH 40, 150LB, ASTM A105",
|
|
"expected_standard": "ASTM A105"
|
|
},
|
|
{
|
|
"description": "GATE VALVE, ASTM A216 WCB",
|
|
"expected_standard": "ASTM A216"
|
|
},
|
|
{
|
|
"description": "SPECIAL FITTING, INCONEL 625",
|
|
"expected_standard": "INCONEL"
|
|
}
|
|
]
|
|
|
|
print("🔧 재질 분류 테스트 시작\n")
|
|
|
|
for i, test in enumerate(test_cases, 1):
|
|
result = classify_material(test["description"])
|
|
manufacturing = get_manufacturing_method_from_material(result)
|
|
|
|
print(f"테스트 {i}:")
|
|
print(f" 입력: {test['description']}")
|
|
print(f" 결과: {result['standard']} | {result['grade']}")
|
|
print(f" 재질타입: {result['material_type']}")
|
|
print(f" 제작방법: {manufacturing}")
|
|
print(f" 신뢰도: {result['confidence']}")
|
|
print(f" 증거: {result.get('evidence', [])}")
|
|
print()
|
|
|
|
if __name__ == "__main__":
|
|
test_material_classification()
|