From 1612bcccba1876ce9fbc15371554ff30031cefd8 Mon Sep 17 00:00:00 2001
From: d0cd <pranavsaig@gmail.com>
Date: Fri, 10 Feb 2023 16:25:32 -0800
Subject: [PATCH] Add equivalence integration test for inlined functions

---
 .../flattened_function_and_inline_matches.leo | 82 +++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 tests/tests/execution/flattened_function_and_inline_matches.leo

diff --git a/tests/tests/execution/flattened_function_and_inline_matches.leo b/tests/tests/execution/flattened_function_and_inline_matches.leo
new file mode 100644
index 0000000000..1713215422
--- /dev/null
+++ 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;
+    }
+
+}