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
#![cfg_attr(not(test), no_std)]
#[cfg(test)]
extern crate quickcheck;
pub fn partition<T, P>(data: &mut [T], predicate: P) -> (&mut [T], &mut [T])
where P: Fn(&T) -> bool {
let len = data.len();
if len == 0 { return (&mut [], &mut []); }
let (mut l, mut r) = (0, len - 1);
loop {
while l < len && predicate(&data[l]) { l += 1; }
while r > 0 && !predicate(&data[r]) { r -= 1; }
if l >= r { return data.split_at_mut(l); }
data.swap(l, r);
}
}
#[cfg(test)]
mod tests {
use super::partition;
use quickcheck::QuickCheck;
#[test]
fn test_empty() {
let empty : &mut [usize] = &mut [][..];
assert_eq!((&mut [][..], &mut [][..]), partition(empty, |_| true));
}
#[test]
fn test_single_true() {
assert_eq!((&mut [1][..], &mut [][..]), partition(&mut [1u8][..], |_| true));
}
#[test]
fn test_single_false() {
assert_eq!((&mut [][..], &mut [1][..]), partition(&mut [1u8][..], |_| false));
}
#[test]
fn quickcheck() {
fn prop(data: Vec<u32>) -> bool {
let mut data = data;
let mut trues = data.iter().cloned().filter(|e| e % 2 == 0).collect::<Vec<u32>>();
let mut falses = data.iter().cloned().filter(|e| e % 2 != 0).collect::<Vec<u32>>();
let (left, right) = partition(&mut data, |&e| e % 2 == 0);
trues.sort();
falses.sort();
left.sort();
right.sort();
trues == left && falses == right
}
QuickCheck::new().tests(10000).quickcheck(prop as fn(Vec<u32>) -> bool);
}
}