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 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
mod collect; mod empty; mod flat_scan; mod fold; mod iter; mod list; mod option; mod result; mod take; pub use collect::*; pub use empty::*; pub use flat_scan::*; pub use fold::*; pub use iter::*; pub use list::*; pub use option::*; pub use result::*; pub use take::*; #[cfg(test)] mod tests { use super::*; struct Ref<'a>(&'a mut u32); impl<'a> ListFn for Ref<'a> { type Item = u32; type End = Self; fn state(self) -> ListState<Self> { let first = *self.0; if first < 10 { *self.0 += 1; ListState::Some(first, self) } else { ListState::End(self) } } } struct Range10(u32); impl<'a> ListFn for Range10 { type Item = u32; type End = Self; fn state(self) -> ListState<Self> { let first = self.0; if first < 10 { ListState::Some(first, Self(first + 1)) } else { ListState::End(self) } } } #[test] fn as_ref() { let mut i = 0; let v = Ref(&mut i).iter().collect::<Vec<u32>>(); assert_eq!(v, vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9]); } #[test] fn as_range() { let v = Range10(0).collect(); assert_eq!(v, vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9]); } #[test] fn option() { { let a = Some(5); let v = a.iter().collect::<Vec<u32>>(); assert_eq!(v, vec![5]); } { let a = None; let v = a.iter().collect::<Vec<u32>>(); assert_eq!(v, Vec::new()); } } #[test] fn foreach() { let mut v = Vec::new(); for elem in Range10(2).iter() { v.push(elem); } assert_eq!(v, vec![2, 3, 4, 5, 6, 7, 8, 9]) } }