70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
"""
|
||
运行所有测试的主脚本
|
||
"""
|
||
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
# 添加项目路径
|
||
project_root = Path(__file__).parent.parent
|
||
sys.path.insert(0, str(project_root))
|
||
|
||
from test_consistency import main as test_consistency
|
||
from example_usage import (
|
||
example_single_prediction,
|
||
example_batch_prediction,
|
||
example_smiles_list_prediction
|
||
)
|
||
|
||
|
||
def run_all_tests():
|
||
"""运行所有测试"""
|
||
print("🧪 开始运行广谱抗菌预测API测试套件")
|
||
print("=" * 60)
|
||
|
||
# 1. 一致性测试
|
||
print("\n📊 运行一致性测试...")
|
||
try:
|
||
consistency_passed = test_consistency()
|
||
except Exception as e:
|
||
print(f"一致性测试失败: {e}")
|
||
consistency_passed = False
|
||
|
||
# 2. 功能测试
|
||
print("\n🔧 运行功能测试...")
|
||
try:
|
||
print("测试单分子预测...")
|
||
example_single_prediction()
|
||
|
||
print("\n测试批量预测...")
|
||
example_batch_prediction()
|
||
|
||
print("\n测试SMILES列表预测...")
|
||
example_smiles_list_prediction()
|
||
|
||
functional_passed = True
|
||
print("✅ 功能测试通过")
|
||
|
||
except Exception as e:
|
||
print(f"❌ 功能测试失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
functional_passed = False
|
||
|
||
# 总结
|
||
print("\n" + "=" * 60)
|
||
print("📋 测试总结:")
|
||
print(f" 一致性测试: {'✅ 通过' if consistency_passed else '❌ 失败'}")
|
||
print(f" 功能测试: {'✅ 通过' if functional_passed else '❌ 失败'}")
|
||
|
||
if consistency_passed and functional_passed:
|
||
print("\n🎉 所有测试通过！API可以正常使用。")
|
||
return True
|
||
else:
|
||
print("\n⚠️ 部分测试失败，请检查问题。")
|
||
return False
|
||
|
||
|
||
if __name__ == "__main__":
|
||
success = run_all_tests()
|
||
sys.exit(0 if success else 1) |