trace into functions in testcases

Author: KDr2

test/Project.toml

@@ -12,6 +12,7 @@ DynamicPPL = "366bfd00-2699-11ea-058f-f148b4cae6d8"
 FiniteDifferences = "26cc04aa-876d-5657-8c51-4c34ba976000"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 GalacticOptim = "a75be94c-b780-496d-a8a9-0878b188d577"
+Libtask = "6f1fad26-d15e-5dc8-ae53-837a1d7b8c9f"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MCMCChains = "c7f686f2-ff18-58e9-bc7b-31028e88f75d"
 Memoization = "6fafb56a-5788-4b4e-91ca-c0cea6611c73"

test/test_utils/models.jl

@@ -7,6 +7,9 @@
   return s, m
 end
 
+using Libtask
+Libtask.trace_into(::typeof(gdemo)) = true
+
 @model gdemo_d() = begin
   s ~ InverseGamma(2, 3)
   m ~ Normal(0, sqrt(s))
@@ -16,6 +19,7 @@ end
 end
 
 gdemo_default = gdemo_d()
+Libtask.trace_into(::typeof(gdemo_d)) = true
 
 @model MoGtest(D) = begin
     mu1 ~ Normal(1, 1)
@@ -48,6 +52,7 @@ gdemo_default = gdemo_d()
 end
 
 MoGtest_default = MoGtest([1.0 1.0 4.0 4.0])
+Libtask.trace_into(::typeof(MoGtest)) = true
 
 # Declare empty model to make the Sampler constructor work.
 @model empty_model() = begin x = 1; end