fix test rng

Author: prbzrg

test/ci_tests/regression_tests.jl

@@ -1,9 +1,8 @@
 Test.@testset "Regression Tests" begin
+    rng = StableRNGs.StableRNG(1)
     ndata = 2^10
     ndimension = 1
     data_dist = Distributions.Beta{Float32}(2.0f0, 4.0f0)
-    r = rand(data_dist, ndimension, ndata)
-    r = convert.(Float32, r)
 
     nvars = size(r, 1)
     naugs = nvars
@@ -22,13 +21,16 @@ Test.@testset "Regression Tests" begin
         λ₁ = 1.0f-2,
         λ₂ = 1.0f-2,
         λ₃ = 1.0f-2,
+        rng,
         sol_kwargs = (;
             save_everystep = false,
             alg = OrdinaryDiffEqDefault.DefaultODEAlgorithm(),
             sensealg = SciMLSensitivity.InterpolatingAdjoint(),
         ),
     )
 
+    r = rand(icnf.rng, data_dist, ndimension, ndata)
+    r = convert.(Float32, r)
     df = DataFrames.DataFrame(transpose(r), :auto)
     model = ContinuousNormalizingFlows.ICNFModel(
         icnf;

test/ci_tests/speed_tests.jl

@@ -21,11 +21,10 @@ Test.@testset "Speed Tests" begin
     Test.@testset "$compute_mode" for compute_mode in compute_modes
         @show compute_mode
 
+        rng = StableRNGs.StableRNG(1)
         ndata = 2^10
         ndimension = 1
         data_dist = Distributions.Beta{Float32}(2.0f0, 4.0f0)
-        r = rand(data_dist, ndimension, ndata)
-        r = convert.(Float32, r)
 
         nvars = size(r, 1)
         naugs = nvars
@@ -44,15 +43,17 @@ Test.@testset "Speed Tests" begin
             λ₁ = 1.0f-2,
             λ₂ = 1.0f-2,
             λ₃ = 1.0f-2,
+            rng,
             sol_kwargs = (;
                 save_everystep = false,
                 alg = OrdinaryDiffEqDefault.DefaultODEAlgorithm(),
                 sensealg = SciMLSensitivity.InterpolatingAdjoint(),
             ),
         )
 
+        r = rand(icnf.rng, data_dist, ndimension, ndata)
+        r = convert.(Float32, r)
         df = DataFrames.DataFrame(transpose(r), :auto)
-
         model = ContinuousNormalizingFlows.ICNFModel(
             icnf;
             batchsize = 0,

test/runtests.jl

@@ -31,8 +31,7 @@ if GROUP == "All"
     Logging.global_logger(debuglogger)
 end
 
-Test.@testset verbose = true showtiming = true failfast = false rng =
-    StableRNGs.StableRNG(1) "Overall" begin
+Test.@testset verbose = true showtiming = true failfast = false "Overall" begin
     if GROUP == "All" || GROUP in ["SmokeXOut", "SmokeXIn", "SmokeXYOut", "SmokeXYIn"]
         include(joinpath("ci_tests", "smoke_tests.jl"))
     end