diff --git a/ext/DynamicPPLTestExt/contexts.jl b/ext/DynamicPPLTestExt/contexts.jl index 108af567c..2f630dca4 100644 --- a/ext/DynamicPPLTestExt/contexts.jl +++ b/ext/DynamicPPLTestExt/contexts.jl @@ -40,7 +40,9 @@ function DynamicPPL.setchildcontext( ) return DynamicPPL.TestUtils.TestLogModifyingChildContext(context.mod, child) end -function DynamicPPL.tilde_assume(context::DynamicPPL.TestUtils.TestLogModifyingChildContext, right, vn, vi) +function DynamicPPL.tilde_assume( + context::DynamicPPL.TestUtils.TestLogModifyingChildContext, right, vn, vi +) value, logp, vi = DynamicPPL.tilde_assume(context.context, right, vn, vi) return value, logp * context.mod, vi end @@ -50,7 +52,9 @@ function DynamicPPL.dot_tilde_assume( value, logp, vi = DynamicPPL.dot_tilde_assume(context.context, right, left, vn, vi) return value, logp * context.mod, vi end -function DynamicPPL.tilde_observe(context::DynamicPPL.TestUtils.TestLogModifyingChildContext, right, left, vi) +function DynamicPPL.tilde_observe( + context::DynamicPPL.TestUtils.TestLogModifyingChildContext, right, left, vi +) logp, vi = DynamicPPL.tilde_observe(context.context, right, left, vi) return logp * context.mod, vi end diff --git a/ext/DynamicPPLTestExt/sampler.jl b/ext/DynamicPPLTestExt/sampler.jl index 1134d4121..97c3b391f 100644 --- a/ext/DynamicPPLTestExt/sampler.jl +++ b/ext/DynamicPPLTestExt/sampler.jl @@ -81,6 +81,8 @@ Test that `sampler` produces the correct marginal posterior means on all models As of right now, this is just an alias for [`test_sampler_on_demo_models`](@ref). """ -function DynamicPPL.TestUtils.test_sampler_continuous(sampler::AbstractMCMC.AbstractSampler, args...; kwargs...) +function DynamicPPL.TestUtils.test_sampler_continuous( + sampler::AbstractMCMC.AbstractSampler, args...; kwargs... +) return test_sampler_on_demo_models(sampler, args...; kwargs...) end diff --git a/ext/DynamicPPLTestExt/varinfo.jl b/ext/DynamicPPLTestExt/varinfo.jl index 97c19ac74..a82bc782e 100644 --- a/ext/DynamicPPLTestExt/varinfo.jl +++ b/ext/DynamicPPLTestExt/varinfo.jl @@ -8,7 +8,9 @@ Test that `vi[vn]` corresponds to the correct value in `vals` for every `vn` in `vns`. """ -function DynamicPPL.TestUtils.test_values(vi::AbstractVarInfo, vals::NamedTuple, vns; compare=isequal, kwargs...) +function DynamicPPL.TestUtils.test_values( + vi::AbstractVarInfo, vals::NamedTuple, vns; compare=isequal, kwargs... +) for vn in vns @test compare(vi[vn], get(vals, vn); kwargs...) end diff --git a/src/test_utils.jl b/src/test_utils.jl index 5200b2535..036494bd5 100644 --- a/src/test_utils.jl +++ b/src/test_utils.jl @@ -10,7 +10,6 @@ using Bijectors: Bijectors include("test_utils/model_interface.jl") include("test_utils/models.jl") - ############################################################## # The remainder of this file contains skeleton implementations for # DynamicPPLTestExt