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
|
#![allow(clippy::unit_arg)]
use std::collections::BTreeSet;
use std::fmt::{Debug, Error, Formatter, Write};
use crate::OrdSet;
use proptest::proptest;
use proptest_derive::Arbitrary;
#[derive(Arbitrary, Debug)]
enum Action<A> {
Insert(A),
Remove(A),
}
#[derive(Arbitrary)]
struct Actions<A>(Vec<Action<A>>)
where
A: Ord + Clone;
impl<A> Debug for Actions<A>
where
A: Ord + Debug + Clone,
{
fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error> {
let mut out = String::new();
let mut expected = BTreeSet::new();
writeln!(out, "let mut set = OrdSet::new();")?;
for action in &self.0 {
match action {
Action::Insert(ref value) => {
expected.insert(value.clone());
writeln!(out, "set.insert({:?});", value)?;
}
Action::Remove(ref value) => {
expected.remove(value);
writeln!(out, "set.remove({:?});", value)?;
}
}
}
writeln!(
out,
"let expected = vec!{:?};",
expected.into_iter().collect::<Vec<_>>()
)?;
writeln!(out, "assert_eq!(OrdSet::from(expected), set);")?;
write!(f, "{}", super::code_fmt(&out))
}
}
proptest! {
#[test]
fn comprehensive(actions: Actions<u8>) {
let mut set = OrdSet::new();
let mut nat = BTreeSet::new();
for action in actions.0 {
match action {
Action::Insert(value) => {
let len = nat.len() + if nat.contains(&value) {
0
} else {
1
};
nat.insert(value);
set.insert(value);
assert_eq!(len, set.len());
}
Action::Remove(value) => {
let len = nat.len() - if nat.contains(&value) {
1
} else {
0
};
nat.remove(&value);
set.remove(&value);
assert_eq!(len, set.len());
}
}
assert_eq!(nat.len(), set.len());
assert_eq!(OrdSet::from(nat.clone()), set);
assert!(nat.iter().eq(set.iter()));
}
}
}
|
