first

pc-1

simple_test.py

@@ -0,0 +1,81 @@
+#!/usr/bin/env python3
+"""
+简单测试三体问题求解器
+"""
+
+import sys
+import os
+
+# 添加当前目录到路径
+current_dir = os.path.dirname(os.path.abspath(__file__))
+sys.path.insert(0, current_dir)
+
+print("三体问题求解器 - 简单测试")
+print("=" * 60)
+
+try:
+    # 尝试导入
+    print("1. 导入模块...")
+    from three_body_problem.particle import Particle
+    from three_body_problem.config import ThreeBodyConfig
+    print("   ✓ 导入成功")
+    
+    # 测试创建质点
+    print("\n2. 测试质点创建...")
+    p = Particle(mass=1.0, position=[1.0, 2.0, 3.0], velocity=[0.1, 0.2, 0.3], name="Test")
+    print(f"   名称: {p.name}")
+    print(f"   质量: {p.mass}")
+    print(f"   位置: {p.position}")
+    print(f"   速度: {p.velocity}")
+    print("   ✓ 质点创建成功")
+    
+    # 测试能量计算
+    energy = p.get_energy()
+    print(f"   动能: {energy:.6f}")
+    print("   ✓ 能量计算成功")
+    
+    # 测试更新
+    print("\n3. 测试质点更新...")
+    p.update([2.0, 3.0, 4.0], [0.2, 0.3, 0.4])
+    print(f"   新位置: {p.position}")
+    print(f"   新速度: {p.velocity}")
+    print(f"   历史记录: {len(p.position_history)} 个位置点")
+    print("   ✓ 质点更新成功")
+    
+    # 测试配置创建
+    print("\n4. 测试配置创建...")
+    particles = ThreeBodyConfig.create_figure8_config()
+    print(f"   创建了 {len(particles)} 个质点")
+    for i, particle in enumerate(particles):
+        print(f"   质点{i+1}: {particle.name}, 质量: {particle.mass:.3f}")
+    print("   ✓ 配置创建成功")
+    
+    # 测试随机配置
+    print("\n5. 测试随机配置...")
+    random_particles = ThreeBodyConfig.create_random_config()
+    print(f"   创建了 {len(random_particles)} 个随机质点")
+    total_mass = sum(p.mass for p in random_particles)
+    print(f"   总质量: {total_mass:.3f}")
+    print("   ✓ 随机配置成功")
+    
+    print("\n" + "=" * 60)
+    print("所有基本测试通过!")
+    print("=" * 60)
+    
+    # 显示下一步操作
+    print("\n下一步:")
+    print("1. 安装依赖: pip install numpy matplotlib")
+    print("2. 运行完整测试: python three_body_problem/tests/test_solver.py")
+    print("3. 运行示例: python three_body_problem/run_example.py")
+    print("4. 查看文档: 阅读 three_body_problem/README.md")
+    
+except ImportError as e:
+    print(f"\n✗ 导入错误: {e}")
+    print("\n请确保:")
+    print("1. 在项目根目录运行此测试")
+    print("2. 安装了必要的依赖: pip install numpy matplotlib")
+    
+except Exception as e:
+    print(f"\n✗ 测试错误: {e}")
+    import traceback
+    traceback.print_exc()