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::ascii::AsciiExt;
13 use std::mem::size_of;
15 use std::vec::{Drain, IntoIter};
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();
101 v.extend(w.clone()); // specializes to `append`
102 assert!(v.iter().eq(w.iter().chain(w.iter())));
106 fn test_extend_ref() {
107 let mut v = vec![1, 2];
108 v.extend(&[3, 4, 5]);
110 assert_eq!(v.len(), 5);
111 assert_eq!(v, [1, 2, 3, 4, 5]);
116 assert_eq!(v.len(), 7);
117 assert_eq!(v, [1, 2, 3, 4, 5, 6, 7]);
121 fn test_slice_from_mut() {
122 let mut values = vec![1, 2, 3, 4, 5];
124 let slice = &mut values[2..];
125 assert!(slice == [3, 4, 5]);
131 assert!(values == [1, 2, 5, 6, 7]);
135 fn test_slice_to_mut() {
136 let mut values = vec![1, 2, 3, 4, 5];
138 let slice = &mut values[..2];
139 assert!(slice == [1, 2]);
145 assert!(values == [2, 3, 3, 4, 5]);
149 fn test_split_at_mut() {
150 let mut values = vec![1, 2, 3, 4, 5];
152 let (left, right) = values.split_at_mut(2);
154 let left: &[_] = left;
155 assert!(&left[..left.len()] == &[1, 2]);
162 let right: &[_] = right;
163 assert!(&right[..right.len()] == &[3, 4, 5]);
170 assert_eq!(values, [2, 3, 5, 6, 7]);
175 let v: Vec<i32> = vec![];
176 let w = vec![1, 2, 3];
178 assert_eq!(v, v.clone());
182 // they should be disjoint in memory.
183 assert!(w.as_ptr() != z.as_ptr())
187 fn test_clone_from() {
189 let three: Vec<Box<_>> = vec![box 1, box 2, box 3];
190 let two: Vec<Box<_>> = vec![box 4, box 5];
192 v.clone_from(&three);
193 assert_eq!(v, three);
196 v.clone_from(&three);
197 assert_eq!(v, three);
204 v.clone_from(&three);
210 let mut vec = vec![1, 2, 3, 4];
211 vec.retain(|&x| x % 2 == 0);
212 assert_eq!(vec, [2, 4]);
217 fn case(a: Vec<i32>, b: Vec<i32>) {
222 case(vec![], vec![]);
223 case(vec![1], vec![1]);
224 case(vec![1, 1], vec![1]);
225 case(vec![1, 2, 3], vec![1, 2, 3]);
226 case(vec![1, 1, 2, 3], vec![1, 2, 3]);
227 case(vec![1, 2, 2, 3], vec![1, 2, 3]);
228 case(vec![1, 2, 3, 3], vec![1, 2, 3]);
229 case(vec![1, 1, 2, 2, 2, 3, 3], vec![1, 2, 3]);
233 fn test_dedup_by_key() {
234 fn case(a: Vec<i32>, b: Vec<i32>) {
236 v.dedup_by_key(|i| *i / 10);
239 case(vec![], vec![]);
240 case(vec![10], vec![10]);
241 case(vec![10, 11], vec![10]);
242 case(vec![10, 20, 30], vec![10, 20, 30]);
243 case(vec![10, 11, 20, 30], vec![10, 20, 30]);
244 case(vec![10, 20, 21, 30], vec![10, 20, 30]);
245 case(vec![10, 20, 30, 31], vec![10, 20, 30]);
246 case(vec![10, 11, 20, 21, 22, 30, 31], vec![10, 20, 30]);
251 let mut vec = vec!["foo", "bar", "Bar", "baz", "bar"];
252 vec.dedup_by(|a, b| a.eq_ignore_ascii_case(b));
254 assert_eq!(vec, ["foo", "bar", "baz", "bar"]);
258 fn test_dedup_unique() {
259 let mut v0: Vec<Box<_>> = vec![box 1, box 1, box 2, box 3];
261 let mut v1: Vec<Box<_>> = vec![box 1, box 2, box 2, box 3];
263 let mut v2: Vec<Box<_>> = vec![box 1, box 2, box 3, box 3];
265 // If the boxed pointers were leaked or otherwise misused, valgrind
266 // and/or rt should raise errors.
270 fn zero_sized_values() {
271 let mut v = Vec::new();
272 assert_eq!(v.len(), 0);
274 assert_eq!(v.len(), 1);
276 assert_eq!(v.len(), 2);
277 assert_eq!(v.pop(), Some(()));
278 assert_eq!(v.pop(), Some(()));
279 assert_eq!(v.pop(), None);
281 assert_eq!(v.iter().count(), 0);
283 assert_eq!(v.iter().count(), 1);
285 assert_eq!(v.iter().count(), 2);
289 assert_eq!(v.iter_mut().count(), 2);
291 assert_eq!(v.iter_mut().count(), 3);
293 assert_eq!(v.iter_mut().count(), 4);
295 for &mut () in &mut v {}
299 assert_eq!(v.iter_mut().count(), 0);
303 fn test_partition() {
304 assert_eq!(vec![].into_iter().partition(|x: &i32| *x < 3),
306 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 4),
307 (vec![1, 2, 3], vec![]));
308 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 2),
309 (vec![1], vec![2, 3]));
310 assert_eq!(vec![1, 2, 3].into_iter().partition(|x| *x < 0),
311 (vec![], vec![1, 2, 3]));
315 fn test_zip_unzip() {
316 let z1 = vec![(1, 4), (2, 5), (3, 6)];
318 let (left, right): (Vec<_>, Vec<_>) = z1.iter().cloned().unzip();
320 assert_eq!((1, 4), (left[0], right[0]));
321 assert_eq!((2, 5), (left[1], right[1]));
322 assert_eq!((3, 6), (left[2], right[2]));
326 fn test_vec_truncate_drop() {
327 static mut DROPS: u32 = 0;
337 let mut v = vec![Elem(1), Elem(2), Elem(3), Elem(4), Elem(5)];
338 assert_eq!(unsafe { DROPS }, 0);
340 assert_eq!(unsafe { DROPS }, 2);
342 assert_eq!(unsafe { DROPS }, 5);
347 fn test_vec_truncate_fail() {
349 impl Drop for BadElem {
351 let BadElem(ref mut x) = *self;
353 panic!("BadElem panic: 0xbadbeef")
358 let mut v = vec![BadElem(1), BadElem(2), BadElem(0xbadbeef), BadElem(4)];
364 let vec = vec![1, 2, 3];
365 assert!(vec[1] == 2);
370 fn test_index_out_of_bounds() {
371 let vec = vec![1, 2, 3];
377 fn test_slice_out_of_bounds_1() {
378 let x = vec![1, 2, 3, 4, 5];
384 fn test_slice_out_of_bounds_2() {
385 let x = vec![1, 2, 3, 4, 5];
391 fn test_slice_out_of_bounds_3() {
392 let x = vec![1, 2, 3, 4, 5];
398 fn test_slice_out_of_bounds_4() {
399 let x = vec![1, 2, 3, 4, 5];
405 fn test_slice_out_of_bounds_5() {
406 let x = vec![1, 2, 3, 4, 5];
412 fn test_swap_remove_empty() {
413 let mut vec = Vec::<i32>::new();
418 fn test_move_items() {
419 let vec = vec![1, 2, 3];
420 let mut vec2 = vec![];
424 assert_eq!(vec2, [1, 2, 3]);
428 fn test_move_items_reverse() {
429 let vec = vec![1, 2, 3];
430 let mut vec2 = vec![];
431 for i in vec.into_iter().rev() {
434 assert_eq!(vec2, [3, 2, 1]);
438 fn test_move_items_zero_sized() {
439 let vec = vec![(), (), ()];
440 let mut vec2 = vec![];
444 assert_eq!(vec2, [(), (), ()]);
448 fn test_drain_items() {
449 let mut vec = vec![1, 2, 3];
450 let mut vec2 = vec![];
451 for i in vec.drain(..) {
455 assert_eq!(vec2, [1, 2, 3]);
459 fn test_drain_items_reverse() {
460 let mut vec = vec![1, 2, 3];
461 let mut vec2 = vec![];
462 for i in vec.drain(..).rev() {
466 assert_eq!(vec2, [3, 2, 1]);
470 fn test_drain_items_zero_sized() {
471 let mut vec = vec![(), (), ()];
472 let mut vec2 = vec![];
473 for i in vec.drain(..) {
477 assert_eq!(vec2, [(), (), ()]);
482 fn test_drain_out_of_bounds() {
483 let mut v = vec![1, 2, 3, 4, 5];
488 fn test_drain_range() {
489 let mut v = vec![1, 2, 3, 4, 5];
490 for _ in v.drain(4..) {
492 assert_eq!(v, &[1, 2, 3, 4]);
494 let mut v: Vec<_> = (1..6).map(|x| x.to_string()).collect();
495 for _ in v.drain(1..4) {
497 assert_eq!(v, &[1.to_string(), 5.to_string()]);
499 let mut v: Vec<_> = (1..6).map(|x| x.to_string()).collect();
500 for _ in v.drain(1..4).rev() {
502 assert_eq!(v, &[1.to_string(), 5.to_string()]);
504 let mut v: Vec<_> = vec![(); 5];
505 for _ in v.drain(1..4).rev() {
507 assert_eq!(v, &[(), ()]);
511 fn test_into_boxed_slice() {
512 let xs = vec![1, 2, 3];
513 let ys = xs.into_boxed_slice();
514 assert_eq!(&*ys, [1, 2, 3]);
519 let mut vec = vec![1, 2, 3];
520 let mut vec2 = vec![4, 5, 6];
521 vec.append(&mut vec2);
522 assert_eq!(vec, [1, 2, 3, 4, 5, 6]);
523 assert_eq!(vec2, []);
527 fn test_split_off() {
528 let mut vec = vec![1, 2, 3, 4, 5, 6];
529 let vec2 = vec.split_off(4);
530 assert_eq!(vec, [1, 2, 3, 4]);
531 assert_eq!(vec2, [5, 6]);
535 fn test_into_iter_as_slice() {
536 let vec = vec!['a', 'b', 'c'];
537 let mut into_iter = vec.into_iter();
538 assert_eq!(into_iter.as_slice(), &['a', 'b', 'c']);
539 let _ = into_iter.next().unwrap();
540 assert_eq!(into_iter.as_slice(), &['b', 'c']);
541 let _ = into_iter.next().unwrap();
542 let _ = into_iter.next().unwrap();
543 assert_eq!(into_iter.as_slice(), &[]);
547 fn test_into_iter_as_mut_slice() {
548 let vec = vec!['a', 'b', 'c'];
549 let mut into_iter = vec.into_iter();
550 assert_eq!(into_iter.as_slice(), &['a', 'b', 'c']);
551 into_iter.as_mut_slice()[0] = 'x';
552 into_iter.as_mut_slice()[1] = 'y';
553 assert_eq!(into_iter.next().unwrap(), 'x');
554 assert_eq!(into_iter.as_slice(), &['y', 'c']);
558 fn test_into_iter_debug() {
559 let vec = vec!['a', 'b', 'c'];
560 let into_iter = vec.into_iter();
561 let debug = format!("{:?}", into_iter);
562 assert_eq!(debug, "IntoIter(['a', 'b', 'c'])");
566 fn test_into_iter_count() {
567 assert_eq!(vec![1, 2, 3].into_iter().count(), 3);
571 fn test_into_iter_clone() {
572 fn iter_equal<I: Iterator<Item = i32>>(it: I, slice: &[i32]) {
573 let v: Vec<i32> = it.collect();
574 assert_eq!(&v[..], slice);
576 let mut it = vec![1, 2, 3].into_iter();
577 iter_equal(it.clone(), &[1, 2, 3]);
578 assert_eq!(it.next(), Some(1));
579 let mut it = it.rev();
580 iter_equal(it.clone(), &[3, 2]);
581 assert_eq!(it.next(), Some(3));
582 iter_equal(it.clone(), &[2]);
583 assert_eq!(it.next(), Some(2));
584 iter_equal(it.clone(), &[]);
585 assert_eq!(it.next(), None);
590 let borrowed: &[_] = &["borrowed", "(slice)"];
591 let owned = vec!["owned", "(vec)"];
592 match (Cow::from(owned.clone()), Cow::from(borrowed)) {
593 (Cow::Owned(o), Cow::Borrowed(b)) => assert!(o == owned && b == borrowed),
594 _ => panic!("invalid `Cow::from`"),
600 let borrowed: &[_] = &["borrowed", "(slice)"];
601 let owned = vec!["owned", "(vec)"];
602 assert_eq!(Vec::from(Cow::Borrowed(borrowed)), vec!["borrowed", "(slice)"]);
603 assert_eq!(Vec::from(Cow::Owned(owned)), vec!["owned", "(vec)"]);
607 fn assert_covariance() {
608 fn drain<'new>(d: Drain<'static, &'static str>) -> Drain<'new, &'new str> {
611 fn into_iter<'new>(i: IntoIter<&'static str>) -> IntoIter<&'new str> {
617 fn test_placement() {
618 let mut vec = vec![1];
619 assert_eq!(vec.place_back() <- 2, &2);
620 assert_eq!(vec.len(), 2);
621 assert_eq!(vec.place_back() <- 3, &3);
622 assert_eq!(vec.len(), 3);
623 assert_eq!(&vec, &[1, 2, 3]);
627 fn test_placement_panic() {
628 let mut vec = vec![1, 2, 3];
629 fn mkpanic() -> usize { panic!() }
630 let _ = panic::catch_unwind(panic::AssertUnwindSafe(|| { vec.place_back() <- mkpanic(); }));
631 assert_eq!(vec.len(), 3);