1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
#[macro_export] macro_rules! vecset { () => ({ $crate::VecSet::default() }); ($($x:expr),*$(,)*) => ({ let mut set = $crate::VecSet::default(); $(set.insert($x);)* set }); } #[macro_export] macro_rules! vecmap { ($($key:expr => $value:expr,)+) => { vecmap!($($key => $value),+) }; ($($key:expr => $value:expr),*) => { { let mut _map = $crate::VecMap::default(); $( let _ = _map.insert($key, $value); )* _map } }; } #[cfg(test)] mod tests { use crate::{VecMap, VecSet}; #[test] fn vecset_macro() { let from_macro_empty: VecSet<[u32; 4]> = vecset! {}; let from_macro: VecSet<[u32; 4]> = vecset! {1,2,3}; let mut manual: VecSet<[u32; 4]> = VecSet::default(); assert_eq!(from_macro_empty, manual); manual.insert(1); manual.insert(2); manual.insert(3); assert_eq!(from_macro, manual); } #[test] fn vecmap_macro() { let from_macro_empty: VecMap<[(u32, u32); 4]> = vecmap! {}; let from_macro: VecMap<[(u32, u32); 4]> = vecmap! {1 => 2, 2 => 4,3 => 6}; let mut manual: VecMap<[(u32, u32); 4]> = VecMap::default(); assert_eq!(from_macro_empty, manual); manual.insert(1, 2); manual.insert(2, 4); manual.insert(3, 6); assert_eq!(from_macro, manual); } }