From aa82c2430bea33c7ac55787d971ccdfc7604e44e Mon Sep 17 00:00:00 2001 From: David Widmann Date: Wed, 1 Jun 2022 13:26:47 +0200 Subject: [PATCH] Update stepper.jl --- test/stepper.jl | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/test/stepper.jl b/test/stepper.jl index bc0ea8b2..80143344 100644 --- a/test/stepper.jl +++ b/test/stepper.jl @@ -48,8 +48,8 @@ chain = sample( MyModel(), MySampler(), N; discard_initial=discard_initial, progress=false ) - @test all(as[i] === chain[i].a for i in 1:N) - @test all(bs[i] === chain[i].b for i in 1:N) + @test all(as[i] == chain[i].a for i in 1:N) + @test all(bs[i] == chain[i].b for i in 1:N) end @testset "Thin chain by a factor of `thinning`" begin @@ -68,7 +68,8 @@ # Repeat sampling with `sample`. Random.seed!(1234) chain = sample(MyModel(), MySampler(), N; thinning=thinning, progress=false) - @test all(as[i] === chain[i].a for i in 1:N) - @test all(bs[i] === chain[i].b for i in 1:N) + @test as[1] === chain[1].a === missing + @test all(as[i] == chain[i].a for i in 2:N) + @test all(bs[i] == chain[i].b for i in 1:N) end end