@@ -6,7 +6,7 @@ using DynamicPPL: Selector, reconstruct, invlink, CACHERESET,
6
6
set_flag!, unset_flag!, VarInfo, TypedVarInfo,
7
7
getlogp, setlogp!, resetlogp!, acclogp!, vectorize,
8
8
setorder!, updategid!
9
- using DynamicPPL
9
+ using DynamicPPL, LinearAlgebra
10
10
using Distributions
11
11
using ForwardDiff: Dual
12
12
using Test
@@ -167,32 +167,32 @@ include(dir*"/test/test_utils/AllUtils.jl")
167
167
meta = vi. metadata
168
168
model (vi, SampleFromUniform ())
169
169
170
- @test all (x -> ~ istrans (vi, x), meta. vns)
170
+ @test all (x -> istrans (vi, x), meta. vns)
171
171
alg = HMC (0.1 , 5 )
172
172
spl = Sampler (alg, model)
173
173
v = copy (meta. vals)
174
- link! (vi, spl)
175
- @test all (x -> istrans (vi, x), meta. vns)
176
174
invlink! (vi, spl)
177
175
@test all (x -> ~ istrans (vi, x), meta. vns)
178
- @test meta. vals == v
176
+ link! (vi, spl)
177
+ @test all (x -> istrans (vi, x), meta. vns)
178
+ @test norm (meta. vals - v) <= 1e-6
179
179
180
180
vi = TypedVarInfo (vi)
181
181
meta = vi. metadata
182
182
alg = HMC (0.1 , 5 )
183
183
spl = Sampler (alg, model)
184
- @test all (x -> ~ istrans (vi, x), meta. s. vns)
185
- @test all (x -> ~ istrans (vi, x), meta. m. vns)
186
- v_s = copy (meta. s. vals)
187
- v_m = copy (meta. m. vals)
188
- link! (vi, spl)
189
184
@test all (x -> istrans (vi, x), meta. s. vns)
190
185
@test all (x -> istrans (vi, x), meta. m. vns)
186
+ v_s = copy (meta. s. vals)
187
+ v_m = copy (meta. m. vals)
191
188
invlink! (vi, spl)
192
189
@test all (x -> ~ istrans (vi, x), meta. s. vns)
193
190
@test all (x -> ~ istrans (vi, x), meta. m. vns)
194
- @test meta. s. vals == v_s
195
- @test meta. m. vals == v_m
191
+ link! (vi, spl)
192
+ @test all (x -> istrans (vi, x), meta. s. vns)
193
+ @test all (x -> istrans (vi, x), meta. m. vns)
194
+ @test norm (meta. s. vals - v_s) <= 1e-6
195
+ @test norm (meta. m. vals - v_m) <= 1e-6
196
196
end
197
197
@testset " setgid!" begin
198
198
vi = VarInfo ()
0 commit comments