From 79e072eddf0ec2a97ca455c27cb5ff6f2132bbab Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Tue, 22 Oct 2024 22:02:06 +0200
Subject: Differentiate Replicate

---
 test/Main.hs | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'test')

diff --git a/test/Main.hs b/test/Main.hs
index d90d9cd..f779352 100644
--- a/test/Main.hs
+++ b/test/Main.hs
@@ -229,8 +229,18 @@ tests :: IO Bool
 tests = checkParallel $ Group "AD"
   [("id", adTest $ fromNamed $ lambda #x $ body $ #x)
 
+  ,("idx0", adTest $ fromNamed $ lambda #x $ body $ idx0 #x)
+
   ,("sum-vec", adTest $ fromNamed $ lambda #x $ body $ idx0 (sum1i #x))
 
+  ,("sum-replicate", adTest $ fromNamed $ lambda #x $ body $
+      idx0 $ sum1i $ replicate1i 10 #x)
+
+  ,("pairs", adTest $ fromNamed $ lambda #x $ lambda #y $ body $
+      let_ #p (pair #x #y) $
+      let_ #q (pair (snd_ #p * fst_ #p + #y) #x) $
+        fst_ #q * #x + snd_ #q * fst_ #p)
+
   ,("build0", adTest $ fromNamed $ lambda @(TArr N0 _) #x $ body $
       idx0 $
         build SZ (shape #x) $ #idx :-> #x ! #idx)
-- 
cgit v1.2.3-70-g09d2