1 files changed, 44 insertions, 2 deletions
diff --git a/test/unheard/cycles_test.clj b/test/unheard/cycles_test.clj
index ef9aca7..0f55a2c 100644
--- a/test/unheard/cycles_test.clj
+++ b/test/unheard/cycles_test.clj
@@ -1,6 +1,6 @@
 (ns unheard.cycles-test
   (:require [clojure.test :refer [deftest is testing]]
-            [unheard.cycles :refer [l f p unfold]]))
+            [unheard.cycles :refer [l f p rate unfold]]))
 
 (deftest unfold-tests
   (testing "single scalar"
@@ -86,4 +86,46 @@
   (testing "parallel with different cycle lengths"
     (is (= [[0 1 :a] [0 1 :b]
             [1 2 :a] [1 2 :c]]
-           (unfold 1 (p :a (f :b :c)))))))
+           (unfold 1 (p :a (f :b :c))))))
+
+  (testing "rate 2 - doubles speed of scalar"
+    (is (= [[0 1/2 :a] [1/2 1 :a]]
+           (unfold 1 (rate 2 :a)))))
+
+  (testing "rate 1/2 - halves speed of scalar"
+    (is (= [[0 2 :a]]
+           (unfold 1 (rate 1/2 :a)))))
+
+  (testing "rate 2 - doubles speed of list"
+    (is (= [[0 1/6 :a] [1/6 1/3 :b] [1/3 1/2 :c]
+            [1/2 2/3 :a] [2/3 5/6 :b] [5/6 1 :c]]
+           (unfold 1 (rate 2 (l :a :b :c))))))
+
+  (testing "rate 1/2 - halves speed of list"
+    (is (= [[0 1 :a] [1 2 :b]]
+           (unfold 1 (rate 1/2 (l :a :b))))))
+
+  (testing "rate 2 - doubles speed of fork"
+    ;; Fork has 3 children, cycle count is 3
+    ;; rate 2 makes it repeat 2x, so 6 total cycles
+    (is (= [[0 1/6 :a] [1/6 1/3 :b] [1/3 1/2 :c]
+            [1/2 2/3 :a] [2/3 5/6 :b] [5/6 1 :c]
+            [1 7/6 :a] [7/6 4/3 :b] [4/3 3/2 :c]
+            [3/2 5/3 :a] [5/3 11/6 :b] [11/6 2 :c]
+            [2 13/6 :a] [13/6 7/3 :b] [7/3 5/2 :c]
+            [5/2 8/3 :a] [8/3 17/6 :b] [17/6 3 :c]]
+           (unfold 1 (rate 2 (f :a :b :c))))))
+
+  (testing "rate with parallel"
+    ;; Parallel has cycle 1, rate 2 repeats it twice
+    (is (= [[0 1/2 :a] [0 1/2 :b]
+            [1/2 1 :a] [1/2 1 :b]]
+           (unfold 1 (rate 2 (p :a :b))))))
+
+  (testing "rate inside list"
+    (is (= [[0 1/2 :x] [1/2 3/4 :y] [3/4 1 :y]]
+           (unfold 1 (l :x (rate 2 :y))))))
+
+  (testing "nested rates"
+    (is (= [[0 1/4 :a] [1/4 1/2 :a] [1/2 3/4 :a] [3/4 1 :a]]
+           (unfold 1 (rate 2 (rate 2 :a)))))))