2 files changed, 41 insertions, 28 deletions

diff --git a/tests/smoke.rs b/tests/smoke.rs
index 059cf9f..a9ee367 100644
--- a/tests/smoke.rs
+++ b/tests/smoke.rs
@@ -1,33 +1,13 @@
 #![feature(type_alias_impl_trait)]
 
-pub mod helper_functions {
-    use currying::curry;
-
-    #[curry]
-    pub fn product(x: u32, y: u32) -> u32 {
-        return x * y;
-    }
-    #[curry]
-    pub fn add(x: u32, y: u32) -> u32 {
-        x + y
-    }
+#[cutlass::curry]
+fn add(x: u32, y: u32, z: u32) -> u32 {
+    return x + y + z;
 }
 
-#[cfg(test)]
-mod tests {
-    use super::helper_functions::{add, product};
-    #[test]
-    fn test_product() {
-        assert_eq!(
-            (1..=3).map(product(5)).collect::<Vec<u32>>(),
-            vec![5, 10, 15]
-        );
-    }
-
-    #[test]
-    fn test_add() {
-        let increment = add(1);
-        let v: Vec<u32> = (1..=3).map(increment).collect();
-        assert_eq!(v, vec![2, 3, 4]);
-    }
+#[test]
+fn add_works() {
+    let plus_3 = add(1)(2);
+    let v: Vec<u32> = (1..=3).map(plus_3).collect();
+    assert_eq!(v, vec![4, 5, 6]);
 }
diff --git a/tests/with_map.rs b/tests/with_map.rs
new file mode 100644
index 0000000..141ee5d
--- /dev/null
+++ b/tests/with_map.rs
@@ -0,0 +1,33 @@
+#![feature(type_alias_impl_trait)]
+
+pub mod helper_functions {
+    use cutlass::curry;
+
+    #[curry]
+    pub fn product(x: u32, y: u32) -> u32 {
+        return x * y;
+    }
+    #[curry]
+    pub fn add(x: u32, y: u32) -> u32 {
+        x + y
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::helper_functions::{add, product};
+    #[test]
+    fn test_product() {
+        assert_eq!(
+            (1..=3).map(product(5)).collect::<Vec<u32>>(),
+            vec![5, 10, 15]
+        );
+    }
+
+    #[test]
+    fn test_add() {
+        let increment = add(1);
+        let v: Vec<u32> = (1..=3).map(increment).collect();
+        assert_eq!(v, vec![2, 3, 4]);
+    }
+}