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;
17 struct DropCounter<'a> {
22 impl<'a> Drop for DropCounter<'a> {
30 let xs = [1u8, 2u8, 3u8];
31 assert_eq!(&**as_vec(&xs), xs);
35 fn test_as_vec_dtor() {
36 let (mut count_x, mut count_y) = (0, 0);
38 let xs = &[DropCounter { count: &mut count_x }, DropCounter { count: &mut count_y }];
39 assert_eq!(as_vec(xs).len(), 2);
41 assert_eq!(count_x, 1);
42 assert_eq!(count_y, 1);
46 fn test_small_vec_struct() {
47 assert!(size_of::<Vec<u8>>() == size_of::<usize>() * 3);
51 fn test_double_drop() {
57 let (mut count_x, mut count_y) = (0, 0);
63 tv.x.push(DropCounter {count: &mut count_x});
64 tv.y.push(DropCounter {count: &mut count_y});
66 // If Vec had a drop flag, here is where it would be zeroed.
67 // Instead, it should rely on its internal state to prevent
68 // doing anything significant when dropped multiple times.
71 // Here tv goes out of scope, tv.y should be dropped, but not tv.x.
74 assert_eq!(count_x, 1);
75 assert_eq!(count_y, 1);
80 let mut v = Vec::new();
81 assert_eq!(v.capacity(), 0);
84 assert!(v.capacity() >= 2);
90 assert!(v.capacity() >= 16);
92 assert!(v.capacity() >= 32);
97 assert!(v.capacity() >= 33)
102 let mut v = Vec::new();
103 let mut w = Vec::new();
106 for i in 0..3 { w.push(i) }
111 for i in 3..10 { w.push(i) }
117 fn test_slice_from_mut() {
118 let mut values = vec![1, 2, 3, 4, 5];
120 let slice = &mut values[2 ..];
121 assert!(slice == [3, 4, 5]);
127 assert!(values == [1, 2, 5, 6, 7]);
131 fn test_slice_to_mut() {
132 let mut values = vec![1, 2, 3, 4, 5];
134 let slice = &mut values[.. 2];
135 assert!(slice == [1, 2]);
141 assert!(values == [2, 3, 3, 4, 5]);
145 fn test_split_at_mut() {
146 let mut values = vec![1, 2, 3, 4, 5];
148 let (left, right) = values.split_at_mut(2);
150 let left: &[_] = left;
151 assert!(&left[..left.len()] == &[1, 2]);
158 let right: &[_] = right;
159 assert!(&right[..right.len()] == &[3, 4, 5]);
166 assert_eq!(values, [2, 3, 5, 6, 7]);
171 let v: Vec<i32> = vec![];
172 let w = vec!(1, 2, 3);
174 assert_eq!(v, v.clone());
178 // they should be disjoint in memory.
179 assert!(w.as_ptr() != z.as_ptr())
183 fn test_clone_from() {
185 let three: Vec<Box<_>> = vec!(box 1, box 2, box 3);
186 let two: Vec<Box<_>> = vec!(box 4, box 5);
188 v.clone_from(&three);
189 assert_eq!(v, three);
192 v.clone_from(&three);
193 assert_eq!(v, three);
200 v.clone_from(&three);
206 let mut vec = vec![1, 2, 3, 4];
207 vec.retain(|&x| x % 2 == 0);
208 assert_eq!(vec, [2, 4]);
212 fn zero_sized_values() {
213 let mut v = Vec::new();
214 assert_eq!(v.len(), 0);
216 assert_eq!(v.len(), 1);
218 assert_eq!(v.len(), 2);
219 assert_eq!(v.pop(), Some(()));
220 assert_eq!(v.pop(), Some(()));
221 assert_eq!(v.pop(), None);
223 assert_eq!(v.iter().count(), 0);
225 assert_eq!(v.iter().count(), 1);
227 assert_eq!(v.iter().count(), 2);
231 assert_eq!(v.iter_mut().count(), 2);
233 assert_eq!(v.iter_mut().count(), 3);
235 assert_eq!(v.iter_mut().count(), 4);
237 for &mut () in &mut v {}
238 unsafe { v.set_len(0); }
239 assert_eq!(v.iter_mut().count(), 0);
243 fn test_partition() {
244 assert_eq!(vec![].into_iter().partition(|x: &i32| *x < 3), (vec![], vec![]));
245 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 4), (vec![1, 2, 3], vec![]));
246 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 2), (vec![1], vec![2, 3]));
247 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 0), (vec![], vec![1, 2, 3]));
251 fn test_zip_unzip() {
252 let z1 = vec![(1, 4), (2, 5), (3, 6)];
254 let (left, right): (Vec<_>, Vec<_>) = z1.iter().cloned().unzip();
256 assert_eq!((1, 4), (left[0], right[0]));
257 assert_eq!((2, 5), (left[1], right[1]));
258 assert_eq!((3, 6), (left[2], right[2]));
262 fn test_unsafe_ptrs() {
264 // Test on-stack copy-from-buf.
266 let ptr = a.as_ptr();
267 let b = Vec::from_raw_buf(ptr, 3);
268 assert_eq!(b, [1, 2, 3]);
270 // Test on-heap copy-from-buf.
271 let c = vec![1, 2, 3, 4, 5];
272 let ptr = c.as_ptr();
273 let d = Vec::from_raw_buf(ptr, 5);
274 assert_eq!(d, [1, 2, 3, 4, 5]);
279 fn test_vec_truncate_drop() {
280 static mut drops: u32 = 0;
284 unsafe { drops += 1; }
288 let mut v = vec![Elem(1), Elem(2), Elem(3), Elem(4), Elem(5)];
289 assert_eq!(unsafe { drops }, 0);
291 assert_eq!(unsafe { drops }, 2);
293 assert_eq!(unsafe { drops }, 5);
298 fn test_vec_truncate_fail() {
300 impl Drop for BadElem {
302 let BadElem(ref mut x) = *self;
304 panic!("BadElem panic: 0xbadbeef")
309 let mut v = vec![BadElem(1), BadElem(2), BadElem(0xbadbeef), BadElem(4)];
315 let vec = vec![1, 2, 3];
316 assert!(vec[1] == 2);
321 fn test_index_out_of_bounds() {
322 let vec = vec![1, 2, 3];
328 fn test_slice_out_of_bounds_1() {
329 let x = vec![1, 2, 3, 4, 5];
335 fn test_slice_out_of_bounds_2() {
336 let x = vec![1, 2, 3, 4, 5];
342 fn test_slice_out_of_bounds_3() {
343 let x = vec![1, 2, 3, 4, 5];
349 fn test_slice_out_of_bounds_4() {
350 let x = vec![1, 2, 3, 4, 5];
356 fn test_slice_out_of_bounds_5() {
357 let x = vec![1, 2, 3, 4, 5];
363 fn test_swap_remove_empty() {
364 let mut vec= Vec::<i32>::new();
369 fn test_move_iter_unwrap() {
370 let mut vec = Vec::with_capacity(7);
373 let ptr = vec.as_ptr();
374 vec = vec.into_iter().into_inner();
375 assert_eq!(vec.as_ptr(), ptr);
376 assert_eq!(vec.capacity(), 7);
377 assert_eq!(vec.len(), 0);
382 fn test_map_in_place_incompatible_types_fail() {
383 let v = vec![0, 1, 2];
384 v.map_in_place(|_| ());
388 fn test_map_in_place() {
389 let v = vec![0, 1, 2];
390 assert_eq!(v.map_in_place(|i: u32| i as i32 - 1), [-1, 0, 1]);
394 fn test_map_in_place_zero_sized() {
395 let v = vec![(), ()];
396 #[derive(PartialEq, Debug)]
398 assert_eq!(v.map_in_place(|_| ZeroSized), [ZeroSized, ZeroSized]);
402 fn test_map_in_place_zero_drop_count() {
403 use std::sync::atomic::{AtomicUsize, Ordering, ATOMIC_USIZE_INIT};
405 #[derive(Clone, PartialEq, Debug)]
407 impl Drop for Nothing { fn drop(&mut self) { } }
409 #[derive(Clone, PartialEq, Debug)]
411 impl Drop for ZeroSized {
413 DROP_COUNTER.fetch_add(1, Ordering::Relaxed);
416 const NUM_ELEMENTS: usize = 2;
417 static DROP_COUNTER: AtomicUsize = ATOMIC_USIZE_INIT;
419 let v = repeat(Nothing).take(NUM_ELEMENTS).collect::<Vec<_>>();
421 DROP_COUNTER.store(0, Ordering::Relaxed);
423 let v = v.map_in_place(|_| ZeroSized);
424 assert_eq!(DROP_COUNTER.load(Ordering::Relaxed), 0);
426 assert_eq!(DROP_COUNTER.load(Ordering::Relaxed), NUM_ELEMENTS);
430 fn test_move_items() {
431 let vec = vec![1, 2, 3];
432 let mut vec2 = vec![];
436 assert_eq!(vec2, [1, 2, 3]);
440 fn test_move_items_reverse() {
441 let vec = vec![1, 2, 3];
442 let mut vec2 = vec![];
443 for i in vec.into_iter().rev() {
446 assert_eq!(vec2, [3, 2, 1]);
450 fn test_move_items_zero_sized() {
451 let vec = vec![(), (), ()];
452 let mut vec2 = vec![];
456 assert_eq!(vec2, [(), (), ()]);
460 fn test_drain_items() {
461 let mut vec = vec![1, 2, 3];
462 let mut vec2 = vec![];
463 for i in vec.drain() {
467 assert_eq!(vec2, [ 1, 2, 3 ]);
471 fn test_drain_items_reverse() {
472 let mut vec = vec![1, 2, 3];
473 let mut vec2 = vec![];
474 for i in vec.drain().rev() {
478 assert_eq!(vec2, [3, 2, 1]);
482 fn test_drain_items_zero_sized() {
483 let mut vec = vec![(), (), ()];
484 let mut vec2 = vec![];
485 for i in vec.drain() {
489 assert_eq!(vec2, [(), (), ()]);
493 fn test_into_boxed_slice() {
494 let xs = vec![1, 2, 3];
495 let ys = xs.into_boxed_slice();
496 assert_eq!(&*ys, [1, 2, 3]);
501 let mut vec = vec![1, 2, 3];
502 let mut vec2 = vec![4, 5, 6];
503 vec.append(&mut vec2);
504 assert_eq!(vec, [1, 2, 3, 4, 5, 6]);
505 assert_eq!(vec2, []);
509 fn test_split_off() {
510 let mut vec = vec![1, 2, 3, 4, 5, 6];
511 let vec2 = vec.split_off(4);
512 assert_eq!(vec, [1, 2, 3, 4]);
513 assert_eq!(vec2, [5, 6]);
517 fn bench_new(b: &mut Bencher) {
519 let v: Vec<u32> = Vec::new();
520 assert_eq!(v.len(), 0);
521 assert_eq!(v.capacity(), 0);
525 fn do_bench_with_capacity(b: &mut Bencher, src_len: usize) {
526 b.bytes = src_len as u64;
529 let v: Vec<u32> = Vec::with_capacity(src_len);
530 assert_eq!(v.len(), 0);
531 assert_eq!(v.capacity(), src_len);
536 fn bench_with_capacity_0000(b: &mut Bencher) {
537 do_bench_with_capacity(b, 0)
541 fn bench_with_capacity_0010(b: &mut Bencher) {
542 do_bench_with_capacity(b, 10)
546 fn bench_with_capacity_0100(b: &mut Bencher) {
547 do_bench_with_capacity(b, 100)
551 fn bench_with_capacity_1000(b: &mut Bencher) {
552 do_bench_with_capacity(b, 1000)
555 fn do_bench_from_fn(b: &mut Bencher, src_len: usize) {
556 b.bytes = src_len as u64;
559 let dst = (0..src_len).collect::<Vec<_>>();
560 assert_eq!(dst.len(), src_len);
561 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
566 fn bench_from_fn_0000(b: &mut Bencher) {
567 do_bench_from_fn(b, 0)
571 fn bench_from_fn_0010(b: &mut Bencher) {
572 do_bench_from_fn(b, 10)
576 fn bench_from_fn_0100(b: &mut Bencher) {
577 do_bench_from_fn(b, 100)
581 fn bench_from_fn_1000(b: &mut Bencher) {
582 do_bench_from_fn(b, 1000)
585 fn do_bench_from_elem(b: &mut Bencher, src_len: usize) {
586 b.bytes = src_len as u64;
589 let dst: Vec<usize> = repeat(5).take(src_len).collect();
590 assert_eq!(dst.len(), src_len);
591 assert!(dst.iter().all(|x| *x == 5));
596 fn bench_from_elem_0000(b: &mut Bencher) {
597 do_bench_from_elem(b, 0)
601 fn bench_from_elem_0010(b: &mut Bencher) {
602 do_bench_from_elem(b, 10)
606 fn bench_from_elem_0100(b: &mut Bencher) {
607 do_bench_from_elem(b, 100)
611 fn bench_from_elem_1000(b: &mut Bencher) {
612 do_bench_from_elem(b, 1000)
615 fn do_bench_from_slice(b: &mut Bencher, src_len: usize) {
616 let src: Vec<_> = FromIterator::from_iter(0..src_len);
618 b.bytes = src_len as u64;
621 let dst = src.clone()[..].to_vec();
622 assert_eq!(dst.len(), src_len);
623 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
628 fn bench_from_slice_0000(b: &mut Bencher) {
629 do_bench_from_slice(b, 0)
633 fn bench_from_slice_0010(b: &mut Bencher) {
634 do_bench_from_slice(b, 10)
638 fn bench_from_slice_0100(b: &mut Bencher) {
639 do_bench_from_slice(b, 100)
643 fn bench_from_slice_1000(b: &mut Bencher) {
644 do_bench_from_slice(b, 1000)
647 fn do_bench_from_iter(b: &mut Bencher, src_len: usize) {
648 let src: Vec<_> = FromIterator::from_iter(0..src_len);
650 b.bytes = src_len as u64;
653 let dst: Vec<_> = FromIterator::from_iter(src.clone().into_iter());
654 assert_eq!(dst.len(), src_len);
655 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
660 fn bench_from_iter_0000(b: &mut Bencher) {
661 do_bench_from_iter(b, 0)
665 fn bench_from_iter_0010(b: &mut Bencher) {
666 do_bench_from_iter(b, 10)
670 fn bench_from_iter_0100(b: &mut Bencher) {
671 do_bench_from_iter(b, 100)
675 fn bench_from_iter_1000(b: &mut Bencher) {
676 do_bench_from_iter(b, 1000)
679 fn do_bench_extend(b: &mut Bencher, dst_len: usize, src_len: usize) {
680 let dst: Vec<_> = FromIterator::from_iter(0..dst_len);
681 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
683 b.bytes = src_len as u64;
686 let mut dst = dst.clone();
687 dst.extend(src.clone().into_iter());
688 assert_eq!(dst.len(), dst_len + src_len);
689 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
694 fn bench_extend_0000_0000(b: &mut Bencher) {
695 do_bench_extend(b, 0, 0)
699 fn bench_extend_0000_0010(b: &mut Bencher) {
700 do_bench_extend(b, 0, 10)
704 fn bench_extend_0000_0100(b: &mut Bencher) {
705 do_bench_extend(b, 0, 100)
709 fn bench_extend_0000_1000(b: &mut Bencher) {
710 do_bench_extend(b, 0, 1000)
714 fn bench_extend_0010_0010(b: &mut Bencher) {
715 do_bench_extend(b, 10, 10)
719 fn bench_extend_0100_0100(b: &mut Bencher) {
720 do_bench_extend(b, 100, 100)
724 fn bench_extend_1000_1000(b: &mut Bencher) {
725 do_bench_extend(b, 1000, 1000)
728 fn do_bench_push_all(b: &mut Bencher, dst_len: usize, src_len: usize) {
729 let dst: Vec<_> = FromIterator::from_iter(0..dst_len);
730 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
732 b.bytes = src_len as u64;
735 let mut dst = dst.clone();
737 assert_eq!(dst.len(), dst_len + src_len);
738 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
743 fn bench_push_all_0000_0000(b: &mut Bencher) {
744 do_bench_push_all(b, 0, 0)
748 fn bench_push_all_0000_0010(b: &mut Bencher) {
749 do_bench_push_all(b, 0, 10)
753 fn bench_push_all_0000_0100(b: &mut Bencher) {
754 do_bench_push_all(b, 0, 100)
758 fn bench_push_all_0000_1000(b: &mut Bencher) {
759 do_bench_push_all(b, 0, 1000)
763 fn bench_push_all_0010_0010(b: &mut Bencher) {
764 do_bench_push_all(b, 10, 10)
768 fn bench_push_all_0100_0100(b: &mut Bencher) {
769 do_bench_push_all(b, 100, 100)
773 fn bench_push_all_1000_1000(b: &mut Bencher) {
774 do_bench_push_all(b, 1000, 1000)
777 fn do_bench_push_all_move(b: &mut Bencher, dst_len: usize, src_len: usize) {
778 let dst: Vec<_> = FromIterator::from_iter(0..dst_len);
779 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
781 b.bytes = src_len as u64;
784 let mut dst = dst.clone();
785 dst.extend(src.clone().into_iter());
786 assert_eq!(dst.len(), dst_len + src_len);
787 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
792 fn bench_push_all_move_0000_0000(b: &mut Bencher) {
793 do_bench_push_all_move(b, 0, 0)
797 fn bench_push_all_move_0000_0010(b: &mut Bencher) {
798 do_bench_push_all_move(b, 0, 10)
802 fn bench_push_all_move_0000_0100(b: &mut Bencher) {
803 do_bench_push_all_move(b, 0, 100)
807 fn bench_push_all_move_0000_1000(b: &mut Bencher) {
808 do_bench_push_all_move(b, 0, 1000)
812 fn bench_push_all_move_0010_0010(b: &mut Bencher) {
813 do_bench_push_all_move(b, 10, 10)
817 fn bench_push_all_move_0100_0100(b: &mut Bencher) {
818 do_bench_push_all_move(b, 100, 100)
822 fn bench_push_all_move_1000_1000(b: &mut Bencher) {
823 do_bench_push_all_move(b, 1000, 1000)
826 fn do_bench_clone(b: &mut Bencher, src_len: usize) {
827 let src: Vec<usize> = FromIterator::from_iter(0..src_len);
829 b.bytes = src_len as u64;
832 let dst = src.clone();
833 assert_eq!(dst.len(), src_len);
834 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
839 fn bench_clone_0000(b: &mut Bencher) {
844 fn bench_clone_0010(b: &mut Bencher) {
845 do_bench_clone(b, 10)
849 fn bench_clone_0100(b: &mut Bencher) {
850 do_bench_clone(b, 100)
854 fn bench_clone_1000(b: &mut Bencher) {
855 do_bench_clone(b, 1000)
858 fn do_bench_clone_from(b: &mut Bencher, times: usize, dst_len: usize, src_len: usize) {
859 let dst: Vec<_> = FromIterator::from_iter(0..src_len);
860 let src: Vec<_> = FromIterator::from_iter(dst_len..dst_len + src_len);
862 b.bytes = (times * src_len) as u64;
865 let mut dst = dst.clone();
868 dst.clone_from(&src);
870 assert_eq!(dst.len(), src_len);
871 assert!(dst.iter().enumerate().all(|(i, x)| dst_len + i == *x));
877 fn bench_clone_from_01_0000_0000(b: &mut Bencher) {
878 do_bench_clone_from(b, 1, 0, 0)
882 fn bench_clone_from_01_0000_0010(b: &mut Bencher) {
883 do_bench_clone_from(b, 1, 0, 10)
887 fn bench_clone_from_01_0000_0100(b: &mut Bencher) {
888 do_bench_clone_from(b, 1, 0, 100)
892 fn bench_clone_from_01_0000_1000(b: &mut Bencher) {
893 do_bench_clone_from(b, 1, 0, 1000)
897 fn bench_clone_from_01_0010_0010(b: &mut Bencher) {
898 do_bench_clone_from(b, 1, 10, 10)
902 fn bench_clone_from_01_0100_0100(b: &mut Bencher) {
903 do_bench_clone_from(b, 1, 100, 100)
907 fn bench_clone_from_01_1000_1000(b: &mut Bencher) {
908 do_bench_clone_from(b, 1, 1000, 1000)
912 fn bench_clone_from_01_0010_0100(b: &mut Bencher) {
913 do_bench_clone_from(b, 1, 10, 100)
917 fn bench_clone_from_01_0100_1000(b: &mut Bencher) {
918 do_bench_clone_from(b, 1, 100, 1000)
922 fn bench_clone_from_01_0010_0000(b: &mut Bencher) {
923 do_bench_clone_from(b, 1, 10, 0)
927 fn bench_clone_from_01_0100_0010(b: &mut Bencher) {
928 do_bench_clone_from(b, 1, 100, 10)
932 fn bench_clone_from_01_1000_0100(b: &mut Bencher) {
933 do_bench_clone_from(b, 1, 1000, 100)
937 fn bench_clone_from_10_0000_0000(b: &mut Bencher) {
938 do_bench_clone_from(b, 10, 0, 0)
942 fn bench_clone_from_10_0000_0010(b: &mut Bencher) {
943 do_bench_clone_from(b, 10, 0, 10)
947 fn bench_clone_from_10_0000_0100(b: &mut Bencher) {
948 do_bench_clone_from(b, 10, 0, 100)
952 fn bench_clone_from_10_0000_1000(b: &mut Bencher) {
953 do_bench_clone_from(b, 10, 0, 1000)
957 fn bench_clone_from_10_0010_0010(b: &mut Bencher) {
958 do_bench_clone_from(b, 10, 10, 10)
962 fn bench_clone_from_10_0100_0100(b: &mut Bencher) {
963 do_bench_clone_from(b, 10, 100, 100)
967 fn bench_clone_from_10_1000_1000(b: &mut Bencher) {
968 do_bench_clone_from(b, 10, 1000, 1000)
972 fn bench_clone_from_10_0010_0100(b: &mut Bencher) {
973 do_bench_clone_from(b, 10, 10, 100)
977 fn bench_clone_from_10_0100_1000(b: &mut Bencher) {
978 do_bench_clone_from(b, 10, 100, 1000)
982 fn bench_clone_from_10_0010_0000(b: &mut Bencher) {
983 do_bench_clone_from(b, 10, 10, 0)
987 fn bench_clone_from_10_0100_0010(b: &mut Bencher) {
988 do_bench_clone_from(b, 10, 100, 10)
992 fn bench_clone_from_10_1000_0100(b: &mut Bencher) {
993 do_bench_clone_from(b, 10, 1000, 100)