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
/// Creates new collection of element by running each element in collection thru iteratee.
///
/// * iteratee function signature: ```fn(item: &T, index: usize) -> U```
///
/// # Arguments
///
/// * `collection` - The collection to iterate over.
///
/// * `predicate` - The function invoked per iteration.
///
/// # Returns
///
/// Returns the new mapped collection.
///
/// # Examples
///
/// ```
/// use rufl::collection;
///
/// assert_eq!(vec![2, 3, 4, 5, 6], collection::map(&vec![1, 2, 3, 4, 5], &|n: &i32, _i: usize| { *n + 1 }));
///
/// assert_eq!(vec![1, 0, 1, 0, 1], collection::map(&vec![1, 2, 3, 4, 5], &|n: &i32, _i: usize| { *n % 2 }));
///
/// ```
pub fn map<T, U>(vector: &Vec<T>, iteratee: impl Fn(&T, usize) -> U) -> Vec<U> {
let mut result: Vec<U> = Vec::new();
for (index, item) in vector.iter().enumerate() {
result.push(iteratee(item, index));
}
result
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_filter() {
assert_eq!(
vec![2, 3, 4, 5, 6],
map(&vec![1, 2, 3, 4, 5], &|n: &i32, _i: usize| { *n + 1 })
);
assert_eq!(
vec![1, 0, 1, 0, 1],
map(&vec![1, 2, 3, 4, 5], &|n: &i32, _i: usize| { *n % 2 })
);
}
}