From f610514e014f26497b05cc9eeff53995faa1e8f3 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Wed, 18 Jun 2025 11:25:13 +0200
Subject: test: Test both default and accum configs

---
 test/Main.hs | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/test/Main.hs b/test/Main.hs
index 1b83a2e..79051de 100644
--- a/test/Main.hs
+++ b/test/Main.hs
@@ -305,7 +305,9 @@ adTestGen name expr envGenerator =
      testGroupCollapse name
        [adTestGenPrimal env envGenerator expr exprS primalfun primalSfun
        ,adTestGenFwd env envGenerator exprS
-       ,adTestGenChad env envGenerator expr exprS primalSfun]
+       ,testGroup "chad"
+          [adTestGenChad "default" defaultConfig env envGenerator expr exprS primalSfun
+          ,adTestGenChad "accum" (chcSetAccum defaultConfig) env envGenerator expr exprS primalSfun]]
 
 adTestGenPrimal :: SList STy env -> Gen (SList Value env)
                 -> Ex env R -> Ex env R
@@ -336,19 +338,19 @@ adTestGenFwd env envGenerator exprS =
       diff outDNI1 (closeIsh' 1e-8) outDNC1
       diff outDNI2 (closeIsh' 1e-8) outDNC2
 
-adTestGenChad :: forall env. SList STy env -> Gen (SList Value env)
+adTestGenChad :: forall env. String -> CHADConfig -> SList STy env -> Gen (SList Value env)
               -> Ex env R -> Ex env R
               -> (SList Value env -> IO Double)
               -> TestTree
-adTestGenChad env envGenerator expr exprS primalSfun | Dict <- envKnown env =
-  let dtermChad0 = ELet ext (EConst ext STF64 1.0) $ chad' defaultConfig env expr
+adTestGenChad testname config env envGenerator expr exprS primalSfun | Dict <- envKnown env =
+  let dtermChad0 = ELet ext (EConst ext STF64 1.0) $ chad' config env expr
       dtermChadS = simplifyFix dtermChad0
-      dtermSChad0 = ELet ext (EConst ext STF64 1.0) $ chad' defaultConfig env exprS
+      dtermSChad0 = ELet ext (EConst ext STF64 1.0) $ chad' config env exprS
       dtermSChadS = simplifyFix dtermSChad0
   in
   withResource (makeFwdADArtifactCompile env exprS) (\_ -> pure ()) $ \fwdartifactC ->
   withCompiled env (simplifyFix (unMonoid dtermSChadS)) $ \dcompSChadS ->
-    testProperty "chad" $ property $ do
+    testProperty testname $ property $ do
       annotate (concat (unSList (\t -> ppSTy 0 t ++ " -> ") env) ++ ppSTy 0 (typeOf expr))
 
       -- pack Text for less GC pressure (these values are retained for some reason)
-- 
cgit v1.2.3-70-g09d2