From 6b5290c2fd151105c540a98c643061a233c0a1bc Mon Sep 17 00:00:00 2001
From: juanbono <juanbono94@gmail.com>
Date: Thu, 18 Apr 2024 12:55:18 -0300
Subject: [PATCH] fix segment_arena tests to take into account that we no
 longer subtract INITIAL_SEGMENT_SIZE

---
 vm/src/vm/runners/builtin_runner/segment_arena.rs | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/vm/src/vm/runners/builtin_runner/segment_arena.rs b/vm/src/vm/runners/builtin_runner/segment_arena.rs
index c673e5118b..57cd43253d 100644
--- a/vm/src/vm/runners/builtin_runner/segment_arena.rs
+++ b/vm/src/vm/runners/builtin_runner/segment_arena.rs
@@ -119,7 +119,7 @@ mod tests {
         let mut vm = vm!();
         vm.segments.segment_used_sizes = Some(vec![3]);
 
-        assert_eq!(builtin.get_used_instances(&vm.segments), Ok(0));
+        assert_eq!(builtin.get_used_instances(&vm.segments), Ok(1));
     }
 
     #[test]
@@ -130,7 +130,7 @@ mod tests {
         let mut vm = vm!();
         vm.segments.segment_used_sizes = Some(vec![3]);
 
-        assert_eq!(builtin.get_used_instances(&vm.segments), Ok(0));
+        assert_eq!(builtin.get_used_instances(&vm.segments), Ok(1));
     }
 
     #[test]
@@ -154,7 +154,7 @@ mod tests {
             builtin.final_stack(&vm.segments, pointer),
             Err(RunnerError::InvalidStopPointer(Box::new((
                 SEGMENT_ARENA_BUILTIN_NAME,
-                relocatable!(0, 3),
+                relocatable!(0, 6),
                 relocatable!(0, 0)
             ))))
         );
@@ -245,7 +245,7 @@ mod tests {
 
         assert_eq!(
             builtin.get_used_cells_and_allocated_size(&vm),
-            Ok((0_usize, 0))
+            Ok((3_usize, 3))
         );
     }
 
@@ -303,7 +303,7 @@ mod tests {
         let mut vm = vm!();
 
         vm.segments.segment_used_sizes = Some(vec![4]);
-        assert_eq!(builtin.get_used_cells(&vm.segments), Ok(1));
+        assert_eq!(builtin.get_used_cells(&vm.segments), Ok(1 + 3));
     }
 
     #[test]
@@ -325,7 +325,7 @@ mod tests {
         let mut memory_segment_manager = MemorySegmentManager::new();
         memory_segment_manager.segment_used_sizes = Some(vec![6]);
         // SIZE(6) / CELLS_PER_INSTANCE(3)
-        assert_eq!(builtin.get_used_instances(&memory_segment_manager), Ok(1));
+        assert_eq!(builtin.get_used_instances(&memory_segment_manager), Ok(2));
     }
 
     #[test]