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.
12 use std::iter::{FromIterator, repeat};
13 use std::mem::size_of;
17 struct DropCounter<'a> {
21 impl<'a> Drop for DropCounter<'a> {
28 fn test_small_vec_struct() {
29 assert!(size_of::<Vec<u8>>() == size_of::<usize>() * 3);
33 fn test_double_drop() {
39 let (mut count_x, mut count_y) = (0, 0);
45 tv.x.push(DropCounter {count: &mut count_x});
46 tv.y.push(DropCounter {count: &mut count_y});
48 // If Vec had a drop flag, here is where it would be zeroed.
49 // Instead, it should rely on its internal state to prevent
50 // doing anything significant when dropped multiple times.
53 // Here tv goes out of scope, tv.y should be dropped, but not tv.x.
56 assert_eq!(count_x, 1);
57 assert_eq!(count_y, 1);
62 let mut v = Vec::new();
63 assert_eq!(v.capacity(), 0);
66 assert!(v.capacity() >= 2);
72 assert!(v.capacity() >= 16);
74 assert!(v.capacity() >= 32);
79 assert!(v.capacity() >= 33)
84 let mut v = Vec::new();
85 let mut w = Vec::new();
88 for i in 0..3 { w.push(i) }
93 for i in 3..10 { w.push(i) }
99 fn test_extend_ref() {
100 let mut v = vec![1, 2];
101 v.extend(&[3, 4, 5]);
103 assert_eq!(v.len(), 5);
104 assert_eq!(v, [1, 2, 3, 4, 5]);
109 assert_eq!(v.len(), 7);
110 assert_eq!(v, [1, 2, 3, 4, 5, 6, 7]);
114 fn test_slice_from_mut() {
115 let mut values = vec![1, 2, 3, 4, 5];
117 let slice = &mut values[2 ..];
118 assert!(slice == [3, 4, 5]);
124 assert!(values == [1, 2, 5, 6, 7]);
128 fn test_slice_to_mut() {
129 let mut values = vec![1, 2, 3, 4, 5];
131 let slice = &mut values[.. 2];
132 assert!(slice == [1, 2]);
138 assert!(values == [2, 3, 3, 4, 5]);
142 fn test_split_at_mut() {
143 let mut values = vec![1, 2, 3, 4, 5];
145 let (left, right) = values.split_at_mut(2);
147 let left: &[_] = left;
148 assert!(&left[..left.len()] == &[1, 2]);
155 let right: &[_] = right;
156 assert!(&right[..right.len()] == &[3, 4, 5]);
163 assert_eq!(values, [2, 3, 5, 6, 7]);
168 let v: Vec<i32> = vec![];
169 let w = vec!(1, 2, 3);
171 assert_eq!(v, v.clone());
175 // they should be disjoint in memory.
176 assert!(w.as_ptr() != z.as_ptr())
180 fn test_clone_from() {
182 let three: Vec<Box<_>> = vec!(box 1, box 2, box 3);
183 let two: Vec<Box<_>> = vec!(box 4, box 5);
185 v.clone_from(&three);
186 assert_eq!(v, three);
189 v.clone_from(&three);
190 assert_eq!(v, three);
197 v.clone_from(&three);
203 let mut vec = vec![1, 2, 3, 4];
204 vec.retain(|&x| x % 2 == 0);
205 assert_eq!(vec, [2, 4]);
209 fn zero_sized_values() {
210 let mut v = Vec::new();
211 assert_eq!(v.len(), 0);
213 assert_eq!(v.len(), 1);
215 assert_eq!(v.len(), 2);
216 assert_eq!(v.pop(), Some(()));
217 assert_eq!(v.pop(), Some(()));
218 assert_eq!(v.pop(), None);
220 assert_eq!(v.iter().count(), 0);
222 assert_eq!(v.iter().count(), 1);
224 assert_eq!(v.iter().count(), 2);
228 assert_eq!(v.iter_mut().count(), 2);
230 assert_eq!(v.iter_mut().count(), 3);
232 assert_eq!(v.iter_mut().count(), 4);
234 for &mut () in &mut v {}
235 unsafe { v.set_len(0); }
236 assert_eq!(v.iter_mut().count(), 0);
240 fn test_partition() {
241 assert_eq!(vec![].into_iter().partition(|x: &i32| *x < 3), (vec![], vec![]));
242 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 4), (vec![1, 2, 3], vec![]));
243 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 2), (vec![1], vec![2, 3]));
244 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 0), (vec![], vec![1, 2, 3]));
248 fn test_zip_unzip() {
249 let z1 = vec![(1, 4), (2, 5), (3, 6)];
251 let (left, right): (Vec<_>, Vec<_>) = z1.iter().cloned().unzip();
253 assert_eq!((1, 4), (left[0], right[0]));
254 assert_eq!((2, 5), (left[1], right[1]));
255 assert_eq!((3, 6), (left[2], right[2]));
259 fn test_vec_truncate_drop() {
260 static mut drops: u32 = 0;
264 unsafe { drops += 1; }
268 let mut v = vec![Elem(1), Elem(2), Elem(3), Elem(4), Elem(5)];
269 assert_eq!(unsafe { drops }, 0);
271 assert_eq!(unsafe { drops }, 2);
273 assert_eq!(unsafe { drops }, 5);
278 fn test_vec_truncate_fail() {
280 impl Drop for BadElem {
282 let BadElem(ref mut x) = *self;
284 panic!("BadElem panic: 0xbadbeef")
289 let mut v = vec![BadElem(1), BadElem(2), BadElem(0xbadbeef), BadElem(4)];
295 let vec = vec![1, 2, 3];
296 assert!(vec[1] == 2);
301 fn test_index_out_of_bounds() {
302 let vec = vec![1, 2, 3];
308 fn test_slice_out_of_bounds_1() {
309 let x = vec![1, 2, 3, 4, 5];
315 fn test_slice_out_of_bounds_2() {
316 let x = vec![1, 2, 3, 4, 5];
322 fn test_slice_out_of_bounds_3() {
323 let x = vec![1, 2, 3, 4, 5];
329 fn test_slice_out_of_bounds_4() {
330 let x = vec![1, 2, 3, 4, 5];
336 fn test_slice_out_of_bounds_5() {
337 let x = vec![1, 2, 3, 4, 5];
343 fn test_swap_remove_empty() {
344 let mut vec= Vec::<i32>::new();
349 fn test_move_items() {
350 let vec = vec![1, 2, 3];
351 let mut vec2 = vec![];
355 assert_eq!(vec2, [1, 2, 3]);
359 fn test_move_items_reverse() {
360 let vec = vec![1, 2, 3];
361 let mut vec2 = vec![];
362 for i in vec.into_iter().rev() {
365 assert_eq!(vec2, [3, 2, 1]);
369 fn test_move_items_zero_sized() {
370 let vec = vec![(), (), ()];
371 let mut vec2 = vec![];
375 assert_eq!(vec2, [(), (), ()]);
379 fn test_drain_items() {
380 let mut vec = vec![1, 2, 3];
381 let mut vec2 = vec![];
382 for i in vec.drain(..) {
386 assert_eq!(vec2, [ 1, 2, 3 ]);
390 fn test_drain_items_reverse() {
391 let mut vec = vec![1, 2, 3];
392 let mut vec2 = vec![];
393 for i in vec.drain(..).rev() {
397 assert_eq!(vec2, [3, 2, 1]);
401 fn test_drain_items_zero_sized() {
402 let mut vec = vec![(), (), ()];
403 let mut vec2 = vec![];
404 for i in vec.drain(..) {
408 assert_eq!(vec2, [(), (), ()]);
413 fn test_drain_out_of_bounds() {
414 let mut v = vec![1, 2, 3, 4, 5];
419 fn test_drain_range() {
420 let mut v = vec![1, 2, 3, 4, 5];
421 for _ in v.drain(4..) {
423 assert_eq!(v, &[1, 2, 3, 4]);
425 let mut v: Vec<_> = (1..6).map(|x| x.to_string()).collect();
426 for _ in v.drain(1..4) {
428 assert_eq!(v, &[1.to_string(), 5.to_string()]);
430 let mut v: Vec<_> = (1..6).map(|x| x.to_string()).collect();
431 for _ in v.drain(1..4).rev() {
433 assert_eq!(v, &[1.to_string(), 5.to_string()]);
435 let mut v: Vec<_> = vec![(); 5];
436 for _ in v.drain(1..4).rev() {
438 assert_eq!(v, &[(), ()]);
442 fn test_into_boxed_slice() {
443 let xs = vec![1, 2, 3];
444 let ys = xs.into_boxed_slice();
445 assert_eq!(&*ys, [1, 2, 3]);
450 let mut vec = vec![1, 2, 3];
451 let mut vec2 = vec![4, 5, 6];
452 vec.append(&mut vec2);
453 assert_eq!(vec, [1, 2, 3, 4, 5, 6]);
454 assert_eq!(vec2, []);
458 fn test_split_off() {
459 let mut vec = vec![1, 2, 3, 4, 5, 6];
460 let vec2 = vec.split_off(4);
461 assert_eq!(vec, [1, 2, 3, 4]);
462 assert_eq!(vec2, [5, 6]);
466 fn test_into_iter_count() {
467 assert_eq!(vec![1, 2, 3].into_iter().count(), 3);
472 let borrowed: &[_] = &["borrowed", "(slice)"];
473 let owned = vec!["owned", "(vec)"];
474 match (Cow::from(owned.clone()), Cow::from(borrowed)) {
475 (Cow::Owned(o), Cow::Borrowed(b)) => assert!(o == owned && b == borrowed),
476 _ => panic!("invalid `Cow::from`"),
481 fn bench_new(b: &mut Bencher) {
483 let v: Vec<u32> = Vec::new();
484 assert_eq!(v.len(), 0);
485 assert_eq!(v.capacity(), 0);
489 fn do_bench_with_capacity(b: &mut Bencher, src_len: usize) {
490 b.bytes = src_len as u64;
493 let v: Vec<u32> = Vec::with_capacity(src_len);
494 assert_eq!(v.len(), 0);
495 assert_eq!(v.capacity(), src_len);
500 fn bench_with_capacity_0000(b: &mut Bencher) {
501 do_bench_with_capacity(b, 0)
505 fn bench_with_capacity_0010(b: &mut Bencher) {
506 do_bench_with_capacity(b, 10)
510 fn bench_with_capacity_0100(b: &mut Bencher) {
511 do_bench_with_capacity(b, 100)
515 fn bench_with_capacity_1000(b: &mut Bencher) {
516 do_bench_with_capacity(b, 1000)
519 fn do_bench_from_fn(b: &mut Bencher, src_len: usize) {
520 b.bytes = src_len as u64;
523 let dst = (0..src_len).collect::<Vec<_>>();
524 assert_eq!(dst.len(), src_len);
525 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
530 fn bench_from_fn_0000(b: &mut Bencher) {
531 do_bench_from_fn(b, 0)
535 fn bench_from_fn_0010(b: &mut Bencher) {
536 do_bench_from_fn(b, 10)
540 fn bench_from_fn_0100(b: &mut Bencher) {
541 do_bench_from_fn(b, 100)
545 fn bench_from_fn_1000(b: &mut Bencher) {
546 do_bench_from_fn(b, 1000)
549 fn do_bench_from_elem(b: &mut Bencher, src_len: usize) {
550 b.bytes = src_len as u64;
553 let dst: Vec<usize> = repeat(5).take(src_len).collect();
554 assert_eq!(dst.len(), src_len);
555 assert!(dst.iter().all(|x| *x == 5));
560 fn bench_from_elem_0000(b: &mut Bencher) {
561 do_bench_from_elem(b, 0)
565 fn bench_from_elem_0010(b: &mut Bencher) {
566 do_bench_from_elem(b, 10)
570 fn bench_from_elem_0100(b: &mut Bencher) {
571 do_bench_from_elem(b, 100)
575 fn bench_from_elem_1000(b: &mut Bencher) {
576 do_bench_from_elem(b, 1000)
579 fn do_bench_from_slice(b: &mut Bencher, src_len: usize) {
580 let src: Vec<_> = FromIterator::from_iter(0..src_len);
582 b.bytes = src_len as u64;
585 let dst = src.clone()[..].to_vec();
586 assert_eq!(dst.len(), src_len);
587 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
592 fn bench_from_slice_0000(b: &mut Bencher) {
593 do_bench_from_slice(b, 0)
597 fn bench_from_slice_0010(b: &mut Bencher) {
598 do_bench_from_slice(b, 10)
602 fn bench_from_slice_0100(b: &mut Bencher) {
603 do_bench_from_slice(b, 100)
607 fn bench_from_slice_1000(b: &mut Bencher) {
608 do_bench_from_slice(b, 1000)
611 fn do_bench_from_iter(b: &mut Bencher, src_len: usize) {
612 let src: Vec<_> = FromIterator::from_iter(0..src_len);
614 b.bytes = src_len as u64;
617 let dst: Vec<_> = FromIterator::from_iter(src.clone());
618 assert_eq!(dst.len(), src_len);
619 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
624 fn bench_from_iter_0000(b: &mut Bencher) {
625 do_bench_from_iter(b, 0)
629 fn bench_from_iter_0010(b: &mut Bencher) {
630 do_bench_from_iter(b, 10)
634 fn bench_from_iter_0100(b: &mut Bencher) {
635 do_bench_from_iter(b, 100)
639 fn bench_from_iter_1000(b: &mut Bencher) {
640 do_bench_from_iter(b, 1000)
643 fn do_bench_extend(b: &mut Bencher, dst_len: usize, src_len: usize) {
644 let dst: Vec<_> = FromIterator::from_iter(0..dst_len);
645 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
647 b.bytes = src_len as u64;
650 let mut dst = dst.clone();
651 dst.extend(src.clone());
652 assert_eq!(dst.len(), dst_len + src_len);
653 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
658 fn bench_extend_0000_0000(b: &mut Bencher) {
659 do_bench_extend(b, 0, 0)
663 fn bench_extend_0000_0010(b: &mut Bencher) {
664 do_bench_extend(b, 0, 10)
668 fn bench_extend_0000_0100(b: &mut Bencher) {
669 do_bench_extend(b, 0, 100)
673 fn bench_extend_0000_1000(b: &mut Bencher) {
674 do_bench_extend(b, 0, 1000)
678 fn bench_extend_0010_0010(b: &mut Bencher) {
679 do_bench_extend(b, 10, 10)
683 fn bench_extend_0100_0100(b: &mut Bencher) {
684 do_bench_extend(b, 100, 100)
688 fn bench_extend_1000_1000(b: &mut Bencher) {
689 do_bench_extend(b, 1000, 1000)
692 fn do_bench_push_all(b: &mut Bencher, dst_len: usize, src_len: usize) {
693 let dst: Vec<_> = FromIterator::from_iter(0..dst_len);
694 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
696 b.bytes = src_len as u64;
699 let mut dst = dst.clone();
700 dst.extend_from_slice(&src);
701 assert_eq!(dst.len(), dst_len + src_len);
702 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
707 fn bench_push_all_0000_0000(b: &mut Bencher) {
708 do_bench_push_all(b, 0, 0)
712 fn bench_push_all_0000_0010(b: &mut Bencher) {
713 do_bench_push_all(b, 0, 10)
717 fn bench_push_all_0000_0100(b: &mut Bencher) {
718 do_bench_push_all(b, 0, 100)
722 fn bench_push_all_0000_1000(b: &mut Bencher) {
723 do_bench_push_all(b, 0, 1000)
727 fn bench_push_all_0010_0010(b: &mut Bencher) {
728 do_bench_push_all(b, 10, 10)
732 fn bench_push_all_0100_0100(b: &mut Bencher) {
733 do_bench_push_all(b, 100, 100)
737 fn bench_push_all_1000_1000(b: &mut Bencher) {
738 do_bench_push_all(b, 1000, 1000)
741 fn do_bench_push_all_move(b: &mut Bencher, dst_len: usize, src_len: usize) {
742 let dst: Vec<_> = FromIterator::from_iter(0..dst_len);
743 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
745 b.bytes = src_len as u64;
748 let mut dst = dst.clone();
749 dst.extend(src.clone());
750 assert_eq!(dst.len(), dst_len + src_len);
751 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
756 fn bench_push_all_move_0000_0000(b: &mut Bencher) {
757 do_bench_push_all_move(b, 0, 0)
761 fn bench_push_all_move_0000_0010(b: &mut Bencher) {
762 do_bench_push_all_move(b, 0, 10)
766 fn bench_push_all_move_0000_0100(b: &mut Bencher) {
767 do_bench_push_all_move(b, 0, 100)
771 fn bench_push_all_move_0000_1000(b: &mut Bencher) {
772 do_bench_push_all_move(b, 0, 1000)
776 fn bench_push_all_move_0010_0010(b: &mut Bencher) {
777 do_bench_push_all_move(b, 10, 10)
781 fn bench_push_all_move_0100_0100(b: &mut Bencher) {
782 do_bench_push_all_move(b, 100, 100)
786 fn bench_push_all_move_1000_1000(b: &mut Bencher) {
787 do_bench_push_all_move(b, 1000, 1000)
790 fn do_bench_clone(b: &mut Bencher, src_len: usize) {
791 let src: Vec<usize> = FromIterator::from_iter(0..src_len);
793 b.bytes = src_len as u64;
796 let dst = src.clone();
797 assert_eq!(dst.len(), src_len);
798 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
803 fn bench_clone_0000(b: &mut Bencher) {
808 fn bench_clone_0010(b: &mut Bencher) {
809 do_bench_clone(b, 10)
813 fn bench_clone_0100(b: &mut Bencher) {
814 do_bench_clone(b, 100)
818 fn bench_clone_1000(b: &mut Bencher) {
819 do_bench_clone(b, 1000)
822 fn do_bench_clone_from(b: &mut Bencher, times: usize, dst_len: usize, src_len: usize) {
823 let dst: Vec<_> = FromIterator::from_iter(0..src_len);
824 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
826 b.bytes = (times * src_len) as u64;
829 let mut dst = dst.clone();
832 dst.clone_from(&src);
834 assert_eq!(dst.len(), src_len);
835 assert!(dst.iter().enumerate().all(|(i, x)| dst_len + i == *x));
841 fn bench_clone_from_01_0000_0000(b: &mut Bencher) {
842 do_bench_clone_from(b, 1, 0, 0)
846 fn bench_clone_from_01_0000_0010(b: &mut Bencher) {
847 do_bench_clone_from(b, 1, 0, 10)
851 fn bench_clone_from_01_0000_0100(b: &mut Bencher) {
852 do_bench_clone_from(b, 1, 0, 100)
856 fn bench_clone_from_01_0000_1000(b: &mut Bencher) {
857 do_bench_clone_from(b, 1, 0, 1000)
861 fn bench_clone_from_01_0010_0010(b: &mut Bencher) {
862 do_bench_clone_from(b, 1, 10, 10)
866 fn bench_clone_from_01_0100_0100(b: &mut Bencher) {
867 do_bench_clone_from(b, 1, 100, 100)
871 fn bench_clone_from_01_1000_1000(b: &mut Bencher) {
872 do_bench_clone_from(b, 1, 1000, 1000)
876 fn bench_clone_from_01_0010_0100(b: &mut Bencher) {
877 do_bench_clone_from(b, 1, 10, 100)
881 fn bench_clone_from_01_0100_1000(b: &mut Bencher) {
882 do_bench_clone_from(b, 1, 100, 1000)
886 fn bench_clone_from_01_0010_0000(b: &mut Bencher) {
887 do_bench_clone_from(b, 1, 10, 0)
891 fn bench_clone_from_01_0100_0010(b: &mut Bencher) {
892 do_bench_clone_from(b, 1, 100, 10)
896 fn bench_clone_from_01_1000_0100(b: &mut Bencher) {
897 do_bench_clone_from(b, 1, 1000, 100)
901 fn bench_clone_from_10_0000_0000(b: &mut Bencher) {
902 do_bench_clone_from(b, 10, 0, 0)
906 fn bench_clone_from_10_0000_0010(b: &mut Bencher) {
907 do_bench_clone_from(b, 10, 0, 10)
911 fn bench_clone_from_10_0000_0100(b: &mut Bencher) {
912 do_bench_clone_from(b, 10, 0, 100)
916 fn bench_clone_from_10_0000_1000(b: &mut Bencher) {
917 do_bench_clone_from(b, 10, 0, 1000)
921 fn bench_clone_from_10_0010_0010(b: &mut Bencher) {
922 do_bench_clone_from(b, 10, 10, 10)
926 fn bench_clone_from_10_0100_0100(b: &mut Bencher) {
927 do_bench_clone_from(b, 10, 100, 100)
931 fn bench_clone_from_10_1000_1000(b: &mut Bencher) {
932 do_bench_clone_from(b, 10, 1000, 1000)
936 fn bench_clone_from_10_0010_0100(b: &mut Bencher) {
937 do_bench_clone_from(b, 10, 10, 100)
941 fn bench_clone_from_10_0100_1000(b: &mut Bencher) {
942 do_bench_clone_from(b, 10, 100, 1000)
946 fn bench_clone_from_10_0010_0000(b: &mut Bencher) {
947 do_bench_clone_from(b, 10, 10, 0)
951 fn bench_clone_from_10_0100_0010(b: &mut Bencher) {
952 do_bench_clone_from(b, 10, 100, 10)
956 fn bench_clone_from_10_1000_0100(b: &mut Bencher) {
957 do_bench_clone_from(b, 10, 1000, 100)