add some new tests

TuringLang · Oct 22, 2024 · 933efe5 · 933efe5
1 parent 161efff
commit 933efe5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/abstractmcmc.jl b/test/abstractmcmc.jl
@@ -27,6 +27,10 @@ using Statistics: mean
         θ = AbstractMCMC.getparams(s)
         @test θ == t.z.θ
         @test AbstractMCMC.setparams!!(s, θ) == s
+
+        new_θ = randn(rng, 2)
+        new_state = AbstractMCMC.setparams!!(s, new_θ)
+        @test AbstractMCMC.getparams(new_state) == new_θ
     end
 
     samples_nuts = AbstractMCMC.sample(