diff options
Diffstat (limited to 'src/libstd')
| -rw-r--r-- | src/libstd/treemap.rs | 37 |
1 files changed, 21 insertions, 16 deletions
diff --git a/src/libstd/treemap.rs b/src/libstd/treemap.rs index 9cef77cfa76..55a714309af 100644 --- a/src/libstd/treemap.rs +++ b/src/libstd/treemap.rs @@ -1141,25 +1141,30 @@ mod test_set { #[test] fn test_difference() { - let mut a = TreeSet::new(); - let mut b = TreeSet::new(); - - fail_unless!(a.insert(1)); - fail_unless!(a.insert(3)); - fail_unless!(a.insert(5)); - fail_unless!(a.insert(9)); - fail_unless!(a.insert(11)); + fn check_difference(a: &[int], b: &[int], expected: &[int]) { + let mut set_a = TreeSet::new(); + let mut set_b = TreeSet::new(); - fail_unless!(b.insert(3)); - fail_unless!(b.insert(9)); + for a.each |x| { fail_unless!(set_a.insert(*x)) } + for b.each |y| { fail_unless!(set_b.insert(*y)) } - let mut i = 0; - let expected = [1, 5, 11]; - for a.difference(&b) |x| { - fail_unless!(*x == expected[i]); - i += 1 + let mut i = 0; + for set_a.difference(&set_b) |x| { + fail_unless!(*x == expected[i]); + i += 1; + } + fail_unless!(i == expected.len()); } - fail_unless!(i == expected.len()); + + check_difference([], [], []); + check_difference([1, 12], [], [1, 12]); + check_difference([], [1, 2, 3, 9], []); + check_difference([1, 3, 5, 9, 11], + [3, 9], + [1, 5, 11]); + check_difference([-5, 11, 22, 33, 40, 42], + [-12, -5, 14, 23, 34, 38, 39, 50], + [11, 22, 33, 40, 42]); } #[test] |