Use ADTypeCheckContext with hmc tests

Author: mhauru

test/mcmc/hmc.jl

@@ -1,6 +1,7 @@
 module HMCTests
 
 using ..Models: gdemo_default
+using ..ADUtils: ADTypeCheckContext
 #using ..Models: gdemo
 using ..NumericalTests: check_gdemo, check_numerical
 using Distributions: Bernoulli, Beta, Categorical, Dirichlet, Normal, Wishart, sample
@@ -321,6 +322,15 @@ using Turing
         # KS will compare the empirical CDFs, which seems like a reasonable thing to do here.
         @test pvalue(ApproximateTwoSampleKSTest(vec(results), vec(results_prior))) > 0.001
     end
+
+    @testset "Check ADType" begin
+        alg = HMC(0.1, 10; adtype=adbackend)
+        m = DynamicPPL.contextualize(
+            gdemo_default, ADTypeCheckContext(adbackend, gdemo_default.context)
+        )
+        # These will error if the adbackend being used is not the one set.
+        sample(rng, m, alg, 10)
+    end
 end
 
 end

test/optimisation/Optimisation.jl

@@ -623,6 +623,7 @@ using Turing
             m = DynamicPPL.contextualize(
                 gdemo_default, ADTypeCheckContext(adbackend, gdemo_default.context)
             )
+            # These will error if the adbackend being used is not the one set.
             maximum_likelihood(m; adtype=adbackend)
             maximum_a_posteriori(m; adtype=adbackend)
         end