Format

TuringLang · Jan 9, 2025 · 268aa3a · 268aa3a
1 parent b5573a3
commit 268aa3a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/test/contexts.jl b/test/contexts.jl
@@ -243,7 +243,9 @@ end
             end
 
             @testset "Dict" begin
-                ctx = ConditionContext(Dict(@varname(x) => 1, @varname(y) => 2, @varname(z) => 3))
+                ctx = ConditionContext(
+                    Dict(@varname(x) => 1, @varname(y) => 2, @varname(z) => 3)
+                )
                 # Decondition all variables
                 @test decondition_context(ctx) isa DefaultContext
                 # Decondition only some variables
@@ -254,7 +256,8 @@ end
                 @test dctx isa ConditionContext
                 @test dctx.values == Dict(@varname(x) => 1)
                 # Decondition all variables manually
-                @test decondition_context(ctx, @varname(x), @varname(y), @varname(z)) isa DefaultContext
+                @test decondition_context(ctx, @varname(x), @varname(y), @varname(z)) isa
+                    DefaultContext
             end
         end
     end

diff --git a/test/model.jl b/test/model.jl
@@ -103,22 +103,22 @@ const GDEMO_DEFAULT = DynamicPPL.TestUtils.demo_assume_observe_literal()
     @testset "model conditioning with various arguments" begin
         @model function demo_condition()
             x ~ Normal()
-            y ~ Normal(x)
+            return y ~ Normal(x)
         end
         model = demo_condition()
         # Test that different syntaxes work and give the same underlying ConditionContext
         @testset "NamedTuple ConditionContext" begin
-            expected_values = (y = 2,)
+            expected_values = (y=2,)
             @test condition(model, (y=2,)).context.values == expected_values
-            @test condition(model, y=2).context.values == expected_values
             @test condition(model; y=2).context.values == expected_values
-            @test (model | (y = 2, )).context.values == expected_values
+            @test condition(model; y=2).context.values == expected_values
+            @test (model | (y=2,)).context.values == expected_values
         end
         @testset "AbstractDict ConditionContext" begin
             expected_values = Dict(@varname(y) => 2)
             @test condition(model, Dict(@varname(y) => 2)).context.values == expected_values
             @test condition(model, @varname(y) => 2).context.values == expected_values
-            @test (model | (@varname(y) => 2, )).context.values == expected_values
+            @test (model | (@varname(y) => 2,)).context.values == expected_values
         end
     end