Add equivalence integration test for inlined functions

2024-11-30 03:34:51 +03:00 · 2023-02-10 16:25:32 -08:00 · 2023-02-10 16:25:32 -08:00 · 1612bcccba
commit 1612bcccba
parent 0be98e30b7
1 changed files with 82 additions and 0 deletions
--- a/tests/tests/execution/flattened_function_and_inline_matches.leo
+++ b/tests/tests/execution/flattened_function_and_inline_matches.leo
@ -0,0 +1,82 @@
+/*
+namespace: Execute
+expectation: Pass
+cases:
+    bar:
+    - input: ["true", "true", "1u8", "0u8"]
+    - input: ["true", "false", "1u8", "1u8"]
+    - input: ["false", "true", "2u8", "1u8"]
+    - input: ["false", "false", "2u8", "2u8"]
+    blar:
+    - input: ["true", "true", "1u8", "0u8"]
+    - input: ["true", "false", "1u8", "1u8"]
+    - input: ["false", "true", "2u8", "1u8"]
+    - input: ["false", "false", "2u8", "2u8"]
+*/
+
+// In this test, we expect bar and blar to produce the same result for all inputs.
+
+program test.aleo {
+    struct Extra {
+        c: u8,
+    }
+
+    struct Data {
+        a: u8,
+        b: u8,
+        c: Extra,
+    }
+
+    inline foo(a: u8, b: u8) -> (u8, u8, Data) {
+        let extra: Extra = Extra { c: a };
+        let data: Data = Data { a: a, b: b, c: extra };
+        if (a == b) {
+            return (a, b, data);
+        }
+        let c: u8 = a + b;
+        let d: u8 = a - b;
+
+        return (c, d, data);
+    }
+
+    function floo(a: u8, b: u8) -> (u8, u8, Data) {
+        let extra: Extra = Extra { c: a };
+        let data: Data = Data { a: a, b: b, c: extra };
+        if (a == b) {
+            return (a, b, data);
+        }
+        let c: u8 = a + b;
+        let d: u8 = a - b;
+
+        return (c, d, data);
+    }
+
+    transition bar(flag1: bool, flag2: bool, a: u8, b: u8) -> (u8, u8, Data) {
+        let start: (u8, u8, Data) = foo(a, b);
+        if flag1 {
+            start = foo(start.0, start.2.c.c);
+        } else {
+            if flag2 {
+                start = foo(start.1, start.2.b);
+            } else {
+                start = foo(start.2.a, start.1);
+            }
+        }
+        return start;
+    }
+
+    transition blar(flag1: bool, flag2: bool, a: u8, b: u8) -> (u8, u8, Data) {
+        let start: (u8, u8, Data) = floo(a, b);
+        if flag1 {
+            start = floo(start.0, start.2.c.c);
+        } else {
+            if flag2 {
+                start = floo(start.1, start.2.b);
+            } else {
+                start = floo(start.2.a, start.1);
+            }
+        }
+        return start;
+    }
+
+}