Add Display trait to Memory struct

src/vm/vm_memory/memory.rs

@@ -7,6 +7,7 @@ use felt::Felt;
 use std::{
     borrow::Cow,
     collections::{HashMap, HashSet},
+    fmt::{Display, Formatter},
 };
 
 pub struct ValidationRule(
@@ -300,6 +301,27 @@ impl Memory {
     }
 }
 
+impl Display for Memory {
+    fn fmt(&self, f: &mut Formatter) -> std::fmt::Result {
+        for (i, segment) in self.temp_data.iter().enumerate() {
+            for (j, cell) in segment.iter().enumerate() {
+                if let Some(cell) = cell {
+                    let temp_segment = i + 1;
+                    writeln!(f, "(-{temp_segment},{j}) : {cell}")?;
+                }
+            }
+        }
+        for (i, segment) in self.data.iter().enumerate() {
+            for (j, cell) in segment.iter().enumerate() {
+                if let Some(cell) = cell {
+                    writeln!(f, "({i},{j}) : {cell}")?;
+                }
+            }
+        }
+        writeln!(f, "}}")
+    }
+}
+
 pub(crate) trait RelocateValue<'a, Input: 'a, Output: 'a> {
     fn relocate_value(&self, value: Input) -> Output;
 }
@@ -1225,6 +1247,28 @@ mod memory_tests {
         assert!(memory.temp_data.is_empty());
     }
 
+    #[test]
+    fn test_memory_display() {
+        let mut memory = memory![
+            ((0, 0), 1),
+            ((0, 1), (-1, 0)),
+            ((0, 2), 3),
+            ((1, 0), (-1, 1)),
+            ((1, 1), 5),
+            ((1, 2), (-1, 2))
+        ];
+
+        memory.temp_data = vec![vec![
+            mayberelocatable!(-1, 0).into(),
+            mayberelocatable!(8).into(),
+            mayberelocatable!(9).into(),
+        ]];
+
+        assert_eq!(
+            format!("{}", memory),
+            "(-1,0) : -1:0\n(-1,1) : 8\n(-1,2) : 9\n(0,0) : 1\n(0,1) : -1:0\n(0,2) : 3\n(1,0) : -1:1\n(1,1) : 5\n(1,2) : -1:2\n}\n");
+    }
+
     #[test]
     fn relocate_memory_into_existing_segment_temporary_values_in_temporary_memory() {
         let mut memory = memory![