fix allocation tests

m3g · Mar 2, 2024 · 082e62d · 082e62d
1 parent afde299
commit 082e62d
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/test/allocations.jl b/test/allocations.jl
@@ -26,16 +26,18 @@
         ) samples = 1 evals = 1
         @test t_options.allocs == 0
 
-        protein = Selection(select(atoms, "protein"), nmols = 1)
+        protein_atoms = select(atoms, "protein")
+        protein = Selection(protein_atoms, nmols = 1)
         t_selection1 =
-            @benchmark Selection(select($atoms, "protein"), nmols = 1) samples = 1 evals = 1
+            @benchmark Selection(protein_atoms, nmols = 1) samples = 1 evals = 1
         @test t_selection1.allocs < 100 
 
-        tmao = Selection(select(atoms, "resname TMAO"), natomspermol = 14)
+        tmao_atoms = select(atoms, "resname TMAO")
+        tmao = Selection(tmao_atoms, natomspermol = 14)
         t_selection2 =
-            @benchmark Selection(select($atoms, "resname TMAO"), natomspermol = 14) samples =
+            @benchmark Selection(tmao_atoms, natomspermol = 14) samples =
                 1 evals = 1
-        @test t_selection2.allocs < 200000
+        @test t_selection2.allocs < 100
 
         trajfile = "$dir/trajectory.dcd" # because of the interpolation of @benchmark
         traj = Trajectory(trajfile, protein, tmao)