Some clean up of contexts

src/test_utils.jl

@@ -1129,7 +1129,9 @@ function test_context(context::DynamicPPL.AbstractContext, model::DynamicPPL.Mod
 
     # Setting the child context to a leaf should now change the leafcontext accordingly.
     context_with_new_leaf = DynamicPPL.setleafcontext(context, leafcontext_new)
-    @test childcontext(context_with_new_leaf) === leafcontext(context_with_new_leaf) === leafcontext_new
+    @test childcontext(context_with_new_leaf) ===
+        leafcontext(context_with_new_leaf) ===
+        leafcontext_new
 
     # Make sure that the we can evaluate the model with the context (i.e. that none of the tilde-functions are incorrectly overloaded).
     # The tilde-pipeline contains two different paths: with `SamplingContext` as a parent, and without it.