Squash dict add test tags

cairo_programs/squash_dict.cairo

@@ -64,6 +64,7 @@ func squash_dict{range_check_ptr}(
         squashed_dict=squashed_dict,
         big_keys=big_keys)
     %{ 
+        #TEST
         vm_exit_scope() 
     %}
     return (squashed_dict=squashed_dict)
@@ -173,7 +174,10 @@ func squash_dict_inner(
     [ap] = dict_accesses_end_minus1 - cast(last_loop_locals.access_ptr, felt)
     [ap] = [last_loop_locals.range_check_ptr]; ap++
     tempvar n_used_accesses = last_loop_locals.range_check_ptr - range_check_ptr
-    %{ assert ids.n_used_accesses == len(access_indices[key]) %}
+    %{ 
+        #TEST
+        assert ids.n_used_accesses == len(access_indices[key]) 
+    %}
 
     # Write last value to dict_diff.
     last_loop_locals.value = dict_diff.new_value

hints_tests.py

@@ -32,11 +32,11 @@ def test_program(program_name: str):
     test_program("memcpy")
     test_program("memset")
     test_program("dict_new")
-    # test_program("dict_read") # Waiting on starkware PR
-    # test_program("dict_write") # ValueError: Custom Hint Error: AttributeError: 'PyTypeId' object has no attribute 'segment_index'
-    # test_program("dict_update") # Waiting on starkware PR
+    test_program("dict_read")
+    test_program("dict_write")
+    test_program("dict_update")
     test_program("default_dict_new")
-    # test_program("squash_dict") # ValueError: Custom Hint Error: ValueError: Failed to get ids value
+    test_program("squash_dict") 
     # test_program("dict_squash") # Custom Hint Error: AttributeError: 'PyTypeId' object has no attribute 'segment_index'
     test_program("ids_size")
     test_program("split_felt")

src/cairo_run.rs

@@ -1,6 +1,7 @@
 #[cfg(test)]
 mod test {
     use crate::cairo_runner::PyCairoRunner;
+    use pyo3::Python;
     use std::fs;
 
     #[test]
@@ -9,9 +10,11 @@ mod test {
         let program = fs::read_to_string(path).unwrap();
         let mut runner =
             PyCairoRunner::new(program, Some("main".to_string()), None, false).unwrap();
-        runner
-            .cairo_run_py(false, None, None, None, None, None)
-            .expect("Couldn't run program");
+        Python::with_gil(|py| {
+            runner
+                .cairo_run_py(false, None, None, None, py, None, None)
+                .expect("Couldn't run program");
+        });
     }
 
     #[test]
@@ -25,9 +28,11 @@ mod test {
             false,
         )
         .unwrap();
-        runner
-            .cairo_run_py(false, None, None, None, None, None)
-            .expect("Couldn't run program");
+        Python::with_gil(|py| {
+            runner
+                .cairo_run_py(false, None, None, None, py, None, None)
+                .expect("Couldn't run program");
+        });
     }
 
     #[test]
@@ -36,8 +41,10 @@ mod test {
         let program = fs::read_to_string(path).unwrap();
         let mut runner =
             PyCairoRunner::new(program, Some("main".to_string()), None, false).unwrap();
-        runner
-            .cairo_run_py(false, None, None, None, None, None)
-            .expect("Couldn't run program");
+        Python::with_gil(|py| {
+            runner
+                .cairo_run_py(false, None, None, None, py, None, None)
+                .expect("Couldn't run program");
+        });
     }
 }