changes

zink-lang · Nov 27, 2024 · 8b0c4ca · 8b0c4ca
1 parent ace7d84
commit 8b0c4ca
Showing 1 changed file with 27 additions and 24 deletions.
diff --git a/codegen/src/wasm/host.rs b/codegen/src/wasm/host.rs
@@ -101,63 +101,66 @@ mod tests {
         let addmod_func = HostFunc::try_from(("zinkc", "u256_addmod"));
         assert!(addmod_func.is_ok());
         let addmod_host_func = addmod_func.unwrap();
-        
+
         // Verify it maps to the correct EVM opcode
         match addmod_host_func {
-            HostFunc::Evm(OpCode::ADDMOD) => {},
+            HostFunc::Evm(OpCode::ADDMOD) => {}
             _ => panic!("Expected ADDMOD opcode"),
         }
 
         // Verify stack input/output for ADDMOD
-        assert_eq!(addmod_host_func.stack_in(), 3);  
-        assert_eq!(addmod_host_func.stack_out(), 1);  
+        assert_eq!(addmod_host_func.stack_in(), 3);
+        assert_eq!(addmod_host_func.stack_out(), 1);
 
         // Test MULMOD host function conversion
         let mulmod_func = HostFunc::try_from(("zinkc", "u256_mulmod"));
         assert!(mulmod_func.is_ok());
         let mulmod_host_func = mulmod_func.unwrap();
-        
+
         // Verify it maps to the correct EVM opcode
         match mulmod_host_func {
-            HostFunc::Evm(OpCode::MULMOD) => {},
+            HostFunc::Evm(OpCode::MULMOD) => {}
             _ => panic!("Expected MULMOD opcode"),
         }
 
         // Verify stack input/output for MULMOD
-        assert_eq!(mulmod_host_func.stack_in(), 3);  
-        assert_eq!(mulmod_host_func.stack_out(), 1);  
+        assert_eq!(mulmod_host_func.stack_in(), 3);
+        assert_eq!(mulmod_host_func.stack_out(), 1);
     }
-    
+
     #[test]
     fn test_addmod_mulmod_example_scenarios() {
         fn modular_add(a: u64, b: u64, n: u64) -> u64 {
             ((a % n) + (b % n)) % n
         }
-    
+
         let test_cases = [
-            (10u64, 20, 7, 2, 4),     
-            (100, 200, 3, 0, 2),  
-            (5, 6, 7, 4, 2),  
-            (10, 11, 12, 9, 2)  
+            (10u64, 20, 7, 2, 4),
+            (100, 200, 3, 0, 2),
+            (5, 6, 7, 4, 2),
+            (10, 11, 12, 9, 2),
         ];
-    
+
         for (a, b, n, expected_add, expected_mul) in test_cases.iter() {
             let actual_add = modular_add(*a, *b, *n);
             let actual_mul = (*a * *b) % *n;
-    
+
             println!(
                 "Debug: a={}, b={}, N={}, actual_add={}, expected_add={}",
                 a, b, n, actual_add, expected_add
             );
-
-            assert_eq!(actual_add, *expected_add,
+
+            assert_eq!(
+                actual_add, *expected_add,
                 "ADDMOD failed for inputs {}, {}, {} (got {}, expected {})",
-                a, b, n, actual_add, expected_add);
-
-            assert_eq!(actual_mul, *expected_mul,
+                a, b, n, actual_add, expected_add
+            );
+
+            assert_eq!(
+                actual_mul, *expected_mul,
                 "MULMOD failed for inputs {}, {}, {} (got {}, expected {})",
-                a, b, n, actual_mul, expected_mul);
+                a, b, n, actual_mul, expected_mul
+            );
         }
     }
-
-}
+}