Skip to content

Commit 9eca78d

Browse files
committed
Fix some tests
1 parent 71300d6 commit 9eca78d

File tree

1 file changed

+7
-13
lines changed

1 file changed

+7
-13
lines changed

test/sampler.jl

Lines changed: 7 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -81,10 +81,8 @@
8181
model = coinflip()
8282
sampler = Sampler(alg)
8383
lptrue = logpdf(Binomial(25, 0.2), 10)
84-
let inits = (; p=0.2)
85-
chain = sample(
86-
model, sampler, 1; initial_params=ParamsInit(inits), progress=false
87-
)
84+
let inits = ParamsInit((; p=0.2))
85+
chain = sample(model, sampler, 1; initial_params=inits, progress=false)
8886
@test chain[1].metadata.p.vals == [0.2]
8987
@test getlogjoint(chain[1]) == lptrue
9088

@@ -111,10 +109,8 @@
111109
end
112110
model = twovars()
113111
lptrue = logpdf(InverseGamma(2, 3), 4) + logpdf(Normal(0, 2), -1)
114-
for inits in ([4, -1], (; s=4, m=-1))
115-
chain = sample(
116-
model, sampler, 1; initial_params=ParamsInit(inits), progress=false
117-
)
112+
let inits = ParamsInit((; s=4, m=-1))
113+
chain = sample(model, sampler, 1; initial_params=inits, progress=false)
118114
@test chain[1].metadata.s.vals == [4]
119115
@test chain[1].metadata.m.vals == [-1]
120116
@test getlogjoint(chain[1]) == lptrue
@@ -126,7 +122,7 @@
126122
MCMCThreads(),
127123
1,
128124
10;
129-
initial_params=fill(ParamsInit(inits), 10),
125+
initial_params=fill(inits, 10),
130126
progress=false,
131127
)
132128
for c in chains
@@ -137,10 +133,8 @@
137133
end
138134

139135
# set only m = -1
140-
for inits in ((; s=missing, m=-1), (; m=-1))
141-
chain = sample(
142-
model, sampler, 1; initial_params=ParamsInit(inits), progress=false
143-
)
136+
for inits in (ParamsInit((; s=missing, m=-1)), ParamsInit((; m=-1)))
137+
chain = sample(model, sampler, 1; initial_params=inits, progress=false)
144138
@test !ismissing(chain[1].metadata.s.vals[1])
145139
@test chain[1].metadata.m.vals == [-1]
146140

0 commit comments

Comments
 (0)