From 9b6d4e20c76ace7537cae5c16b6358db53235889 Mon Sep 17 00:00:00 2001
From: Kasey White <kwhitemsg@gmail.com>
Date: Thu, 22 Sep 2022 03:13:30 -0400
Subject: [PATCH] check exunits for eval tests

---
 crates/uplc/src/tx/tests.rs | 76 +++++++++++++++++++++++++++++++++++++
 1 file changed, 76 insertions(+)

diff --git a/crates/uplc/src/tx/tests.rs b/crates/uplc/src/tx/tests.rs
index a7a39266..f456dbca 100644
--- a/crates/uplc/src/tx/tests.rs
+++ b/crates/uplc/src/tx/tests.rs
@@ -499,6 +499,25 @@ fn test_eval_1() {
             .unwrap();
 
             println!("{:?}", redeemers.len());
+
+            let total_budget_used =
+                redeemers
+                    .iter()
+                    .fold(ExBudget { mem: 0, cpu: 0 }, |accum, curr| ExBudget {
+                        mem: accum.mem + curr.ex_units.mem as i64,
+                        cpu: accum.cpu + curr.ex_units.steps as i64,
+                    });
+
+            println!("{:?}", total_budget_used);
+
+            assert_eq!(
+                total_budget_used,
+                // Numbers came uplc evaluate
+                ExBudget {
+                    cpu: 210171376,
+                    mem: 721484
+                }
+            );
         }
         _ => unreachable!(),
     };
@@ -584,6 +603,25 @@ fn test_eval_2() {
             .unwrap();
 
             println!("{:?}", redeemers.len());
+
+            let total_budget_used =
+                redeemers
+                    .iter()
+                    .fold(ExBudget { mem: 0, cpu: 0 }, |accum, curr| ExBudget {
+                        mem: accum.mem + curr.ex_units.mem as i64,
+                        cpu: accum.cpu + curr.ex_units.steps as i64,
+                    });
+
+            println!("{:?}", total_budget_used);
+
+            assert_eq!(
+                total_budget_used,
+                // Numbers came uplc evaluate
+                ExBudget {
+                    cpu: 183304563,
+                    mem: 638338
+                }
+            );
         }
         _ => unreachable!(),
     };
@@ -828,6 +866,25 @@ fn test_eval_3() {
             .unwrap();
 
             println!("{:?}", redeemers.len());
+
+            let total_budget_used =
+                redeemers
+                    .iter()
+                    .fold(ExBudget { mem: 0, cpu: 0 }, |accum, curr| ExBudget {
+                        mem: accum.mem + curr.ex_units.mem as i64,
+                        cpu: accum.cpu + curr.ex_units.steps as i64,
+                    });
+
+            println!("{:?}", total_budget_used);
+
+            assert_eq!(
+                total_budget_used,
+                // Numbers came uplc evaluate
+                ExBudget {
+                    cpu: 177894084,
+                    mem: 566628
+                }
+            );
         }
         _ => unreachable!(),
     };
@@ -991,6 +1048,25 @@ fn test_eval_5() {
             .unwrap();
 
             println!("{:?}", redeemers.len());
+
+            let total_budget_used =
+                redeemers
+                    .iter()
+                    .fold(ExBudget { mem: 0, cpu: 0 }, |accum, curr| ExBudget {
+                        mem: accum.mem + curr.ex_units.mem as i64,
+                        cpu: accum.cpu + curr.ex_units.steps as i64,
+                    });
+
+            println!("{:?}", total_budget_used);
+
+            assert_eq!(
+                total_budget_used,
+                // Numbers came uplc evaluate
+                ExBudget {
+                    cpu: 40211433,
+                    mem: 114126
+                }
+            );
         }
         _ => unreachable!(),
     };