Add another test

Author: penelopeysm

test/ext/DynamicPPLMCMCChainsExt.jl

@@ -17,7 +17,7 @@ using LinearAlgebra: I
     @testset "varinfos_to_chains" begin
         @model function f()
             x ~ Normal()
-            y ~ Normal(x)
+            y := x + 1
             return z ~ MvNormal(zeros(3), I)
         end
 
@@ -35,6 +35,17 @@ using LinearAlgebra: I
             @test c.info.varname_to_symbol[@varname(z[3])] == Symbol("z[3]")
         end
 
+        @testset "vector, without include_colon_eq" begin
+            vis = [VarInfo(model) for _ in 1:50]
+            c = DynamicPPL.varinfos_to_chains(MCMCChains.Chains, model, vis, false)
+            @test c isa MCMCChains.Chains
+            @test size(c) == (50, 4, 1)
+            @test c.info.varname_to_symbol[@varname(x)] == :x
+            @test c.info.varname_to_symbol[@varname(z[1])] == Symbol("z[1]")
+            @test c.info.varname_to_symbol[@varname(z[2])] == Symbol("z[2]")
+            @test c.info.varname_to_symbol[@varname(z[3])] == Symbol("z[3]")
+        end
+
         @testset "matrix" begin
             vis = [VarInfo(model) for _ in 1:50, _ in 1:3]
             c = DynamicPPL.varinfos_to_chains(MCMCChains.Chains, model, vis)