@@ -116,19 +116,16 @@ using Turing
116
116
@testset " Prior" begin
117
117
N = 10_000
118
118
119
- # Note that all chains contain 3 values per sample: 2 variables + log probability
120
119
@testset " Single-threaded vanilla" begin
121
120
chains = sample (StableRNG (seed), gdemo_d (), Prior (), N)
122
121
@test chains isa MCMCChains. Chains
123
- @test size (chains) == (N, 3 , 1 )
124
122
@test mean (chains, :s ) ≈ 3 atol = 0.11
125
123
@test mean (chains, :m ) ≈ 0 atol = 0.1
126
124
end
127
125
128
126
@testset " Multi-threaded" begin
129
127
chains = sample (StableRNG (seed), gdemo_d (), Prior (), MCMCThreads (), N, 4 )
130
128
@test chains isa MCMCChains. Chains
131
- @test size (chains) == (N, 3 , 4 )
132
129
@test mean (chains, :s ) ≈ 3 atol = 0.11
133
130
@test mean (chains, :m ) ≈ 0 atol = 0.1
134
131
end
@@ -139,8 +136,9 @@ using Turing
139
136
)
140
137
@test chains isa Vector{<: NamedTuple }
141
138
@test length (chains) == N
142
- @test all (length (x) == 3 for x in chains)
143
139
@test all (haskey (x, :lp ) for x in chains)
140
+ @test all (haskey (x, :logprior ) for x in chains)
141
+ @test all (haskey (x, :loglikelihood ) for x in chains)
144
142
@test mean (x[:s ][1 ] for x in chains) ≈ 3 atol = 0.11
145
143
@test mean (x[:m ][1 ] for x in chains) ≈ 0 atol = 0.1
146
144
end
0 commit comments