test the context interface for DebugContext on multiple models

Author: torfjelde

test/debug_utils.jl

@@ -1,9 +1,9 @@
 @testset "check_model" begin
     @testset "context interface" begin
-        # HACK: Require a model to instantiate it, so let's just grab one.
-        model = first(DynamicPPL.TestUtils.DEMO_MODELS)
-        context = DynamicPPL.DebugUtils.DebugContext(model)
-        DynamicPPL.TestUtils.test_context(context, model)
+        @testset "$(model.f)" for model in DynamicPPL.TestUtils.DEMO_MODELS
+            context = DynamicPPL.DebugUtils.DebugContext(model)
+            DynamicPPL.TestUtils.test_context(context, model)
+        end
     end
 
     @testset "$(model.f)" for model in DynamicPPL.TestUtils.DEMO_MODELS
@@ -14,7 +14,7 @@
         # Check that the trace contains all the variables in the model.
         varnames_in_trace = DynamicPPL.DebugUtils.varnames_in_trace(trace)
         for vn in DynamicPPL.TestUtils.varnames(model)
-            @test vn in varnames_in_trace
+            @test vn in varnames_in_traces
         end
 
         # Quick checks for `show` of trace.