From 14124168205a0f171797621c66de9bc63c6d5f40 Mon Sep 17 00:00:00 2001 From: Penelope Yong Date: Wed, 27 Nov 2024 16:57:02 +0000 Subject: [PATCH] Fix test for invlink on reduced space --- test/dynamicppl/varinfo.jl | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/test/dynamicppl/varinfo.jl b/test/dynamicppl/varinfo.jl index 54f3baf1d..28826f9f0 100644 --- a/test/dynamicppl/varinfo.jl +++ b/test/dynamicppl/varinfo.jl @@ -58,7 +58,7 @@ using Turing @test all(x -> DynamicPPL.istrans(vi, x), meta.vns) DynamicPPL.invlink!!(vi, spl, model) @test all(x -> !DynamicPPL.istrans(vi, x), meta.vns) - @test meta.vals == v + @test meta.vals ≈ v atol=1e-10 vi = DynamicPPL.TypedVarInfo(vi) meta = vi.metadata @@ -74,19 +74,21 @@ using Turing DynamicPPL.invlink!!(vi, spl, model) @test all(x -> !DynamicPPL.istrans(vi, x), meta.s.vns) @test all(x -> !DynamicPPL.istrans(vi, x), meta.m.vns) - @test meta.s.vals == v_s - @test meta.m.vals == v_m + @test meta.s.vals ≈ v_s atol=1e-10 + @test meta.m.vals ≈ v_m atol=1e-10 # Transforming only a subset of the variables - DynamicPPL.link!(vi, spl, Val((:m,))) + spl = DynamicPPL.Sampler(HMC(0.1, 5, :m), model) + DynamicPPL.link!!(vi, spl, model) @test all(x -> !DynamicPPL.istrans(vi, x), meta.s.vns) @test all(x -> DynamicPPL.istrans(vi, x), meta.m.vns) - DynamicPPL.invlink!(vi, spl, Val((:m,))) + DynamicPPL.invlink!!(vi, spl, model) @test all(x -> !DynamicPPL.istrans(vi, x), meta.s.vns) @test all(x -> !DynamicPPL.istrans(vi, x), meta.m.vns) - @test meta.s.vals == v_s - @test meta.m.vals == v_m + @test meta.s.vals ≈ v_s atol=1e-10 + @test meta.m.vals ≈ v_m atol=1e-10 end + @testset "orders" begin csym = gensym() # unique per model vn_z1 = @varname z[1]