Medical Calculator MCP Service

import asyncio import json import sys import os from fastmcp import Client sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from config import MCP_SERVER_URL async def test_steroid_conversion_calculator(client): """测试 Steroid Conversion 计算器的各种功能""" def print_header(): print("\n" + "=" * 60) print("Steroid Conversion 计算器测试套件") print("=" * 60) def print_test_case(i, test_case): print(f"\n测试 {i:2d} | {test_case['name']}") print(f"- {test_case['description']}") print(f"- 输入参数: {test_case['params']}") def print_validation_result(expected, actual, errors=None, warnings=None): if expected == actual: status = "✅ 通过" else: status = "❌ 失败" expected_text = "有效" if expected else "无效" actual_text = "有效" if actual else "无效" print(f"- 验证结果: {status} (期望: {expected_text}, 实际: {actual_text})") if errors: print(f"- ⚠️ 错误: {errors}") if warnings: print(f"- ⚠️ 警告: {warnings}") def print_calculation_result(data, expected_result=None): """打印完整的计算结果""" result_value = data.get("value", "N/A") explanation = data.get("explanation", "") metadata = data.get("metadata", {}) warnings = data.get("warnings", []) # 基本结果 print(f"- 转换结果: {result_value} mg") if expected_result is not None: print(f"- 期望结果: {expected_result} mg") # 检查数值是否匹配（允许小的浮点误差） try: actual_val = float(result_value) expected_val = float(expected_result) diff = abs(actual_val - expected_val) if diff < 0.001: # 允许0.001的误差 print("- 结果匹配: ✅ 通过") else: print(f"- 结果匹配: ❌ 失败 (差异: {diff})") except: print("- 结果匹配: ❌ 失败 (格式错误)") # 详细信息 if metadata: input_steroid = metadata.get("input_steroid", "N/A") input_dose = metadata.get("input_dose_mg", "N/A") target_steroid = metadata.get("target_steroid", "N/A") conversion_factor = metadata.get("conversion_factor", "N/A") print(f"- 输入类固醇: {input_steroid}") print(f"- 输入剂量: {input_dose} mg") print(f"- 目标类固醇: {target_steroid}") print(f"- 转换因子: {conversion_factor}") # 警告信息 if warnings: for warning in warnings: print(f"- ⚠️ 警告: {warning}") # 解释（截取前几行显示） if explanation: lines = explanation.split('\n')[:3] # 只显示前3行 print(f"- 解释: {' '.join(lines)}") def print_test_result(i, passed): if passed: status = "✅ 通过" else: status = "❌ 失败" print(f"- 测试结果: {status}") print("-" * 60) def print_summary(total, passed, failed): print(f"\n测试总结:") print(f" 总测试数: {total}") print(f" 通过数: {passed}") print(f" 失败数: {failed}") print(f" 成功率: {(passed/total*100):.1f}%") if failed == 0: print("\n✅ 所有测试都通过了！Steroid Conversion 计算器工作正常。") else: print(f"\n❌ {failed} 个测试失败，请检查实现。") print("\n测试覆盖范围:") features = [ "多种类固醇类型转换", "IV和PO给药途径", "剂量转换精度", "参数验证", "错误处理", "边界测试", ] for feature in features: print(f" - {feature}") # Test statistics total_tests = 0 passed_tests = 0 # Test cases based on data file test_cases = [ { "name": "Dexamethasone PO to Betamethasone IV", "params": {"input_steroid": "Dexamethasone PO", "input_dose": 7.52, "target_steroid": "Betamethasone IV"}, "expected_valid": True, "expected_result": 7.52, "description": "地塞米松口服转换为倍他米松静脉注射", }, { "name": "Dexamethasone PO to Hydrocortisone IV", "params": {"input_steroid": "Dexamethasone PO", "input_dose": 7.61, "target_steroid": "Hydrocortisone IV"}, "expected_valid": True, "expected_result": 202.959, "description": "地塞米松口服转换为氢化可的松静脉注射", }, { "name": "Betamethasone IV to MethylPrednisoLONE IV", "params": {"input_steroid": "Betamethasone IV", "input_dose": 1.35, "target_steroid": "MethylPrednisoLONE IV"}, "expected_valid": True, "expected_result": 7.196, "description": "倍他米松静脉注射转换为甲基强的松龙静脉注射", }, { "name": "PrednisoLONE PO to Dexamethasone PO", "params": {"input_steroid": "PrednisoLONE PO", "input_dose": 34.068, "target_steroid": "Dexamethasone PO"}, "expected_valid": True, "expected_result": 5.11, "description": "强的松龙口服转换为地塞米松口服", }, { "name": "Hydrocortisone IV to MethylPrednisoLONE IV", "params": {"input_steroid": "Hydrocortisone IV", "input_dose": 107.468, "target_steroid": "MethylPrednisoLONE IV"}, "expected_valid": True, "expected_result": 21.494, "description": "氢化可的松静脉注射转换为甲基强的松龙静脉注射", }, { "name": "PredniSONE PO to Hydrocortisone IV", "params": {"input_steroid": "PredniSONE PO", "input_dose": 11.267, "target_steroid": "Hydrocortisone IV"}, "expected_valid": True, "expected_result": 45.057, "description": "强的松口服转换为氢化可的松静脉注射", }, { "name": "Cortisone PO to PrednisoLONE PO", "params": {"input_steroid": "Cortisone PO", "input_dose": 222.331, "target_steroid": "PrednisoLONE PO"}, "expected_valid": True, "expected_result": 44.466, "description": "可的松口服转换为强的松龙口服", }, { "name": "Invalid input steroid", "params": {"input_steroid": "Invalid Steroid", "input_dose": 5.0, "target_steroid": "Dexamethasone PO"}, "expected_valid": False, "description": "无效输入类固醇", }, { "name": "Invalid target steroid", "params": {"input_steroid": "Dexamethasone PO", "input_dose": 5.0, "target_steroid": "Invalid Steroid"}, "expected_valid": False, "description": "无效目标类固醇", }, { "name": "Negative dose", "params": {"input_steroid": "Dexamethasone PO", "input_dose": -5.0, "target_steroid": "Betamethasone IV"}, "expected_valid": False, "description": "负剂量测试", }, { "name": "Zero dose", "params": {"input_steroid": "Dexamethasone PO", "input_dose": 0, "target_steroid": "Betamethasone IV"}, "expected_valid": False, "description": "零剂量测试", }, ] print_header() # Execute test cases for i, test_case in enumerate(test_cases, 1): total_tests += 1 test_passed = True print_test_case(i, test_case) # Calculation test (validation is included in calculate) try: calc_result = await client.call_tool( "calculate", { "calculator_id": 24, "parameters": test_case["params"], }, ) # 使用 structured_content 或 data 属性获取实际数据 calc_data = calc_result.structured_content or calc_result.data or {} if isinstance(calc_data, dict) and calc_data.get("success") and "result" in calc_data: # 成功计算 data = calc_data["result"] expected_result = test_case.get("expected_result") print_calculation_result(data, expected_result) # 检查是否符合预期 if not test_case["expected_valid"]: print("- 错误: 预期失败但计算成功") test_passed = False elif expected_result is not None: # 检查结果精度 try: actual_val = float(data.get("value", 0)) expected_val = float(expected_result) diff = abs(actual_val - expected_val) if diff >= 0.001: # 如果差异大于0.001则认为失败 test_passed = False except: test_passed = False else: # 计算失败（可能是参数验证失败） error_msg = calc_data.get("error", "未知错误") if isinstance(calc_data, dict) else str(calc_data) print(f"- 计算失败: {error_msg}") # 检查是否符合预期 if test_case["expected_valid"]: print("- 错误: 预期成功但计算失败") test_passed = False except Exception as e: print(f"- 计算错误: {e}") # 检查是否符合预期 if test_case["expected_valid"]: test_passed = False # Update statistics if test_passed: passed_tests += 1 print_test_result(i, test_passed) print_summary(total_tests, passed_tests, total_tests - passed_tests) return passed_tests, total_tests - passed_tests async def main(): def print_header(): print("Steroid Conversion 计算器 MCP 测试") print("=" * 60) def print_connection_status(success, error=None): if success: print("✅ 成功连接到 MCP 服务器") else: print(f"❌ 连接失败: {error}") def print_overall_results(total_passed, total_failed): total_tests = total_passed + total_failed if total_tests == 0: return print("\n" + "=" * 60) print("Steroid Conversion 计算器测试结果") print("=" * 60) print(f"总测试数: {total_tests}") print(f"通过数: {total_passed}") print(f"失败数: {total_failed}") print(f"成功率: {(total_passed/total_tests*100):.1f}%") if total_failed == 0: print("\n✅ Steroid Conversion 计算器所有测试都通过了！") else: print(f"\n❌ {total_failed} 个测试失败，请检查 Steroid Conversion 计算器实现。") print_header() try: async with Client(MCP_SERVER_URL) as client: print_connection_status(True) passed, failed = await test_steroid_conversion_calculator(client) print_overall_results(passed, failed) except Exception as e: print_connection_status(False, str(e)) import traceback traceback.print_exc() return print("\n" + "=" * 60) print("✅ Steroid Conversion 计算器测试完成") if __name__ == "__main__": asyncio.run(main())