a couple more ComponentArray tests

ElOceanografo · Dec 3, 2024 · 78cec89 · 78cec89
1 parent 4a0a1be
commit 78cec89
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -155,14 +155,18 @@ end
 
     marginalizer = LaplaceApprox(NelderMead(); adtype=AutoForwardDiff())
     mld = MarginalLogDensity(ld, u, iw, (), marginalizer)
+    mld_component = MarginalLogDensity(ld, u_component, [:w], (), marginalizer)
     L0 = mld(v, ())
-
+    @test L0 ≈ mld_component(u_component[[:v]])
+
     results = []
     for adtype in adtypes
         for solver in solvers
             marginalizer = LaplaceApprox(solver(), adtype=adtype)
             mld = MarginalLogDensity(ld, u, iw, (), marginalizer)
+            mld_component = MarginalLogDensity(ld, u_component, [:w], (), marginalizer)
             @test L0 ≈ mld(v, ())
+            @test L0 ≈ mld_component(u_component[[:v]])
             t0 = time()
                 for i in 1:100 
                     mld(v, ())