Merge pull request #15 from abhijeetgangan/Tersoff_unit_test

Try #8
abhijeetgangan · May 22, 2024 · 21cc95b · 21cc95b
2 parents 7ef06d3 + bec2667
commit 21cc95b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/rigid_body_test.py b/tests/rigid_body_test.py
@@ -302,7 +302,7 @@ def total_energy(state, nbrs):
       return (energy_fn(pos, neighbor=nbrs) +
               simulate.kinetic_energy(state))
 
-    nbrs = neighbor_fn.allocate(body)
+    nbrs = neighbor_fn.allocate(body, extra_capacity=2)
     state = init_fn(key, body, 1e-3, mass=shape.mass(), neighbor=nbrs)
     E_initial = total_energy(state, nbrs)