lssa-zkvm-testing/shared/memory_allocations/src/lib.rs

use std::collections::{BTreeMap, BinaryHeap, HashMap, HashSet};

pub const CAP_LIM: usize = 10000;
pub const TO_INSERT_ITEM: usize = 145;

pub fn alloc_btree_map() -> BTreeMap<usize, usize> {
    BTreeMap::new()
}

pub fn insert_btree_map(mmap: &mut BTreeMap<usize, usize>) {
    for i in 1..CAP_LIM {
        mmap.insert(TO_INSERT_ITEM + i, TO_INSERT_ITEM);
    }
}

pub fn remove_btree_map(mmap: &mut BTreeMap<usize, usize>) {
    for i in 1..CAP_LIM {
        mmap.remove(&(TO_INSERT_ITEM + i));
    }
}

macro_rules! write_tests_maps {
    ($t:ty, $fn_name_1:ident, $fn_name_2:ident, $fn_name_3:ident, $fn_name_4:ident) => {
        pub fn $fn_name_1() -> $t {
            <$t>::with_capacity(CAP_LIM)
        }

        pub fn $fn_name_2(mmap: &mut $t) {
            for i in 1..CAP_LIM {
                mmap.insert(TO_INSERT_ITEM + i, TO_INSERT_ITEM);
            }
        }

        pub fn $fn_name_3(mmap: &mut $t) {
            for i in 0..(CAP_LIM + 1) {
                mmap.insert(TO_INSERT_ITEM + i, TO_INSERT_ITEM);
            }
        }

        pub fn $fn_name_4(mmap: &mut $t) {
            for i in 1..CAP_LIM {
                mmap.remove(&(TO_INSERT_ITEM + i));
            }
        }
    };
}

macro_rules! write_tests_sets {
    ($t:ty, $fn_name_1:ident, $fn_name_2:ident, $fn_name_3:ident, $fn_name_4:ident) => {
        pub fn $fn_name_1() -> $t {
            <$t>::with_capacity(CAP_LIM)
        }

        pub fn $fn_name_2(mmap: &mut $t) {
            for i in 1..CAP_LIM {
                mmap.insert(TO_INSERT_ITEM + i);
            }
        }

        pub fn $fn_name_3(mmap: &mut $t) {
            for i in 0..(CAP_LIM + 1) {
                mmap.insert(TO_INSERT_ITEM + i);
            }
        }

        pub fn $fn_name_4(mmap: &mut $t) {
            for i in 1..CAP_LIM {
                mmap.remove(&(TO_INSERT_ITEM + i));
            }
        }
    };
}

macro_rules! write_tests_vec {
    ($t:ty, $fn_name_1:ident, $fn_name_2:ident, $fn_name_3:ident, $fn_name_4:ident) => {
        pub fn $fn_name_1() -> $t {
            <$t>::with_capacity(CAP_LIM)
        }

        pub fn $fn_name_2(vecc: &mut $t) {
            for _ in 1..CAP_LIM {
                vecc.push(TO_INSERT_ITEM);
            }
        }

        pub fn $fn_name_3(vecc: &mut $t) {
            for _ in 0..(CAP_LIM + 1) {
                vecc.push(TO_INSERT_ITEM);
            }
        }

        pub fn $fn_name_4(vecc: &mut $t) {
            for _ in 1..CAP_LIM {
                vecc.pop();
            }
        }
    };
}

write_tests_maps!(HashMap<usize, usize>, alloc_hash_map, insert_hash_map, dyn_alloc_hash_map, remove_hash_map);
write_tests_sets!(HashSet<usize>, alloc_hash_set, insert_hash_set, dyn_alloc_hash_set, remove_hash_set);
write_tests_vec!(Vec<usize>, alloc_vec, push_vec, dyn_alloc_vec, pop_vec);
write_tests_vec!(BinaryHeap<usize>, alloc_b_heap, push_b_heap, dyn_alloc_b_heap, pop_b_heap);

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test2() {
        let mut mmap = alloc_hash_map();

        insert_hash_map(&mut mmap);

        remove_hash_map(&mut mmap);
    }
}