Return pyrelocatable from add_additional_hash_builtin (#149)

* return a pyrelocatable

* update test

* check that it returns a pyrelocatable

* Update src/cairo_runner.rs

Co-authored-by: fmoletta <[email protected]>

* change Cargo.toml

* fix test

Co-authored-by: Juan Rigada <[email protected]>
Co-authored-by: fmoletta <[email protected]>

Author: 3 people

Cargo.toml

@@ -13,7 +13,7 @@ default = ["pyo3/num-bigint", "pyo3/auto-initialize"]
 
 [dependencies]
 pyo3 = { version = "0.16.5" }
-cairo-rs = { git = "https://github.com/lambdaclass/cairo-rs.git", rev = "c7c3e4fca78d38da968e92c6586ec3a518697745" }
+cairo-rs = { git = "https://github.com/lambdaclass/cairo-rs.git", rev = "2ddf78e20cc25e660263a0c9c1b942780d95a0e6" }
 num-bigint = "0.4"
 lazy_static = "1.4.0"
 

src/cairo_runner.rs

@@ -536,9 +536,10 @@ impl PyCairoRunner {
     }
 
     /// Add (or replace if already present) a custom hash builtin.
-    pub fn add_additional_hash_builtin(&self) {
+    /// Returns a Relocatable with the new hash builtin base.
+    pub fn add_additional_hash_builtin(&self) -> PyRelocatable {
         let mut vm = (*self.pyvm.vm).borrow_mut();
-        self.inner.add_additional_hash_builtin(&mut vm);
+        self.inner.add_additional_hash_builtin(&mut vm).into()
     }
 }
 
@@ -1515,7 +1516,13 @@ mod test {
             )
             .unwrap();
 
-            runner.add_additional_hash_builtin();
+            let expected_relocatable = PyRelocatable {
+                segment_index: 0,
+                offset: 0,
+            };
+            let relocatable = runner.add_additional_hash_builtin();
+            assert_eq!(expected_relocatable, relocatable);
+
             assert_eq!(
                 (*runner.pyvm.vm)
                     .borrow()