From 721ffa79a013e0322899dd3d198f219760e93907 Mon Sep 17 00:00:00 2001
From: Rostyslav Tyshko <tyshko.rostyslav@gmail.com>
Date: Sun, 10 Nov 2024 02:12:20 +0100
Subject: [PATCH] add simple_sum test

---
 zkvm/src/lib.rs | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/zkvm/src/lib.rs b/zkvm/src/lib.rs
index bd81b66..cef1864 100644
--- a/zkvm/src/lib.rs
+++ b/zkvm/src/lib.rs
@@ -26,3 +26,24 @@ pub fn verify(receipt: Receipt, image_id: impl Into<Digest>) {
     .verify(image_id)
     .expect("receipt verification failed");
 }
+
+#[cfg(test)]
+mod tests {
+    use test_methods::{SUMMATION_ELF, SUMMATION_ID};
+    use test_methods::{MULTIPLICATION_ELF, MULTIPLICATION_ID};
+    use super::*;
+
+    #[test]
+    fn simple_sum() {
+        let message = 1;
+        let message_2 = 2;
+    
+        let (digest, receipt) = prove(vec![message, message_2], SUMMATION_ELF);
+    
+        verify(receipt, SUMMATION_ID);
+        assert_eq!(
+            digest,
+            message + message_2
+        );
+    }
+}