add tests for initialization methods

Red-Portal · Aug 11, 2024 · 183a387 · 183a387
1 parent a2b2f3a
commit 183a387
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 0 deletions.
diff --git a/test/multivariate.jl b/test/multivariate.jl
@@ -80,6 +80,32 @@ end
         # Gibbsian polar slice sampling
         GibbsPolarSlice(10),
     ]
+        @testset "initial_params" begin
+            model  = MultiModel(1.0, 1.0, [0.0])
+            θ, y  = MCMCTesting.sample_joint(Random.default_rng(), model)
+            model′ = AbstractMCMC.LogDensityModel(@set model.y = y)
+
+            θ0    = [1.0, 0.1]
+            chain = sample(
+                model,
+                sampler,
+                10;
+                initial_params=θ0,
+                progress=false,
+            )
+            @test first(chain).params == θ0
+        end
+
+        @testset "initial_sample" begin
+            rng = StableRNG(1)
+            model = MultiModel(1.0, 1.0, [0.0])
+            θ0 = SliceSampling.initial_sample(rng, model)
+
+            rng = StableRNG(1)
+            chain = sample(rng, model, sampler, 10; progress=false)
+            @test first(chain).params == θ0
+        end
+
         @testset "determinism" begin
             model  = MultiModel(1.0, 1.0, [0.0])
             θ, y  = MCMCTesting.sample_joint(Random.default_rng(), model)

diff --git a/test/univariate.jl b/test/univariate.jl
@@ -52,6 +52,32 @@ end
         SliceSteppingOut(1),
         SliceDoublingOut(1),
     ]
+        @testset "initialization" begin
+            model  = UniModel([0.0])
+            θ, y  = MCMCTesting.sample_joint(Random.default_rng(), model)
+            model′ = AbstractMCMC.LogDensityModel(@set model.y = y)
+
+            θ0    = [1.0]
+            chain = sample(
+                model,
+                sampler,
+                10;
+                initial_params=θ0,
+                progress=false,
+            )
+            @test first(chain).params == θ0
+        end
+
+        @testset "initial_sample" begin
+            rng = StableRNG(1)
+            model = UniModel([0.0])
+            θ0 = SliceSampling.initial_sample(rng, model)
+
+            rng = StableRNG(1)
+            chain = sample(rng, model, sampler, 10; progress=false)
+            @test first(chain).params == θ0
+        end
+
         @testset "determinism" begin
             model  = UniModel([0.0])
             θ, y  = MCMCTesting.sample_joint(Random.default_rng(), model)