rskit-util 0.1.0-alpha.1

Pure utility functions for rskit
Documentation 
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

/// Split a collection into two collections based on a predicate function.
///
/// The first vector contains elements where the predicate is true, and the second
/// vector contains elements where the predicate is false.
///
/// # Examples
///
/// ```
/// use rskit_util::collections::partition;
/// let items = vec![1, 2, 3, 4, 5];
/// let (evens, odds) = partition(items, |n| n % 2 == 0);
/// assert_eq!(evens, vec![2, 4]);
/// assert_eq!(odds, vec![1, 3, 5]);
/// ```
pub fn partition<T, F>(items: impl IntoIterator<Item = T>, mut predicate: F) -> (Vec<T>, Vec<T>)
where
    F: FnMut(&T) -> bool,
{
    let mut truthy = Vec::new();
    let mut falsy = Vec::new();
    for item in items {
        if predicate(&item) {
            truthy.push(item);
        } else {
            falsy.push(item);
        }
    }
    (truthy, falsy)
}

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

    #[test]
    fn test_partition() {
        let items = vec![1, 2, 3, 4, 5];
        let (evens, odds) = partition(items, |n| n % 2 == 0);
        assert_eq!(evens, vec![2, 4]);
        assert_eq!(odds, vec![1, 3, 5]);
    }
}