def benchmark():
particles = [
particle(uniform(-1.0, 1.0), uniform(-1.0, 1.0), uniform(-1.0, 1.0))
for i in range(1000)
] #生成1000个随机测试用例
simulator = ParticleSimulator(particles)
simulator.evolve(0.1)
def test_evolve():
particles = [
particle(0.3, 0.5, 1),
particle(0.0, -0.5, -1),
particle(-0.1, -0.4, 3)
]
simulator = ParticleSimulator(particles)
simulator.evolve(0.1)
p0, p1, p2 = particles
def fequal(a, b, eps=1e-5):
return abs(a - b) < eps
assert fequal(p0.x, 0.210269, eps=1e-5)
assert fequal(p0.y, 0.543863, eps=1e-5)
assert fequal(p1.x, -0.099334)
assert fequal(p1.y, -0.490034)
assert fequal(p2.x, 0.191358)
assert fequal(p2.y, -0.365227)