Merge pull request #171 from 0xPolygonMiden/greenhat/i154-intrinsics-…

…tests

[7/x] Test suite for felt ops intrinsics translation

hir/src/component/mod.rs

@@ -148,6 +148,10 @@ impl Component {
         &self.modules
     }
 
+    pub fn to_modules(mut self) -> Vec<(Ident, Box<Module>)> {
+        self.modules.drain(..).collect()
+    }
+
     /// Return a mutable reference to the module table for this program
     pub fn modules_mut(&mut self) -> &mut IndexMap<Ident, Box<Module>> {
         &mut self.modules

sdk/prelude/src/intrinsics/felt.rs

@@ -1,4 +1,4 @@
-use core::ops::{Add, Div, Mul, Neg, Sub};
+use core::ops::{Add, AddAssign, Div, DivAssign, Mul, MulAssign, Neg, Sub, SubAssign};
 
 #[link(wasm_import_module = "miden:prelude/intrinsics_felt")]
 extern "C" {
@@ -171,6 +171,13 @@ impl Add for Felt {
     }
 }
 
+impl AddAssign for Felt {
+    #[inline(always)]
+    fn add_assign(&mut self, other: Self) {
+        *self = *self + other;
+    }
+}
+
 impl Sub for Felt {
     type Output = Self;
 
@@ -180,6 +187,13 @@ impl Sub for Felt {
     }
 }
 
+impl SubAssign for Felt {
+    #[inline(always)]
+    fn sub_assign(&mut self, other: Self) {
+        *self = *self - other;
+    }
+}
+
 impl Mul for Felt {
     type Output = Self;
 
@@ -189,6 +203,13 @@ impl Mul for Felt {
     }
 }
 
+impl MulAssign for Felt {
+    #[inline(always)]
+    fn mul_assign(&mut self, other: Self) {
+        *self = *self * other;
+    }
+}
+
 impl Div for Felt {
     type Output = Self;
 
@@ -198,6 +219,13 @@ impl Div for Felt {
     }
 }
 
+impl DivAssign for Felt {
+    #[inline(always)]
+    fn div_assign(&mut self, other: Self) {
+        *self = *self / other;
+    }
+}
+
 impl Neg for Felt {
     type Output = Self;
 

tests/integration/Cargo.toml

@@ -33,8 +33,8 @@ cargo-util = "0.2"
 filetime = "0.2.23"
 glob = "0.3.1"
 walkdir = "2.5.0"
+proptest.workspace = true
 
 [dev-dependencies]
 miden-core.workspace = true
-proptest.workspace = true
 concat-idents = "1.1"

tests/integration/expected/add_felt.hir

@@ -0,0 +1,21 @@
+(component 
+    ;; Modules
+    (module #add_felt
+        ;; Constants
+        (const (id 0) 0x00100000)
+
+        ;; Global Variables
+        (global (export #__stack_pointer) (id 0) (type i32) (const 0))
+
+        ;; Functions
+        (func (export #entrypoint) (param felt) (param felt) (result felt)
+            (block 0 (param v0 felt) (param v1 felt)
+                (let (v3 felt) (add.unchecked v0 v1))
+                (br (block 1 v3)))
+
+            (block 1 (param v2 felt)
+                (ret v2))
+        )
+    )
+
+)