Add a factorial test for loop

hylang · Nov 22, 2024 · 0b4dbe7 · 0b4dbe7
1 parent 7ef8e0a
commit 0b4dbe7
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/tests/test_loop.hy b/tests/test_loop.hy
@@ -1,11 +1,21 @@
 (require
   hyrule [loop])
 (import
+  math
   sys
-  hyrule [inc dec]
+  hyrule [inc dec recur]
   pytest)
 
 
+(defn test-factorial []
+  (assert (=
+    (loop [[i 5] [acc 1]]
+      (if (= i 0)
+       acc
+       (recur (dec i) (* acc i))))
+    (math.factorial 5))))
+
+
 (defn test-tco-sum []
 
   ; This plain old tail-recursive function should exceed Python's