Fix tests + fmt

lambdaclass · Apr 18, 2024 · 5b54cfa · 5b54cfa
1 parent 92e69ee
commit 5b54cfa
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 9 deletions.
diff --git a/vm/src/tests/cairo_pie_test.rs b/vm/src/tests/cairo_pie_test.rs
@@ -102,7 +102,7 @@ fn pedersen_test() {
         ),
         (BuiltinName::range_check, BuiltinAdditionalData::None),
     ]);
-    assert_eq!(cairo_pie.additional_data, expected_additional_data);
+    assert_eq!(cairo_pie.additional_data.0, expected_additional_data);
     // memory
     assert_eq!(
         cairo_pie.memory,
@@ -177,7 +177,7 @@ fn common_signature() {
             ),
         )])),
     )]);
-    assert_eq!(cairo_pie.additional_data, expected_additional_data);
+    assert_eq!(cairo_pie.additional_data.0, expected_additional_data);
     // memory
     assert_eq!(
         cairo_pie.memory,
@@ -236,7 +236,7 @@ fn relocate_segments() {
     };
     assert_eq!(cairo_pie.execution_resources, expected_execution_resources);
     // additional_data
-    assert!(cairo_pie.additional_data.is_empty());
+    assert!(cairo_pie.additional_data.0.is_empty());
     // memory
     assert_eq!(
         cairo_pie.memory,

diff --git a/vm/src/vm/runners/cairo_pie.rs b/vm/src/vm/runners/cairo_pie.rs
@@ -71,7 +71,7 @@ pub enum BuiltinAdditionalData {
 #[derive(Serialize, Clone, Debug, PartialEq, Eq)]
 pub struct CairoPieAdditionalData(
     #[serde(with = "crate::types::builtin_name::serde_generic_map_impl")]
-    pub HashMap<BuiltinName, BuiltinAdditionalData>
+    pub  HashMap<BuiltinName, BuiltinAdditionalData>,
 );
 
 #[derive(Serialize, Clone, Debug, PartialEq, Eq)]

diff --git a/vm/src/vm/runners/cairo_runner.rs b/vm/src/vm/runners/cairo_runner.rs
@@ -1426,11 +1426,12 @@ impl CairoRunner {
             metadata,
             memory: (&vm.segments.memory).into(),
             execution_resources: self.get_execution_resources(vm)?,
-            additional_data: CairoPieAdditionalData(vm
-                .builtin_runners
-                .iter()
-                .map(|b| (b.name(), b.get_additional_data()))
-                .collect()),
+            additional_data: CairoPieAdditionalData(
+                vm.builtin_runners
+                    .iter()
+                    .map(|b| (b.name(), b.get_additional_data()))
+                    .collect(),
+            ),
             version: CairoPieVersion { cairo_pie: () },
         })
     }