1 // Copyright 2012-2015 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
12 use std::cmp::Ordering::{Equal, Greater, Less};
13 use std::cmp::Ordering;
17 use std::sync::atomic::Ordering::Relaxed;
18 use std::sync::atomic::{ATOMIC_USIZE_INIT, AtomicUsize};
21 use rand::{Rng, thread_rng};
23 fn square(n: usize) -> usize {
27 fn is_odd(n: &usize) -> bool {
33 // Test on-stack from_fn.
34 let mut v: Vec<_> = (0..3).map(square).collect();
37 assert_eq!(v.len(), 3);
43 // Test on-heap from_fn.
44 v = (0..5).map(square).collect();
47 assert_eq!(v.len(), 5);
58 // Test on-stack from_elem.
59 let mut v = vec![10, 10];
62 assert_eq!(v.len(), 2);
67 // Test on-heap from_elem.
82 let xs: [i32; 0] = [];
83 assert!(xs.is_empty());
84 assert!(![0].is_empty());
88 fn test_len_divzero() {
92 let v2: &[Z] = &[[], []];
93 assert_eq!(mem::size_of::<Z>(), 0);
94 assert_eq!(v0.len(), 0);
95 assert_eq!(v1.len(), 1);
96 assert_eq!(v2.len(), 2);
101 let mut a = vec![11];
102 assert_eq!(a.get(1), None);
104 assert_eq!(a.get(1).unwrap(), &12);
105 a = vec![11, 12, 13];
106 assert_eq!(a.get(1).unwrap(), &12);
112 assert_eq!(a.first(), None);
114 assert_eq!(a.first().unwrap(), &11);
116 assert_eq!(a.first().unwrap(), &11);
120 fn test_first_mut() {
122 assert_eq!(a.first_mut(), None);
124 assert_eq!(*a.first_mut().unwrap(), 11);
126 assert_eq!(*a.first_mut().unwrap(), 11);
130 fn test_split_first() {
131 let mut a = vec![11];
133 assert!(b.split_first().is_none());
134 assert_eq!(a.split_first(), Some((&11, b)));
136 let b: &[i32] = &[12];
137 assert_eq!(a.split_first(), Some((&11, b)));
141 fn test_split_first_mut() {
142 let mut a = vec![11];
143 let b: &mut [i32] = &mut [];
144 assert!(b.split_first_mut().is_none());
145 assert!(a.split_first_mut() == Some((&mut 11, b)));
147 let b: &mut [_] = &mut [12];
148 assert!(a.split_first_mut() == Some((&mut 11, b)));
152 fn test_split_last() {
153 let mut a = vec![11];
155 assert!(b.split_last().is_none());
156 assert_eq!(a.split_last(), Some((&11, b)));
159 assert_eq!(a.split_last(), Some((&12, b)));
163 fn test_split_last_mut() {
164 let mut a = vec![11];
165 let b: &mut [i32] = &mut [];
166 assert!(b.split_last_mut().is_none());
167 assert!(a.split_last_mut() == Some((&mut 11, b)));
170 let b: &mut [_] = &mut [11];
171 assert!(a.split_last_mut() == Some((&mut 12, b)));
177 assert_eq!(a.last(), None);
179 assert_eq!(a.last().unwrap(), &11);
181 assert_eq!(a.last().unwrap(), &12);
187 assert_eq!(a.last_mut(), None);
189 assert_eq!(*a.last_mut().unwrap(), 11);
191 assert_eq!(*a.last_mut().unwrap(), 12);
196 // Test fixed length vector.
197 let vec_fixed = [1, 2, 3, 4];
198 let v_a = vec_fixed[1..vec_fixed.len()].to_vec();
199 assert_eq!(v_a.len(), 3);
201 assert_eq!(v_a[0], 2);
202 assert_eq!(v_a[1], 3);
203 assert_eq!(v_a[2], 4);
206 let vec_stack: &[_] = &[1, 2, 3];
207 let v_b = vec_stack[1..3].to_vec();
208 assert_eq!(v_b.len(), 2);
210 assert_eq!(v_b[0], 2);
211 assert_eq!(v_b[1], 3);
214 let vec_unique = vec![1, 2, 3, 4, 5, 6];
215 let v_d = vec_unique[1..6].to_vec();
216 assert_eq!(v_d.len(), 5);
218 assert_eq!(v_d[0], 2);
219 assert_eq!(v_d[1], 3);
220 assert_eq!(v_d[2], 4);
221 assert_eq!(v_d[3], 5);
222 assert_eq!(v_d[4], 6);
226 fn test_slice_from() {
227 let vec: &[_] = &[1, 2, 3, 4];
228 assert_eq!(&vec[..], vec);
229 let b: &[_] = &[3, 4];
230 assert_eq!(&vec[2..], b);
232 assert_eq!(&vec[4..], b);
237 let vec: &[_] = &[1, 2, 3, 4];
238 assert_eq!(&vec[..4], vec);
239 let b: &[_] = &[1, 2];
240 assert_eq!(&vec[..2], b);
242 assert_eq!(&vec[..0], b);
250 assert_eq!(v.len(), 0);
251 assert_eq!(e, Some(5));
259 fn test_swap_remove() {
260 let mut v = vec![1, 2, 3, 4, 5];
261 let mut e = v.swap_remove(0);
263 assert_eq!(v, [5, 2, 3, 4]);
264 e = v.swap_remove(3);
266 assert_eq!(v, [5, 2, 3]);
271 fn test_swap_remove_fail() {
273 let _ = v.swap_remove(0);
274 let _ = v.swap_remove(0);
278 fn test_swap_remove_noncopyable() {
279 // Tests that we don't accidentally run destructors twice.
280 let mut v: Vec<Box<_>> = Vec::new();
284 let mut _e = v.swap_remove(0);
285 assert_eq!(v.len(), 2);
286 _e = v.swap_remove(1);
287 assert_eq!(v.len(), 1);
288 _e = v.swap_remove(0);
289 assert_eq!(v.len(), 0);
294 // Test on-stack push().
297 assert_eq!(v.len(), 1);
300 // Test on-heap push().
302 assert_eq!(v.len(), 2);
309 let mut v: Vec<Box<_>> = vec![box 6, box 5, box 4];
312 assert_eq!(v.len(), 1);
313 assert_eq!(*(v[0]), 6);
314 // If the unsafe block didn't drop things properly, we blow up here.
319 let mut v: Vec<Box<_>> = vec![box 6, box 5, box 4];
321 assert_eq!(v.len(), 0);
322 // If the unsafe block didn't drop things properly, we blow up here.
327 let mut v = vec![1, 2, 3, 4, 5];
329 assert_eq!(v, [1, 3, 5]);
333 fn test_binary_search() {
334 assert_eq!([1, 2, 3, 4, 5].binary_search(&5).ok(), Some(4));
335 assert_eq!([1, 2, 3, 4, 5].binary_search(&4).ok(), Some(3));
336 assert_eq!([1, 2, 3, 4, 5].binary_search(&3).ok(), Some(2));
337 assert_eq!([1, 2, 3, 4, 5].binary_search(&2).ok(), Some(1));
338 assert_eq!([1, 2, 3, 4, 5].binary_search(&1).ok(), Some(0));
340 assert_eq!([2, 4, 6, 8, 10].binary_search(&1).ok(), None);
341 assert_eq!([2, 4, 6, 8, 10].binary_search(&5).ok(), None);
342 assert_eq!([2, 4, 6, 8, 10].binary_search(&4).ok(), Some(1));
343 assert_eq!([2, 4, 6, 8, 10].binary_search(&10).ok(), Some(4));
345 assert_eq!([2, 4, 6, 8].binary_search(&1).ok(), None);
346 assert_eq!([2, 4, 6, 8].binary_search(&5).ok(), None);
347 assert_eq!([2, 4, 6, 8].binary_search(&4).ok(), Some(1));
348 assert_eq!([2, 4, 6, 8].binary_search(&8).ok(), Some(3));
350 assert_eq!([2, 4, 6].binary_search(&1).ok(), None);
351 assert_eq!([2, 4, 6].binary_search(&5).ok(), None);
352 assert_eq!([2, 4, 6].binary_search(&4).ok(), Some(1));
353 assert_eq!([2, 4, 6].binary_search(&6).ok(), Some(2));
355 assert_eq!([2, 4].binary_search(&1).ok(), None);
356 assert_eq!([2, 4].binary_search(&5).ok(), None);
357 assert_eq!([2, 4].binary_search(&2).ok(), Some(0));
358 assert_eq!([2, 4].binary_search(&4).ok(), Some(1));
360 assert_eq!([2].binary_search(&1).ok(), None);
361 assert_eq!([2].binary_search(&5).ok(), None);
362 assert_eq!([2].binary_search(&2).ok(), Some(0));
364 assert_eq!([].binary_search(&1).ok(), None);
365 assert_eq!([].binary_search(&5).ok(), None);
367 assert!([1, 1, 1, 1, 1].binary_search(&1).ok() != None);
368 assert!([1, 1, 1, 1, 2].binary_search(&1).ok() != None);
369 assert!([1, 1, 1, 2, 2].binary_search(&1).ok() != None);
370 assert!([1, 1, 2, 2, 2].binary_search(&1).ok() != None);
371 assert_eq!([1, 2, 2, 2, 2].binary_search(&1).ok(), Some(0));
373 assert_eq!([1, 2, 3, 4, 5].binary_search(&6).ok(), None);
374 assert_eq!([1, 2, 3, 4, 5].binary_search(&0).ok(), None);
379 let mut v = vec![10, 20];
380 assert_eq!(v[0], 10);
381 assert_eq!(v[1], 20);
383 assert_eq!(v[0], 20);
384 assert_eq!(v[1], 10);
386 let mut v3 = Vec::<i32>::new();
388 assert!(v3.is_empty());
390 // check the 1-byte-types path
391 let mut v = (-50..51i8).collect::<Vec<_>>();
393 assert_eq!(v, (-50..51i8).rev().collect::<Vec<_>>());
395 // check the 2-byte-types path
396 let mut v = (-50..51i16).collect::<Vec<_>>();
398 assert_eq!(v, (-50..51i16).rev().collect::<Vec<_>>());
403 let mut rng = thread_rng();
405 for len in (2..25).chain(500..510) {
406 for &modulus in &[5, 10, 100, 1000] {
408 let orig: Vec<_> = rng.gen_iter::<i32>()
409 .map(|x| x % modulus)
413 // Sort in default order.
414 let mut v = orig.clone();
416 assert!(v.windows(2).all(|w| w[0] <= w[1]));
418 // Sort in ascending order.
419 let mut v = orig.clone();
420 v.sort_by(|a, b| a.cmp(b));
421 assert!(v.windows(2).all(|w| w[0] <= w[1]));
423 // Sort in descending order.
424 let mut v = orig.clone();
425 v.sort_by(|a, b| b.cmp(a));
426 assert!(v.windows(2).all(|w| w[0] >= w[1]));
428 // Sort in lexicographic order.
429 let mut v1 = orig.clone();
430 let mut v2 = orig.clone();
431 v1.sort_by_key(|x| x.to_string());
432 v2.sort_by_cached_key(|x| x.to_string());
433 assert!(v1.windows(2).all(|w| w[0].to_string() <= w[1].to_string()));
436 // Sort with many pre-sorted runs.
437 let mut v = orig.clone();
441 let a = rng.gen::<usize>() % len;
442 let b = rng.gen::<usize>() % len;
450 assert!(v.windows(2).all(|w| w[0] <= w[1]));
455 // Sort using a completely random comparison function.
456 // This will reorder the elements *somehow*, but won't panic.
457 let mut v = [0; 500];
458 for i in 0..v.len() {
461 v.sort_by(|_, _| *rng.choose(&[Less, Equal, Greater]).unwrap());
463 for i in 0..v.len() {
464 assert_eq!(v[i], i as i32);
472 let mut v = [0xDEADBEEFu64];
474 assert!(v == [0xDEADBEEF]);
478 fn test_sort_stability() {
479 for len in (2..25).chain(500..510) {
481 let mut counts = [0; 10];
483 // create a vector like [(6, 1), (5, 1), (6, 2), ...],
484 // where the first item of each tuple is random, but
485 // the second item represents which occurrence of that
486 // number this element is, i.e. the second elements
487 // will occur in sorted order.
488 let mut orig: Vec<_> = (0..len)
490 let n = thread_rng().gen::<usize>() % 10;
496 let mut v = orig.clone();
497 // Only sort on the first element, so an unstable sort
498 // may mix up the counts.
499 v.sort_by(|&(a, _), &(b, _)| a.cmp(&b));
501 // This comparison includes the count (the second item
502 // of the tuple), so elements with equal first items
503 // will need to be ordered with increasing
504 // counts... i.e. exactly asserting that this sort is
506 assert!(v.windows(2).all(|w| w[0] <= w[1]));
508 let mut v = orig.clone();
509 v.sort_by_cached_key(|&(x, _)| x);
510 assert!(v.windows(2).all(|w| w[0] <= w[1]));
516 fn test_rotate_left() {
517 let expected: Vec<_> = (0..13).collect();
518 let mut v = Vec::new();
521 v.clone_from(&expected);
523 assert_eq!(v, expected);
524 v.rotate_left(expected.len());
525 assert_eq!(v, expected);
526 let mut zst_array = [(), (), ()];
527 zst_array.rotate_left(2);
530 v = (5..13).chain(0..5).collect();
532 assert_eq!(v, expected);
534 let expected: Vec<_> = (0..1000).collect();
536 // small rotations in large slice, uses ptr::copy
537 v = (2..1000).chain(0..2).collect();
539 assert_eq!(v, expected);
540 v = (998..1000).chain(0..998).collect();
542 assert_eq!(v, expected);
544 // non-small prime rotation, has a few rounds of swapping
545 v = (389..1000).chain(0..389).collect();
546 v.rotate_left(1000-389);
547 assert_eq!(v, expected);
551 fn test_rotate_right() {
552 let expected: Vec<_> = (0..13).collect();
553 let mut v = Vec::new();
556 v.clone_from(&expected);
558 assert_eq!(v, expected);
559 v.rotate_right(expected.len());
560 assert_eq!(v, expected);
561 let mut zst_array = [(), (), ()];
562 zst_array.rotate_right(2);
565 v = (5..13).chain(0..5).collect();
567 assert_eq!(v, expected);
569 let expected: Vec<_> = (0..1000).collect();
571 // small rotations in large slice, uses ptr::copy
572 v = (2..1000).chain(0..2).collect();
574 assert_eq!(v, expected);
575 v = (998..1000).chain(0..998).collect();
577 assert_eq!(v, expected);
579 // non-small prime rotation, has a few rounds of swapping
580 v = (389..1000).chain(0..389).collect();
582 assert_eq!(v, expected);
587 let v: [Vec<i32>; 0] = [];
590 let d = [vec![1], vec![2, 3]].concat();
591 assert_eq!(d, [1, 2, 3]);
593 let v: &[&[_]] = &[&[1], &[2, 3]];
594 assert_eq!(v.join(&0), [1, 0, 2, 3]);
595 let v: &[&[_]] = &[&[1], &[2], &[3]];
596 assert_eq!(v.join(&0), [1, 0, 2, 0, 3]);
601 let v: [Vec<i32>; 0] = [];
602 assert_eq!(v.join(&0), []);
603 assert_eq!([vec![1], vec![2, 3]].join(&0), [1, 0, 2, 3]);
604 assert_eq!([vec![1], vec![2], vec![3]].join(&0), [1, 0, 2, 0, 3]);
606 let v: [&[_]; 2] = [&[1], &[2, 3]];
607 assert_eq!(v.join(&0), [1, 0, 2, 3]);
608 let v: [&[_]; 3] = [&[1], &[2], &[3]];
609 assert_eq!(v.join(&0), [1, 0, 2, 0, 3]);
614 let mut a = vec![1, 2, 4];
616 assert_eq!(a, [1, 2, 3, 4]);
618 let mut a = vec![1, 2, 3];
620 assert_eq!(a, [0, 1, 2, 3]);
622 let mut a = vec![1, 2, 3];
624 assert_eq!(a, [1, 2, 3, 4]);
633 fn test_insert_oob() {
634 let mut a = vec![1, 2, 3];
640 let mut a = vec![1, 2, 3, 4];
642 assert_eq!(a.remove(2), 3);
643 assert_eq!(a, [1, 2, 4]);
645 assert_eq!(a.remove(2), 4);
646 assert_eq!(a, [1, 2]);
648 assert_eq!(a.remove(0), 1);
651 assert_eq!(a.remove(0), 2);
657 fn test_remove_fail() {
667 assert!(v.capacity() >= 11);
672 let v = vec![1, 2, 3, 4, 5];
674 assert_eq!(v.len(), 2);
679 macro_rules! assert_order {
680 (Greater, $a:expr, $b:expr) => {
681 assert_eq!($a.cmp($b), Greater);
684 (Less, $a:expr, $b:expr) => {
685 assert_eq!($a.cmp($b), Less);
688 (Equal, $a:expr, $b:expr) => {
689 assert_eq!($a.cmp($b), Equal);
695 fn test_total_ord_u8() {
696 let c = &[1u8, 2, 3];
697 assert_order!(Greater, &[1u8, 2, 3, 4][..], &c[..]);
698 let c = &[1u8, 2, 3, 4];
699 assert_order!(Less, &[1u8, 2, 3][..], &c[..]);
700 let c = &[1u8, 2, 3, 6];
701 assert_order!(Equal, &[1u8, 2, 3, 6][..], &c[..]);
702 let c = &[1u8, 2, 3, 4, 5, 6];
703 assert_order!(Less, &[1u8, 2, 3, 4, 5, 5, 5, 5][..], &c[..]);
704 let c = &[1u8, 2, 3, 4];
705 assert_order!(Greater, &[2u8, 2][..], &c[..]);
710 fn test_total_ord_i32() {
712 assert_order!(Greater, &[1, 2, 3, 4][..], &c[..]);
713 let c = &[1, 2, 3, 4];
714 assert_order!(Less, &[1, 2, 3][..], &c[..]);
715 let c = &[1, 2, 3, 6];
716 assert_order!(Equal, &[1, 2, 3, 6][..], &c[..]);
717 let c = &[1, 2, 3, 4, 5, 6];
718 assert_order!(Less, &[1, 2, 3, 4, 5, 5, 5, 5][..], &c[..]);
719 let c = &[1, 2, 3, 4];
720 assert_order!(Greater, &[2, 2][..], &c[..]);
725 let xs = [1, 2, 5, 10, 11];
726 let mut it = xs.iter();
727 assert_eq!(it.size_hint(), (5, Some(5)));
728 assert_eq!(it.next().unwrap(), &1);
729 assert_eq!(it.size_hint(), (4, Some(4)));
730 assert_eq!(it.next().unwrap(), &2);
731 assert_eq!(it.size_hint(), (3, Some(3)));
732 assert_eq!(it.next().unwrap(), &5);
733 assert_eq!(it.size_hint(), (2, Some(2)));
734 assert_eq!(it.next().unwrap(), &10);
735 assert_eq!(it.size_hint(), (1, Some(1)));
736 assert_eq!(it.next().unwrap(), &11);
737 assert_eq!(it.size_hint(), (0, Some(0)));
738 assert!(it.next().is_none());
742 fn test_iter_size_hints() {
743 let mut xs = [1, 2, 5, 10, 11];
744 assert_eq!(xs.iter().size_hint(), (5, Some(5)));
745 assert_eq!(xs.iter_mut().size_hint(), (5, Some(5)));
749 fn test_iter_as_slice() {
750 let xs = [1, 2, 5, 10, 11];
751 let mut iter = xs.iter();
752 assert_eq!(iter.as_slice(), &[1, 2, 5, 10, 11]);
754 assert_eq!(iter.as_slice(), &[2, 5, 10, 11]);
758 fn test_iter_as_ref() {
759 let xs = [1, 2, 5, 10, 11];
760 let mut iter = xs.iter();
761 assert_eq!(iter.as_ref(), &[1, 2, 5, 10, 11]);
763 assert_eq!(iter.as_ref(), &[2, 5, 10, 11]);
767 fn test_iter_clone() {
769 let mut it = xs.iter();
771 let mut jt = it.clone();
772 assert_eq!(it.next(), jt.next());
773 assert_eq!(it.next(), jt.next());
774 assert_eq!(it.next(), jt.next());
778 fn test_iter_is_empty() {
779 let xs = [1, 2, 5, 10, 11];
780 for i in 0..xs.len() {
781 for j in i..xs.len() {
782 assert_eq!(xs[i..j].iter().is_empty(), xs[i..j].is_empty());
788 fn test_mut_iterator() {
789 let mut xs = [1, 2, 3, 4, 5];
793 assert!(xs == [2, 3, 4, 5, 6])
797 fn test_rev_iterator() {
799 let xs = [1, 2, 5, 10, 11];
800 let ys = [11, 10, 5, 2, 1];
802 for &x in xs.iter().rev() {
803 assert_eq!(x, ys[i]);
810 fn test_mut_rev_iterator() {
811 let mut xs = [1, 2, 3, 4, 5];
812 for (i, x) in xs.iter_mut().rev().enumerate() {
815 assert!(xs == [5, 5, 5, 5, 5])
819 fn test_move_iterator() {
820 let xs = vec![1, 2, 3, 4, 5];
821 assert_eq!(xs.into_iter().fold(0, |a: usize, b: usize| 10 * a + b),
826 fn test_move_rev_iterator() {
827 let xs = vec![1, 2, 3, 4, 5];
828 assert_eq!(xs.into_iter().rev().fold(0, |a: usize, b: usize| 10 * a + b),
833 fn test_splitator() {
834 let xs = &[1, 2, 3, 4, 5];
836 let splits: &[&[_]] = &[&[1], &[3], &[5]];
837 assert_eq!(xs.split(|x| *x % 2 == 0).collect::<Vec<_>>(), splits);
838 let splits: &[&[_]] = &[&[], &[2, 3, 4, 5]];
839 assert_eq!(xs.split(|x| *x == 1).collect::<Vec<_>>(), splits);
840 let splits: &[&[_]] = &[&[1, 2, 3, 4], &[]];
841 assert_eq!(xs.split(|x| *x == 5).collect::<Vec<_>>(), splits);
842 let splits: &[&[_]] = &[&[1, 2, 3, 4, 5]];
843 assert_eq!(xs.split(|x| *x == 10).collect::<Vec<_>>(), splits);
844 let splits: &[&[_]] = &[&[], &[], &[], &[], &[], &[]];
845 assert_eq!(xs.split(|_| true).collect::<Vec<&[i32]>>(), splits);
847 let xs: &[i32] = &[];
848 let splits: &[&[i32]] = &[&[]];
849 assert_eq!(xs.split(|x| *x == 5).collect::<Vec<&[i32]>>(), splits);
853 fn test_splitnator() {
854 let xs = &[1, 2, 3, 4, 5];
856 let splits: &[&[_]] = &[&[1, 2, 3, 4, 5]];
857 assert_eq!(xs.splitn(1, |x| *x % 2 == 0).collect::<Vec<_>>(), splits);
858 let splits: &[&[_]] = &[&[1], &[3, 4, 5]];
859 assert_eq!(xs.splitn(2, |x| *x % 2 == 0).collect::<Vec<_>>(), splits);
860 let splits: &[&[_]] = &[&[], &[], &[], &[4, 5]];
861 assert_eq!(xs.splitn(4, |_| true).collect::<Vec<_>>(), splits);
863 let xs: &[i32] = &[];
864 let splits: &[&[i32]] = &[&[]];
865 assert_eq!(xs.splitn(2, |x| *x == 5).collect::<Vec<_>>(), splits);
869 fn test_splitnator_mut() {
870 let xs = &mut [1, 2, 3, 4, 5];
872 let splits: &[&mut [_]] = &[&mut [1, 2, 3, 4, 5]];
873 assert_eq!(xs.splitn_mut(1, |x| *x % 2 == 0).collect::<Vec<_>>(),
875 let splits: &[&mut [_]] = &[&mut [1], &mut [3, 4, 5]];
876 assert_eq!(xs.splitn_mut(2, |x| *x % 2 == 0).collect::<Vec<_>>(),
878 let splits: &[&mut [_]] = &[&mut [], &mut [], &mut [], &mut [4, 5]];
879 assert_eq!(xs.splitn_mut(4, |_| true).collect::<Vec<_>>(), splits);
881 let xs: &mut [i32] = &mut [];
882 let splits: &[&mut [i32]] = &[&mut []];
883 assert_eq!(xs.splitn_mut(2, |x| *x == 5).collect::<Vec<_>>(), splits);
887 fn test_rsplitator() {
888 let xs = &[1, 2, 3, 4, 5];
890 let splits: &[&[_]] = &[&[5], &[3], &[1]];
891 assert_eq!(xs.split(|x| *x % 2 == 0).rev().collect::<Vec<_>>(), splits);
892 let splits: &[&[_]] = &[&[2, 3, 4, 5], &[]];
893 assert_eq!(xs.split(|x| *x == 1).rev().collect::<Vec<_>>(), splits);
894 let splits: &[&[_]] = &[&[], &[1, 2, 3, 4]];
895 assert_eq!(xs.split(|x| *x == 5).rev().collect::<Vec<_>>(), splits);
896 let splits: &[&[_]] = &[&[1, 2, 3, 4, 5]];
897 assert_eq!(xs.split(|x| *x == 10).rev().collect::<Vec<_>>(), splits);
899 let xs: &[i32] = &[];
900 let splits: &[&[i32]] = &[&[]];
901 assert_eq!(xs.split(|x| *x == 5).rev().collect::<Vec<&[i32]>>(), splits);
905 fn test_rsplitnator() {
906 let xs = &[1, 2, 3, 4, 5];
908 let splits: &[&[_]] = &[&[1, 2, 3, 4, 5]];
909 assert_eq!(xs.rsplitn(1, |x| *x % 2 == 0).collect::<Vec<_>>(), splits);
910 let splits: &[&[_]] = &[&[5], &[1, 2, 3]];
911 assert_eq!(xs.rsplitn(2, |x| *x % 2 == 0).collect::<Vec<_>>(), splits);
912 let splits: &[&[_]] = &[&[], &[], &[], &[1, 2]];
913 assert_eq!(xs.rsplitn(4, |_| true).collect::<Vec<_>>(), splits);
915 let xs: &[i32] = &[];
916 let splits: &[&[i32]] = &[&[]];
917 assert_eq!(xs.rsplitn(2, |x| *x == 5).collect::<Vec<&[i32]>>(), splits);
918 assert!(xs.rsplitn(0, |x| *x % 2 == 0).next().is_none());
922 fn test_windowsator() {
923 let v = &[1, 2, 3, 4];
925 let wins: &[&[_]] = &[&[1, 2], &[2, 3], &[3, 4]];
926 assert_eq!(v.windows(2).collect::<Vec<_>>(), wins);
928 let wins: &[&[_]] = &[&[1, 2, 3], &[2, 3, 4]];
929 assert_eq!(v.windows(3).collect::<Vec<_>>(), wins);
930 assert!(v.windows(6).next().is_none());
932 let wins: &[&[_]] = &[&[3, 4], &[2, 3], &[1, 2]];
933 assert_eq!(v.windows(2).rev().collect::<Vec<&[_]>>(), wins);
938 fn test_windowsator_0() {
939 let v = &[1, 2, 3, 4];
940 let _it = v.windows(0);
944 fn test_chunksator() {
945 let v = &[1, 2, 3, 4, 5];
947 assert_eq!(v.chunks(2).len(), 3);
949 let chunks: &[&[_]] = &[&[1, 2], &[3, 4], &[5]];
950 assert_eq!(v.chunks(2).collect::<Vec<_>>(), chunks);
951 let chunks: &[&[_]] = &[&[1, 2, 3], &[4, 5]];
952 assert_eq!(v.chunks(3).collect::<Vec<_>>(), chunks);
953 let chunks: &[&[_]] = &[&[1, 2, 3, 4, 5]];
954 assert_eq!(v.chunks(6).collect::<Vec<_>>(), chunks);
956 let chunks: &[&[_]] = &[&[5], &[3, 4], &[1, 2]];
957 assert_eq!(v.chunks(2).rev().collect::<Vec<_>>(), chunks);
962 fn test_chunksator_0() {
963 let v = &[1, 2, 3, 4];
964 let _it = v.chunks(0);
968 fn test_exact_chunksator() {
969 let v = &[1, 2, 3, 4, 5];
971 assert_eq!(v.exact_chunks(2).len(), 2);
973 let chunks: &[&[_]] = &[&[1, 2], &[3, 4]];
974 assert_eq!(v.exact_chunks(2).collect::<Vec<_>>(), chunks);
975 let chunks: &[&[_]] = &[&[1, 2, 3]];
976 assert_eq!(v.exact_chunks(3).collect::<Vec<_>>(), chunks);
977 let chunks: &[&[_]] = &[];
978 assert_eq!(v.exact_chunks(6).collect::<Vec<_>>(), chunks);
980 let chunks: &[&[_]] = &[&[3, 4], &[1, 2]];
981 assert_eq!(v.exact_chunks(2).rev().collect::<Vec<_>>(), chunks);
986 fn test_exact_chunksator_0() {
987 let v = &[1, 2, 3, 4];
988 let _it = v.exact_chunks(0);
992 fn test_reverse_part() {
993 let mut values = [1, 2, 3, 4, 5];
994 values[1..4].reverse();
995 assert!(values == [1, 4, 3, 2, 5]);
1000 macro_rules! test_show_vec {
1001 ($x:expr, $x_str:expr) => ({
1002 let (x, x_str) = ($x, $x_str);
1003 assert_eq!(format!("{:?}", x), x_str);
1004 assert_eq!(format!("{:?}", x), x_str);
1007 let empty = Vec::<i32>::new();
1008 test_show_vec!(empty, "[]");
1009 test_show_vec!(vec![1], "[1]");
1010 test_show_vec!(vec![1, 2, 3], "[1, 2, 3]");
1011 test_show_vec!(vec![vec![], vec![1], vec![1, 1]], "[[], [1], [1, 1]]");
1013 let empty_mut: &mut [i32] = &mut [];
1014 test_show_vec!(empty_mut, "[]");
1016 test_show_vec!(v, "[1]");
1017 let v = &mut [1, 2, 3];
1018 test_show_vec!(v, "[1, 2, 3]");
1019 let v: &mut [&mut [_]] = &mut [&mut [], &mut [1], &mut [1, 1]];
1020 test_show_vec!(v, "[[], [1], [1, 1]]");
1024 fn test_vec_default() {
1027 let v: $ty = Default::default();
1028 assert!(v.is_empty());
1038 fn test_overflow_does_not_cause_segfault() {
1040 v.reserve_exact(!0);
1047 fn test_overflow_does_not_cause_segfault_managed() {
1048 let mut v = vec![Rc::new(1)];
1049 v.reserve_exact(!0);
1054 fn test_mut_split_at() {
1055 let mut values = [1, 2, 3, 4, 5];
1057 let (left, right) = values.split_at_mut(2);
1059 let left: &[_] = left;
1060 assert!(left[..left.len()] == [1, 2]);
1067 let right: &[_] = right;
1068 assert!(right[..right.len()] == [3, 4, 5]);
1075 assert!(values == [2, 3, 5, 6, 7]);
1078 #[derive(Clone, PartialEq)]
1082 fn test_iter_zero_sized() {
1083 let mut v = vec![Foo, Foo, Foo];
1084 assert_eq!(v.len(), 3);
1109 assert_eq!(cnt, 11);
1111 let xs: [Foo; 3] = [Foo, Foo, Foo];
1121 fn test_shrink_to_fit() {
1122 let mut xs = vec![0, 1, 2, 3];
1126 assert_eq!(xs.capacity(), 128);
1128 assert_eq!(xs.capacity(), 100);
1129 assert_eq!(xs, (0..100).collect::<Vec<_>>());
1133 fn test_starts_with() {
1134 assert!(b"foobar".starts_with(b"foo"));
1135 assert!(!b"foobar".starts_with(b"oob"));
1136 assert!(!b"foobar".starts_with(b"bar"));
1137 assert!(!b"foo".starts_with(b"foobar"));
1138 assert!(!b"bar".starts_with(b"foobar"));
1139 assert!(b"foobar".starts_with(b"foobar"));
1140 let empty: &[u8] = &[];
1141 assert!(empty.starts_with(empty));
1142 assert!(!empty.starts_with(b"foo"));
1143 assert!(b"foobar".starts_with(empty));
1147 fn test_ends_with() {
1148 assert!(b"foobar".ends_with(b"bar"));
1149 assert!(!b"foobar".ends_with(b"oba"));
1150 assert!(!b"foobar".ends_with(b"foo"));
1151 assert!(!b"foo".ends_with(b"foobar"));
1152 assert!(!b"bar".ends_with(b"foobar"));
1153 assert!(b"foobar".ends_with(b"foobar"));
1154 let empty: &[u8] = &[];
1155 assert!(empty.ends_with(empty));
1156 assert!(!empty.ends_with(b"foo"));
1157 assert!(b"foobar".ends_with(empty));
1161 fn test_mut_splitator() {
1162 let mut xs = [0, 1, 0, 2, 3, 0, 0, 4, 5, 0];
1163 assert_eq!(xs.split_mut(|x| *x == 0).count(), 6);
1164 for slice in xs.split_mut(|x| *x == 0) {
1167 assert!(xs == [0, 1, 0, 3, 2, 0, 0, 5, 4, 0]);
1169 let mut xs = [0, 1, 0, 2, 3, 0, 0, 4, 5, 0, 6, 7];
1170 for slice in xs.split_mut(|x| *x == 0).take(5) {
1173 assert!(xs == [0, 1, 0, 3, 2, 0, 0, 5, 4, 0, 6, 7]);
1177 fn test_mut_splitator_rev() {
1178 let mut xs = [1, 2, 0, 3, 4, 0, 0, 5, 6, 0];
1179 for slice in xs.split_mut(|x| *x == 0).rev().take(4) {
1182 assert!(xs == [1, 2, 0, 4, 3, 0, 0, 6, 5, 0]);
1187 let mut v = [0, 1, 2];
1188 assert_eq!(v.get_mut(3), None);
1189 v.get_mut(1).map(|e| *e = 7);
1190 assert_eq!(v[1], 7);
1192 assert_eq!(v.get_mut(2), Some(&mut x));
1196 fn test_mut_chunks() {
1197 let mut v = [0, 1, 2, 3, 4, 5, 6];
1198 assert_eq!(v.chunks_mut(2).len(), 4);
1199 for (i, chunk) in v.chunks_mut(3).enumerate() {
1204 let result = [0, 0, 0, 1, 1, 1, 2];
1205 assert_eq!(v, result);
1209 fn test_mut_chunks_rev() {
1210 let mut v = [0, 1, 2, 3, 4, 5, 6];
1211 for (i, chunk) in v.chunks_mut(3).rev().enumerate() {
1216 let result = [2, 2, 2, 1, 1, 1, 0];
1217 assert_eq!(v, result);
1222 fn test_mut_chunks_0() {
1223 let mut v = [1, 2, 3, 4];
1224 let _it = v.chunks_mut(0);
1228 fn test_mut_exact_chunks() {
1229 let mut v = [0, 1, 2, 3, 4, 5, 6];
1230 assert_eq!(v.exact_chunks_mut(2).len(), 3);
1231 for (i, chunk) in v.exact_chunks_mut(3).enumerate() {
1236 let result = [0, 0, 0, 1, 1, 1, 6];
1237 assert_eq!(v, result);
1241 fn test_mut_exact_chunks_rev() {
1242 let mut v = [0, 1, 2, 3, 4, 5, 6];
1243 for (i, chunk) in v.exact_chunks_mut(3).rev().enumerate() {
1248 let result = [1, 1, 1, 0, 0, 0, 6];
1249 assert_eq!(v, result);
1254 fn test_mut_exact_chunks_0() {
1255 let mut v = [1, 2, 3, 4];
1256 let _it = v.exact_chunks_mut(0);
1260 fn test_mut_last() {
1261 let mut x = [1, 2, 3, 4, 5];
1262 let h = x.last_mut();
1263 assert_eq!(*h.unwrap(), 5);
1265 let y: &mut [i32] = &mut [];
1266 assert!(y.last_mut().is_none());
1271 let xs: Box<_> = box [1, 2, 3];
1272 let ys = xs.to_vec();
1273 assert_eq!(ys, [1, 2, 3]);
1277 fn test_box_slice_clone() {
1278 let data = vec![vec![0, 1], vec![0], vec![1]];
1279 let data2 = data.clone().into_boxed_slice().clone().to_vec();
1281 assert_eq!(data, data2);
1285 #[cfg_attr(target_os = "emscripten", ignore)]
1286 fn test_box_slice_clone_panics() {
1288 use std::sync::atomic::{AtomicUsize, Ordering};
1289 use std::thread::spawn;
1292 count: Arc<AtomicUsize>,
1296 impl Drop for Canary {
1297 fn drop(&mut self) {
1298 self.count.fetch_add(1, Ordering::SeqCst);
1302 impl Clone for Canary {
1303 fn clone(&self) -> Self {
1309 count: self.count.clone(),
1310 panics: self.panics,
1315 let drop_count = Arc::new(AtomicUsize::new(0));
1316 let canary = Canary {
1317 count: drop_count.clone(),
1320 let panic = Canary {
1321 count: drop_count.clone(),
1326 // When xs is dropped, +5.
1327 let xs = vec![canary.clone(), canary.clone(), canary.clone(), panic, canary]
1328 .into_boxed_slice();
1330 // When panic is cloned, +3.
1337 assert_eq!(drop_count.load(Ordering::SeqCst), 8);
1341 fn test_copy_from_slice() {
1342 let src = [0, 1, 2, 3, 4, 5];
1343 let mut dst = [0; 6];
1344 dst.copy_from_slice(&src);
1345 assert_eq!(src, dst)
1349 #[should_panic(expected = "destination and source slices have different lengths")]
1350 fn test_copy_from_slice_dst_longer() {
1351 let src = [0, 1, 2, 3];
1352 let mut dst = [0; 5];
1353 dst.copy_from_slice(&src);
1357 #[should_panic(expected = "destination and source slices have different lengths")]
1358 fn test_copy_from_slice_dst_shorter() {
1359 let src = [0, 1, 2, 3];
1360 let mut dst = [0; 3];
1361 dst.copy_from_slice(&src);
1364 const MAX_LEN: usize = 80;
1366 static DROP_COUNTS: [AtomicUsize; MAX_LEN] = [
1367 // FIXME(RFC 1109): AtomicUsize is not Copy.
1368 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1369 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1370 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1371 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1372 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1373 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1374 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1375 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1376 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1377 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1378 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1379 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1380 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1381 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1382 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1383 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1384 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1385 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1386 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1387 AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0), AtomicUsize::new(0),
1390 static VERSIONS: AtomicUsize = ATOMIC_USIZE_INIT;
1392 #[derive(Clone, Eq)]
1393 struct DropCounter {
1396 version: Cell<usize>,
1399 impl PartialEq for DropCounter {
1400 fn eq(&self, other: &Self) -> bool {
1401 self.partial_cmp(other) == Some(Ordering::Equal)
1405 impl PartialOrd for DropCounter {
1406 fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
1407 self.version.set(self.version.get() + 1);
1408 other.version.set(other.version.get() + 1);
1409 VERSIONS.fetch_add(2, Relaxed);
1410 self.x.partial_cmp(&other.x)
1414 impl Ord for DropCounter {
1415 fn cmp(&self, other: &Self) -> Ordering {
1416 self.partial_cmp(other).unwrap()
1420 impl Drop for DropCounter {
1421 fn drop(&mut self) {
1422 DROP_COUNTS[self.id].fetch_add(1, Relaxed);
1423 VERSIONS.fetch_sub(self.version.get(), Relaxed);
1428 ($input:ident, $func:ident) => {
1429 let len = $input.len();
1431 // Work out the total number of comparisons required to sort
1433 let mut count = 0usize;
1434 $input.to_owned().$func(|a, b| { count += 1; a.cmp(b) });
1436 // ... and then panic on each and every single one.
1437 for panic_countdown in 0..count {
1438 // Refresh the counters.
1439 VERSIONS.store(0, Relaxed);
1441 DROP_COUNTS[i].store(0, Relaxed);
1444 let v = $input.to_owned();
1445 let _ = thread::spawn(move || {
1447 let mut panic_countdown = panic_countdown;
1449 if panic_countdown == 0 {
1450 SILENCE_PANIC.with(|s| s.set(true));
1453 panic_countdown -= 1;
1458 // Check that the number of things dropped is exactly
1459 // what we expect (i.e. the contents of `v`).
1460 for (i, c) in DROP_COUNTS.iter().enumerate().take(len) {
1461 let count = c.load(Relaxed);
1463 "found drop count == {} for i == {}, len == {}",
1467 // Check that the most recent versions of values were dropped.
1468 assert_eq!(VERSIONS.load(Relaxed), 0);
1473 thread_local!(static SILENCE_PANIC: Cell<bool> = Cell::new(false));
1476 #[cfg_attr(target_os = "emscripten", ignore)] // no threads
1478 let prev = panic::take_hook();
1479 panic::set_hook(Box::new(move |info| {
1480 if !SILENCE_PANIC.with(|s| s.get()) {
1485 let mut rng = thread_rng();
1487 for len in (1..20).chain(70..MAX_LEN) {
1488 for &modulus in &[5, 20, 50] {
1489 for &has_runs in &[false, true] {
1490 let mut input = (0..len)
1493 x: rng.next_u32() % modulus,
1495 version: Cell::new(0),
1498 .collect::<Vec<_>>();
1501 for c in &mut input {
1506 let a = rng.gen::<usize>() % len;
1507 let b = rng.gen::<usize>() % len;
1509 input[a..b].reverse();
1516 test!(input, sort_by);
1517 test!(input, sort_unstable_by);