You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

388 lines
10 KiB

#![cfg(feature = "alloc")]
#![allow(bad_style)]
#![allow(clippy::redundant_clone)]
#[cfg(feature = "serde")]
use serde_test::{assert_tokens, Token};
use std::iter::FromIterator;
use tinyvec::*;
#[test]
fn TinyVec_swap_remove() {
let mut tv: TinyVec<[i32; 10]> = Default::default();
tv.push(1);
tv.push(2);
tv.push(3);
tv.push(4);
assert_eq!(tv.swap_remove(3), 4);
assert_eq!(&tv[..], &[1, 2, 3][..]);
assert_eq!(tv.swap_remove(0), 1);
assert_eq!(&tv[..], &[3, 2][..]);
assert_eq!(tv.swap_remove(0), 3);
assert_eq!(&tv[..], &[2][..]);
assert_eq!(tv.swap_remove(0), 2);
assert_eq!(&tv[..], &[][..]);
}
#[test]
fn TinyVec_capacity() {
let mut tv: TinyVec<[i32; 1]> = Default::default();
assert_eq!(tv.capacity(), 1);
tv.move_to_the_heap();
tv.extend_from_slice(&[1, 2, 3, 4]);
assert_eq!(tv.capacity(), 4);
}
#[test]
fn TinyVec_drain() {
let mut tv: TinyVec<[i32; 10]> = Default::default();
tv.push(1);
tv.push(2);
tv.push(3);
assert_eq!(Vec::from_iter(tv.clone().drain(..)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().drain(..2)), vec![1, 2]);
assert_eq!(Vec::from_iter(tv.clone().drain(..3)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().drain(..=1)), vec![1, 2]);
assert_eq!(Vec::from_iter(tv.clone().drain(..=2)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().drain(0..)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().drain(1..)), vec![2, 3]);
assert_eq!(Vec::from_iter(tv.clone().drain(0..2)), vec![1, 2]);
assert_eq!(Vec::from_iter(tv.clone().drain(0..3)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().drain(1..2)), vec![2]);
assert_eq!(Vec::from_iter(tv.clone().drain(1..3)), vec![2, 3]);
assert_eq!(Vec::from_iter(tv.clone().drain(0..=1)), vec![1, 2]);
assert_eq!(Vec::from_iter(tv.clone().drain(0..=2)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().drain(1..=1)), vec![2]);
assert_eq!(Vec::from_iter(tv.clone().drain(1..=2)), vec![2, 3]);
}
#[test]
fn TinyVec_splice() {
let mut tv: TinyVec<[i32; 10]> = Default::default();
tv.push(1);
tv.push(2);
tv.push(3);
// splice returns the same things as drain
assert_eq!(Vec::from_iter(tv.clone().splice(.., None)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().splice(..2, None)), vec![1, 2]);
assert_eq!(Vec::from_iter(tv.clone().splice(..3, None)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().splice(..=1, None)), vec![1, 2]);
assert_eq!(Vec::from_iter(tv.clone().splice(..=2, None)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().splice(0.., None)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().splice(1.., None)), vec![2, 3]);
assert_eq!(Vec::from_iter(tv.clone().splice(0..2, None)), vec![1, 2]);
assert_eq!(Vec::from_iter(tv.clone().splice(0..3, None)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().splice(1..2, None)), vec![2]);
assert_eq!(Vec::from_iter(tv.clone().splice(1..3, None)), vec![2, 3]);
assert_eq!(Vec::from_iter(tv.clone().splice(0..=1, None)), vec![1, 2]);
assert_eq!(Vec::from_iter(tv.clone().splice(0..=2, None)), vec![1, 2, 3]);
assert_eq!(Vec::from_iter(tv.clone().splice(1..=1, None)), vec![2]);
assert_eq!(Vec::from_iter(tv.clone().splice(1..=2, None)), vec![2, 3]);
// splice removes the same things as drain
let mut tv2 = tv.clone();
tv2.splice(.., None);
assert_eq!(tv2, tiny_vec![]);
let mut tv2 = tv.clone();
tv2.splice(..2, None);
assert_eq!(tv2, tiny_vec![3]);
let mut tv2 = tv.clone();
tv2.splice(..3, None);
assert_eq!(tv2, tiny_vec![]);
let mut tv2 = tv.clone();
tv2.splice(..=1, None);
assert_eq!(tv2, tiny_vec![3]);
let mut tv2 = tv.clone();
tv2.splice(..=2, None);
assert_eq!(tv2, tiny_vec![]);
let mut tv2 = tv.clone();
tv2.splice(0.., None);
assert_eq!(tv2, tiny_vec![]);
let mut tv2 = tv.clone();
tv2.splice(1.., None);
assert_eq!(tv2, tiny_vec![1]);
let mut tv2 = tv.clone();
tv2.splice(0..2, None);
assert_eq!(tv2, tiny_vec![3]);
let mut tv2 = tv.clone();
tv2.splice(0..3, None);
assert_eq!(tv2, tiny_vec![]);
let mut tv2 = tv.clone();
tv2.splice(1..2, None);
assert_eq!(tv2, tiny_vec![1, 3]);
let mut tv2 = tv.clone();
tv2.splice(1..3, None);
assert_eq!(tv2, tiny_vec![1]);
let mut tv2 = tv.clone();
tv2.splice(0..=1, None);
assert_eq!(tv2, tiny_vec![3]);
let mut tv2 = tv.clone();
tv2.splice(0..=2, None);
assert_eq!(tv2, tiny_vec![]);
let mut tv2 = tv.clone();
tv2.splice(1..=1, None);
assert_eq!(tv2, tiny_vec![1, 3]);
let mut tv2 = tv.clone();
tv2.splice(1..=2, None);
assert_eq!(tv2, tiny_vec![1]);
// splice adds the elements correctly
let mut tv2 = tv.clone();
tv2.splice(.., 4..=6);
assert_eq!(tv2, tiny_vec![4, 5, 6]);
let mut tv2 = tv.clone();
tv2.splice(..2, 4..=6);
assert_eq!(tv2, tiny_vec![4, 5, 6, 3]);
let mut tv2 = tv.clone();
tv2.splice(..3, 4..=6);
assert_eq!(tv2, tiny_vec![4, 5, 6]);
let mut tv2 = tv.clone();
tv2.splice(..=1, 4..=6);
assert_eq!(tv2, tiny_vec![4, 5, 6, 3]);
let mut tv2 = tv.clone();
tv2.splice(..=2, 4..=6);
assert_eq!(tv2, tiny_vec![4, 5, 6]);
let mut tv2 = tv.clone();
tv2.splice(0.., 4..=6);
assert_eq!(tv2, tiny_vec![4, 5, 6]);
let mut tv2 = tv.clone();
tv2.splice(1.., 4..=6);
assert_eq!(tv2, tiny_vec![1, 4, 5, 6]);
let mut tv2 = tv.clone();
tv2.splice(0..2, 4..=6);
assert_eq!(tv2, tiny_vec![4, 5, 6, 3]);
let mut tv2 = tv.clone();
tv2.splice(0..3, 4..=6);
assert_eq!(tv2, tiny_vec![4, 5, 6]);
let mut tv2 = tv.clone();
tv2.splice(1..2, 4..=6);
assert_eq!(tv2, tiny_vec![1, 4, 5, 6, 3]);
let mut tv2 = tv.clone();
tv2.splice(1..3, 4..=6);
assert_eq!(tv2, tiny_vec![1, 4, 5, 6]);
let mut tv2 = tv.clone();
tv2.splice(0..=1, 4..=6);
assert_eq!(tv2, tiny_vec![4, 5, 6, 3]);
let mut tv2 = tv.clone();
tv2.splice(0..=2, 4..=6);
assert_eq!(tv2, tiny_vec![4, 5, 6]);
let mut tv2 = tv.clone();
tv2.splice(1..=1, 4..=6);
assert_eq!(tv2, tiny_vec![1, 4, 5, 6, 3]);
let mut tv2 = tv.clone();
tv2.splice(1..=2, 4..=6);
assert_eq!(tv2, tiny_vec![1, 4, 5, 6]);
// splice adds the elements correctly when the replacement is smaller
let mut tv2 = tv.clone();
tv2.splice(.., Some(4));
assert_eq!(tv2, tiny_vec![4]);
let mut tv2 = tv.clone();
tv2.splice(..2, Some(4));
assert_eq!(tv2, tiny_vec![4, 3]);
let mut tv2 = tv.clone();
tv2.splice(1.., Some(4));
assert_eq!(tv2, tiny_vec![1, 4]);
let mut tv2 = tv.clone();
tv2.splice(1..=1, Some(4));
assert_eq!(tv2, tiny_vec![1, 4, 3]);
}
#[test]
fn TinyVec_resize() {
let mut tv: TinyVec<[i32; 10]> = Default::default();
tv.resize(20, 5);
assert_eq!(&tv[..], &[5; 20]);
}
#[test]
fn TinyVec_from_slice_impl() {
let bigger_slice: [u8; 11] = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
let tinyvec: TinyVec<[u8; 10]> = TinyVec::Heap((&bigger_slice[..]).into());
assert_eq!(TinyVec::from(&bigger_slice[..]), tinyvec);
let smaller_slice: [u8; 5] = [0, 1, 2, 3, 4];
let tinyvec: TinyVec<[u8; 10]> = TinyVec::Inline(ArrayVec::from_array_len(
[0, 1, 2, 3, 4, 0, 0, 0, 0, 0],
5,
));
assert_eq!(TinyVec::from(&smaller_slice[..]), tinyvec);
let same_size: [u8; 4] = [0, 1, 2, 3];
let tinyvec: TinyVec<[u8; 4]> =
TinyVec::Inline(ArrayVec::from_array_len(same_size, 4));
assert_eq!(TinyVec::from(&same_size[..]), tinyvec);
}
#[test]
fn TinyVec_from_array() {
let array = [9, 8, 7, 6, 5, 4, 3, 2, 1];
let tv = TinyVec::from(array);
assert_eq!(&array, &tv[..]);
}
#[test]
fn TinyVec_macro() {
let mut expected: TinyVec<[i32; 4]> = Default::default();
expected.push(1);
expected.push(2);
expected.push(3);
let actual = tiny_vec!(1, 2, 3);
assert_eq!(expected, actual);
assert_eq!(tiny_vec![0u8; 4], tiny_vec!(0u8, 0u8, 0u8, 0u8));
assert_eq!(tiny_vec![0u8; 4], tiny_vec!([u8; 4] => 0, 0, 0, 0));
assert_eq!(tiny_vec![0; 4], tiny_vec!(0, 0, 0, 0));
assert_eq!(tiny_vec![0; 4], tiny_vec!([u8; 4] => 0, 0, 0, 0));
let expected2 = tiny_vec![1.1; 3];
let actual2 = tiny_vec!([f32; 3] => 1.1, 1.1, 1.1);
assert_eq!(expected2, actual2);
}
#[test]
fn TinyVec_macro_non_copy() {
// must use a variable here to avoid macro shenanigans
let s = String::new();
let _: TinyVec<[String; 10]> = tiny_vec!([String; 10] => s);
}
#[test]
fn TinyVec_reserve() {
let mut tv: TinyVec<[i32; 4]> = Default::default();
assert_eq!(tv.capacity(), 4);
tv.extend_from_slice(&[1, 2]);
assert_eq!(tv.capacity(), 4);
tv.reserve(2);
assert_eq!(tv.capacity(), 4);
tv.reserve(4);
assert!(tv.capacity() >= 6);
tv.extend_from_slice(&[3, 4, 5, 6]);
tv.reserve(4);
assert!(tv.capacity() >= 10);
}
#[test]
fn TinyVec_reserve_exact() {
let mut tv: TinyVec<[i32; 4]> = Default::default();
assert_eq!(tv.capacity(), 4);
tv.extend_from_slice(&[1, 2]);
assert_eq!(tv.capacity(), 4);
tv.reserve_exact(2);
assert_eq!(tv.capacity(), 4);
tv.reserve_exact(4);
assert!(tv.capacity() >= 6);
tv.extend_from_slice(&[3, 4, 5, 6]);
tv.reserve_exact(4);
assert!(tv.capacity() >= 10);
}
#[test]
fn TinyVec_move_to_heap_and_shrink() {
let mut tv: TinyVec<[i32; 4]> = Default::default();
assert!(tv.is_inline());
tv.move_to_the_heap();
assert!(tv.is_heap());
assert_eq!(tv.capacity(), 0);
tv.push(1);
tv.shrink_to_fit();
assert!(tv.is_inline());
assert_eq!(tv.capacity(), 4);
tv.move_to_the_heap_and_reserve(3);
assert!(tv.is_heap());
assert_eq!(tv.capacity(), 4);
tv.extend(2..=4);
assert_eq!(tv.capacity(), 4);
assert_eq!(tv.as_slice(), [1, 2, 3, 4]);
}
#[cfg(feature = "serde")]
#[test]
fn TinyVec_ser_de_empty() {
let tv: TinyVec<[i32; 0]> = tiny_vec![];
assert_tokens(&tv, &[Token::Seq { len: Some(0) }, Token::SeqEnd]);
}
#[cfg(feature = "serde")]
#[test]
fn TinyVec_ser_de() {
let tv: TinyVec<[i32; 4]> = tiny_vec![1, 2, 3, 4];
assert_tokens(
&tv,
&[
Token::Seq { len: Some(4) },
Token::I32(1),
Token::I32(2),
Token::I32(3),
Token::I32(4),
Token::SeqEnd,
],
);
}
#[cfg(feature = "serde")]
#[test]
fn TinyVec_ser_de_heap() {
let mut tv: TinyVec<[i32; 4]> = tiny_vec![1, 2, 3, 4];
tv.move_to_the_heap();
assert_tokens(
&tv,
&[
Token::Seq { len: Some(4) },
Token::I32(1),
Token::I32(2),
Token::I32(3),
Token::I32(4),
Token::SeqEnd,
],
);
}