adjusted allocation tests

m3g · Dec 1, 2023 · 837d53d · 837d53d
1 parent 93b7646
commit 837d53d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/allocations.jl b/test/allocations.jl
@@ -17,15 +17,15 @@
             seed = 321,
             StableRNG = true,
         )
-        t = @ballocated Options(
+        t = @benchmark Options(
             lastframe = 1,
             seed = 321,
             StableRNG = true,
             nthreads = 1,
             silent = true,
             frame_weights = $(Float64[]),
         ) samples = 1 evals = 1
-        @test t == 0
+        @test t.allocs == 0
 
         protein = Selection(select(atoms, "protein"), nmols = 1)
         t_selection1 =
@@ -46,7 +46,7 @@
 
         R = Result(traj, options)
         t_result = @benchmark Result($traj, $options) samples = 1 evals = 1
-        @test t_result.allocs < 100
+        @test t_result.allocs < 200
 
         CM.opentraj!(traj)
         CM.firstframe!(traj)