1 // Copyright 2014 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.
11 use std::iter::{FromIterator, repeat};
12 use std::mem::size_of;
16 struct DropCounter<'a> {
20 impl<'a> Drop for DropCounter<'a> {
27 fn test_small_vec_struct() {
28 assert!(size_of::<Vec<u8>>() == size_of::<usize>() * 3);
32 fn test_double_drop() {
38 let (mut count_x, mut count_y) = (0, 0);
44 tv.x.push(DropCounter {count: &mut count_x});
45 tv.y.push(DropCounter {count: &mut count_y});
47 // If Vec had a drop flag, here is where it would be zeroed.
48 // Instead, it should rely on its internal state to prevent
49 // doing anything significant when dropped multiple times.
52 // Here tv goes out of scope, tv.y should be dropped, but not tv.x.
55 assert_eq!(count_x, 1);
56 assert_eq!(count_y, 1);
61 let mut v = Vec::new();
62 assert_eq!(v.capacity(), 0);
65 assert!(v.capacity() >= 2);
71 assert!(v.capacity() >= 16);
73 assert!(v.capacity() >= 32);
78 assert!(v.capacity() >= 33)
83 let mut v = Vec::new();
84 let mut w = Vec::new();
87 for i in 0..3 { w.push(i) }
92 for i in 3..10 { w.push(i) }
98 fn test_extend_ref() {
99 let mut v = vec![1, 2];
100 v.extend(&[3, 4, 5]);
102 assert_eq!(v.len(), 5);
103 assert_eq!(v, [1, 2, 3, 4, 5]);
108 assert_eq!(v.len(), 7);
109 assert_eq!(v, [1, 2, 3, 4, 5, 6, 7]);
113 fn test_slice_from_mut() {
114 let mut values = vec![1, 2, 3, 4, 5];
116 let slice = &mut values[2 ..];
117 assert!(slice == [3, 4, 5]);
123 assert!(values == [1, 2, 5, 6, 7]);
127 fn test_slice_to_mut() {
128 let mut values = vec![1, 2, 3, 4, 5];
130 let slice = &mut values[.. 2];
131 assert!(slice == [1, 2]);
137 assert!(values == [2, 3, 3, 4, 5]);
141 fn test_split_at_mut() {
142 let mut values = vec![1, 2, 3, 4, 5];
144 let (left, right) = values.split_at_mut(2);
146 let left: &[_] = left;
147 assert!(&left[..left.len()] == &[1, 2]);
154 let right: &[_] = right;
155 assert!(&right[..right.len()] == &[3, 4, 5]);
162 assert_eq!(values, [2, 3, 5, 6, 7]);
167 let v: Vec<i32> = vec![];
168 let w = vec!(1, 2, 3);
170 assert_eq!(v, v.clone());
174 // they should be disjoint in memory.
175 assert!(w.as_ptr() != z.as_ptr())
179 fn test_clone_from() {
181 let three: Vec<Box<_>> = vec!(box 1, box 2, box 3);
182 let two: Vec<Box<_>> = vec!(box 4, box 5);
184 v.clone_from(&three);
185 assert_eq!(v, three);
188 v.clone_from(&three);
189 assert_eq!(v, three);
196 v.clone_from(&three);
202 let mut vec = vec![1, 2, 3, 4];
203 vec.retain(|&x| x % 2 == 0);
204 assert_eq!(vec, [2, 4]);
208 fn zero_sized_values() {
209 let mut v = Vec::new();
210 assert_eq!(v.len(), 0);
212 assert_eq!(v.len(), 1);
214 assert_eq!(v.len(), 2);
215 assert_eq!(v.pop(), Some(()));
216 assert_eq!(v.pop(), Some(()));
217 assert_eq!(v.pop(), None);
219 assert_eq!(v.iter().count(), 0);
221 assert_eq!(v.iter().count(), 1);
223 assert_eq!(v.iter().count(), 2);
227 assert_eq!(v.iter_mut().count(), 2);
229 assert_eq!(v.iter_mut().count(), 3);
231 assert_eq!(v.iter_mut().count(), 4);
233 for &mut () in &mut v {}
234 unsafe { v.set_len(0); }
235 assert_eq!(v.iter_mut().count(), 0);
239 fn test_partition() {
240 assert_eq!(vec![].into_iter().partition(|x: &i32| *x < 3), (vec![], vec![]));
241 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 4), (vec![1, 2, 3], vec![]));
242 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 2), (vec![1], vec![2, 3]));
243 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 0), (vec![], vec![1, 2, 3]));
247 fn test_zip_unzip() {
248 let z1 = vec![(1, 4), (2, 5), (3, 6)];
250 let (left, right): (Vec<_>, Vec<_>) = z1.iter().cloned().unzip();
252 assert_eq!((1, 4), (left[0], right[0]));
253 assert_eq!((2, 5), (left[1], right[1]));
254 assert_eq!((3, 6), (left[2], right[2]));
258 fn test_vec_truncate_drop() {
259 static mut drops: u32 = 0;
263 unsafe { drops += 1; }
267 let mut v = vec![Elem(1), Elem(2), Elem(3), Elem(4), Elem(5)];
268 assert_eq!(unsafe { drops }, 0);
270 assert_eq!(unsafe { drops }, 2);
272 assert_eq!(unsafe { drops }, 5);
277 fn test_vec_truncate_fail() {
279 impl Drop for BadElem {
281 let BadElem(ref mut x) = *self;
283 panic!("BadElem panic: 0xbadbeef")
288 let mut v = vec![BadElem(1), BadElem(2), BadElem(0xbadbeef), BadElem(4)];
294 let vec = vec![1, 2, 3];
295 assert!(vec[1] == 2);
300 fn test_index_out_of_bounds() {
301 let vec = vec![1, 2, 3];
307 fn test_slice_out_of_bounds_1() {
308 let x = vec![1, 2, 3, 4, 5];
314 fn test_slice_out_of_bounds_2() {
315 let x = vec![1, 2, 3, 4, 5];
321 fn test_slice_out_of_bounds_3() {
322 let x = vec![1, 2, 3, 4, 5];
328 fn test_slice_out_of_bounds_4() {
329 let x = vec![1, 2, 3, 4, 5];
335 fn test_slice_out_of_bounds_5() {
336 let x = vec![1, 2, 3, 4, 5];
342 fn test_swap_remove_empty() {
343 let mut vec= Vec::<i32>::new();
348 fn test_move_items() {
349 let vec = vec![1, 2, 3];
350 let mut vec2 = vec![];
354 assert_eq!(vec2, [1, 2, 3]);
358 fn test_move_items_reverse() {
359 let vec = vec![1, 2, 3];
360 let mut vec2 = vec![];
361 for i in vec.into_iter().rev() {
364 assert_eq!(vec2, [3, 2, 1]);
368 fn test_move_items_zero_sized() {
369 let vec = vec![(), (), ()];
370 let mut vec2 = vec![];
374 assert_eq!(vec2, [(), (), ()]);
378 fn test_drain_items() {
379 let mut vec = vec![1, 2, 3];
380 let mut vec2 = vec![];
381 for i in vec.drain(..) {
385 assert_eq!(vec2, [ 1, 2, 3 ]);
389 fn test_drain_items_reverse() {
390 let mut vec = vec![1, 2, 3];
391 let mut vec2 = vec![];
392 for i in vec.drain(..).rev() {
396 assert_eq!(vec2, [3, 2, 1]);
400 fn test_drain_items_zero_sized() {
401 let mut vec = vec![(), (), ()];
402 let mut vec2 = vec![];
403 for i in vec.drain(..) {
407 assert_eq!(vec2, [(), (), ()]);
412 fn test_drain_out_of_bounds() {
413 let mut v = vec![1, 2, 3, 4, 5];
418 fn test_drain_range() {
419 let mut v = vec![1, 2, 3, 4, 5];
420 for _ in v.drain(4..) {
422 assert_eq!(v, &[1, 2, 3, 4]);
424 let mut v: Vec<_> = (1..6).map(|x| x.to_string()).collect();
425 for _ in v.drain(1..4) {
427 assert_eq!(v, &[1.to_string(), 5.to_string()]);
429 let mut v: Vec<_> = (1..6).map(|x| x.to_string()).collect();
430 for _ in v.drain(1..4).rev() {
432 assert_eq!(v, &[1.to_string(), 5.to_string()]);
434 let mut v: Vec<_> = vec![(); 5];
435 for _ in v.drain(1..4).rev() {
437 assert_eq!(v, &[(), ()]);
441 fn test_into_boxed_slice() {
442 let xs = vec![1, 2, 3];
443 let ys = xs.into_boxed_slice();
444 assert_eq!(&*ys, [1, 2, 3]);
449 let mut vec = vec![1, 2, 3];
450 let mut vec2 = vec![4, 5, 6];
451 vec.append(&mut vec2);
452 assert_eq!(vec, [1, 2, 3, 4, 5, 6]);
453 assert_eq!(vec2, []);
457 fn test_split_off() {
458 let mut vec = vec![1, 2, 3, 4, 5, 6];
459 let vec2 = vec.split_off(4);
460 assert_eq!(vec, [1, 2, 3, 4]);
461 assert_eq!(vec2, [5, 6]);
465 fn test_into_iter_count() {
466 assert_eq!(vec![1, 2, 3].into_iter().count(), 3);
470 fn bench_new(b: &mut Bencher) {
472 let v: Vec<u32> = Vec::new();
473 assert_eq!(v.len(), 0);
474 assert_eq!(v.capacity(), 0);
478 fn do_bench_with_capacity(b: &mut Bencher, src_len: usize) {
479 b.bytes = src_len as u64;
482 let v: Vec<u32> = Vec::with_capacity(src_len);
483 assert_eq!(v.len(), 0);
484 assert_eq!(v.capacity(), src_len);
489 fn bench_with_capacity_0000(b: &mut Bencher) {
490 do_bench_with_capacity(b, 0)
494 fn bench_with_capacity_0010(b: &mut Bencher) {
495 do_bench_with_capacity(b, 10)
499 fn bench_with_capacity_0100(b: &mut Bencher) {
500 do_bench_with_capacity(b, 100)
504 fn bench_with_capacity_1000(b: &mut Bencher) {
505 do_bench_with_capacity(b, 1000)
508 fn do_bench_from_fn(b: &mut Bencher, src_len: usize) {
509 b.bytes = src_len as u64;
512 let dst = (0..src_len).collect::<Vec<_>>();
513 assert_eq!(dst.len(), src_len);
514 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
519 fn bench_from_fn_0000(b: &mut Bencher) {
520 do_bench_from_fn(b, 0)
524 fn bench_from_fn_0010(b: &mut Bencher) {
525 do_bench_from_fn(b, 10)
529 fn bench_from_fn_0100(b: &mut Bencher) {
530 do_bench_from_fn(b, 100)
534 fn bench_from_fn_1000(b: &mut Bencher) {
535 do_bench_from_fn(b, 1000)
538 fn do_bench_from_elem(b: &mut Bencher, src_len: usize) {
539 b.bytes = src_len as u64;
542 let dst: Vec<usize> = repeat(5).take(src_len).collect();
543 assert_eq!(dst.len(), src_len);
544 assert!(dst.iter().all(|x| *x == 5));
549 fn bench_from_elem_0000(b: &mut Bencher) {
550 do_bench_from_elem(b, 0)
554 fn bench_from_elem_0010(b: &mut Bencher) {
555 do_bench_from_elem(b, 10)
559 fn bench_from_elem_0100(b: &mut Bencher) {
560 do_bench_from_elem(b, 100)
564 fn bench_from_elem_1000(b: &mut Bencher) {
565 do_bench_from_elem(b, 1000)
568 fn do_bench_from_slice(b: &mut Bencher, src_len: usize) {
569 let src: Vec<_> = FromIterator::from_iter(0..src_len);
571 b.bytes = src_len as u64;
574 let dst = src.clone()[..].to_vec();
575 assert_eq!(dst.len(), src_len);
576 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
581 fn bench_from_slice_0000(b: &mut Bencher) {
582 do_bench_from_slice(b, 0)
586 fn bench_from_slice_0010(b: &mut Bencher) {
587 do_bench_from_slice(b, 10)
591 fn bench_from_slice_0100(b: &mut Bencher) {
592 do_bench_from_slice(b, 100)
596 fn bench_from_slice_1000(b: &mut Bencher) {
597 do_bench_from_slice(b, 1000)
600 fn do_bench_from_iter(b: &mut Bencher, src_len: usize) {
601 let src: Vec<_> = FromIterator::from_iter(0..src_len);
603 b.bytes = src_len as u64;
606 let dst: Vec<_> = FromIterator::from_iter(src.clone());
607 assert_eq!(dst.len(), src_len);
608 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
613 fn bench_from_iter_0000(b: &mut Bencher) {
614 do_bench_from_iter(b, 0)
618 fn bench_from_iter_0010(b: &mut Bencher) {
619 do_bench_from_iter(b, 10)
623 fn bench_from_iter_0100(b: &mut Bencher) {
624 do_bench_from_iter(b, 100)
628 fn bench_from_iter_1000(b: &mut Bencher) {
629 do_bench_from_iter(b, 1000)
632 fn do_bench_extend(b: &mut Bencher, dst_len: usize, src_len: usize) {
633 let dst: Vec<_> = FromIterator::from_iter(0..dst_len);
634 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
636 b.bytes = src_len as u64;
639 let mut dst = dst.clone();
640 dst.extend(src.clone());
641 assert_eq!(dst.len(), dst_len + src_len);
642 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
647 fn bench_extend_0000_0000(b: &mut Bencher) {
648 do_bench_extend(b, 0, 0)
652 fn bench_extend_0000_0010(b: &mut Bencher) {
653 do_bench_extend(b, 0, 10)
657 fn bench_extend_0000_0100(b: &mut Bencher) {
658 do_bench_extend(b, 0, 100)
662 fn bench_extend_0000_1000(b: &mut Bencher) {
663 do_bench_extend(b, 0, 1000)
667 fn bench_extend_0010_0010(b: &mut Bencher) {
668 do_bench_extend(b, 10, 10)
672 fn bench_extend_0100_0100(b: &mut Bencher) {
673 do_bench_extend(b, 100, 100)
677 fn bench_extend_1000_1000(b: &mut Bencher) {
678 do_bench_extend(b, 1000, 1000)
681 fn do_bench_push_all(b: &mut Bencher, dst_len: usize, src_len: usize) {
682 let dst: Vec<_> = FromIterator::from_iter(0..dst_len);
683 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
685 b.bytes = src_len as u64;
688 let mut dst = dst.clone();
690 assert_eq!(dst.len(), dst_len + src_len);
691 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
696 fn bench_push_all_0000_0000(b: &mut Bencher) {
697 do_bench_push_all(b, 0, 0)
701 fn bench_push_all_0000_0010(b: &mut Bencher) {
702 do_bench_push_all(b, 0, 10)
706 fn bench_push_all_0000_0100(b: &mut Bencher) {
707 do_bench_push_all(b, 0, 100)
711 fn bench_push_all_0000_1000(b: &mut Bencher) {
712 do_bench_push_all(b, 0, 1000)
716 fn bench_push_all_0010_0010(b: &mut Bencher) {
717 do_bench_push_all(b, 10, 10)
721 fn bench_push_all_0100_0100(b: &mut Bencher) {
722 do_bench_push_all(b, 100, 100)
726 fn bench_push_all_1000_1000(b: &mut Bencher) {
727 do_bench_push_all(b, 1000, 1000)
730 fn do_bench_push_all_move(b: &mut Bencher, dst_len: usize, src_len: usize) {
731 let dst: Vec<_> = FromIterator::from_iter(0..dst_len);
732 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
734 b.bytes = src_len as u64;
737 let mut dst = dst.clone();
738 dst.extend(src.clone());
739 assert_eq!(dst.len(), dst_len + src_len);
740 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
745 fn bench_push_all_move_0000_0000(b: &mut Bencher) {
746 do_bench_push_all_move(b, 0, 0)
750 fn bench_push_all_move_0000_0010(b: &mut Bencher) {
751 do_bench_push_all_move(b, 0, 10)
755 fn bench_push_all_move_0000_0100(b: &mut Bencher) {
756 do_bench_push_all_move(b, 0, 100)
760 fn bench_push_all_move_0000_1000(b: &mut Bencher) {
761 do_bench_push_all_move(b, 0, 1000)
765 fn bench_push_all_move_0010_0010(b: &mut Bencher) {
766 do_bench_push_all_move(b, 10, 10)
770 fn bench_push_all_move_0100_0100(b: &mut Bencher) {
771 do_bench_push_all_move(b, 100, 100)
775 fn bench_push_all_move_1000_1000(b: &mut Bencher) {
776 do_bench_push_all_move(b, 1000, 1000)
779 fn do_bench_clone(b: &mut Bencher, src_len: usize) {
780 let src: Vec<usize> = FromIterator::from_iter(0..src_len);
782 b.bytes = src_len as u64;
785 let dst = src.clone();
786 assert_eq!(dst.len(), src_len);
787 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
792 fn bench_clone_0000(b: &mut Bencher) {
797 fn bench_clone_0010(b: &mut Bencher) {
798 do_bench_clone(b, 10)
802 fn bench_clone_0100(b: &mut Bencher) {
803 do_bench_clone(b, 100)
807 fn bench_clone_1000(b: &mut Bencher) {
808 do_bench_clone(b, 1000)
811 fn do_bench_clone_from(b: &mut Bencher, times: usize, dst_len: usize, src_len: usize) {
812 let dst: Vec<_> = FromIterator::from_iter(0..src_len);
813 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
815 b.bytes = (times * src_len) as u64;
818 let mut dst = dst.clone();
821 dst.clone_from(&src);
823 assert_eq!(dst.len(), src_len);
824 assert!(dst.iter().enumerate().all(|(i, x)| dst_len + i == *x));
830 fn bench_clone_from_01_0000_0000(b: &mut Bencher) {
831 do_bench_clone_from(b, 1, 0, 0)
835 fn bench_clone_from_01_0000_0010(b: &mut Bencher) {
836 do_bench_clone_from(b, 1, 0, 10)
840 fn bench_clone_from_01_0000_0100(b: &mut Bencher) {
841 do_bench_clone_from(b, 1, 0, 100)
845 fn bench_clone_from_01_0000_1000(b: &mut Bencher) {
846 do_bench_clone_from(b, 1, 0, 1000)
850 fn bench_clone_from_01_0010_0010(b: &mut Bencher) {
851 do_bench_clone_from(b, 1, 10, 10)
855 fn bench_clone_from_01_0100_0100(b: &mut Bencher) {
856 do_bench_clone_from(b, 1, 100, 100)
860 fn bench_clone_from_01_1000_1000(b: &mut Bencher) {
861 do_bench_clone_from(b, 1, 1000, 1000)
865 fn bench_clone_from_01_0010_0100(b: &mut Bencher) {
866 do_bench_clone_from(b, 1, 10, 100)
870 fn bench_clone_from_01_0100_1000(b: &mut Bencher) {
871 do_bench_clone_from(b, 1, 100, 1000)
875 fn bench_clone_from_01_0010_0000(b: &mut Bencher) {
876 do_bench_clone_from(b, 1, 10, 0)
880 fn bench_clone_from_01_0100_0010(b: &mut Bencher) {
881 do_bench_clone_from(b, 1, 100, 10)
885 fn bench_clone_from_01_1000_0100(b: &mut Bencher) {
886 do_bench_clone_from(b, 1, 1000, 100)
890 fn bench_clone_from_10_0000_0000(b: &mut Bencher) {
891 do_bench_clone_from(b, 10, 0, 0)
895 fn bench_clone_from_10_0000_0010(b: &mut Bencher) {
896 do_bench_clone_from(b, 10, 0, 10)
900 fn bench_clone_from_10_0000_0100(b: &mut Bencher) {
901 do_bench_clone_from(b, 10, 0, 100)
905 fn bench_clone_from_10_0000_1000(b: &mut Bencher) {
906 do_bench_clone_from(b, 10, 0, 1000)
910 fn bench_clone_from_10_0010_0010(b: &mut Bencher) {
911 do_bench_clone_from(b, 10, 10, 10)
915 fn bench_clone_from_10_0100_0100(b: &mut Bencher) {
916 do_bench_clone_from(b, 10, 100, 100)
920 fn bench_clone_from_10_1000_1000(b: &mut Bencher) {
921 do_bench_clone_from(b, 10, 1000, 1000)
925 fn bench_clone_from_10_0010_0100(b: &mut Bencher) {
926 do_bench_clone_from(b, 10, 10, 100)
930 fn bench_clone_from_10_0100_1000(b: &mut Bencher) {
931 do_bench_clone_from(b, 10, 100, 1000)
935 fn bench_clone_from_10_0010_0000(b: &mut Bencher) {
936 do_bench_clone_from(b, 10, 10, 0)
940 fn bench_clone_from_10_0100_0010(b: &mut Bencher) {
941 do_bench_clone_from(b, 10, 100, 10)
945 fn bench_clone_from_10_1000_0100(b: &mut Bencher) {
946 do_bench_clone_from(b, 10, 1000, 100)