ai4mats-mcp-tools/test/test_tem.py

36 lines
975 B
Python

# test_tem.py
import asyncio
import sys
import os
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from tools.tem_analysis_mcp_server import analyze_tem_image
async def main():
print("🚀 开始测试 TEM 分析 MCP 服务...")
# 这是一个 1x1 像素的透明 PNG 图片的 Base64 编码(仅用于测试连通性)
dummy_base64 = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg=="
# 测试参数
try:
result = await analyze_tem_image(
filename="test_sample.jpg",
content_base64=dummy_base64,
model="/detection/predict"
)
print("\n✅ TEM 分析 MCP 服务调用成功!")
print("=" * 60)
print("返回结果:")
print(result)
print("=" * 60)
except Exception as e:
print(f"\n❌ TEM 分析 MCP 服务调用失败: {e}")
if __name__ == "__main__":
asyncio.run(main())