use super::*;
#[derive(Clone, Debug, Default)]
pub struct VecDeque<M> {
mutator: M,
}
pub fn vec_deque<M>(mutator: M) -> VecDeque<M> {
VecDeque { mutator }
}
impl<M, T> Mutate<alloc::collections::VecDeque<T>> for VecDeque<M>
where
M: Generate<T> + Mutate<T>,
{
#[inline]
fn mutation_count(
&self,
value: &alloc::collections::VecDeque<T>,
shrink: bool,
) -> core::option::Option<u32> {
let mut count = 0u32;
count += !shrink as u32;
count += !value.is_empty() as u32;
count += (value.len() >= 2) as u32;
for x in value.iter() {
count += self.mutator.mutation_count(x, shrink)?;
}
Some(count)
}
#[inline]
fn mutate(
&mut self,
c: &mut Candidates,
value: &mut alloc::collections::VecDeque<T>,
) -> Result<()> {
if !c.shrink() {
c.mutation(|ctx| {
let index = ctx.rng().gen_index(value.len() + 1).unwrap();
let elem = self.mutator.generate(ctx)?;
value.insert(index, elem);
Ok(())
})?;
}
if !value.is_empty() {
c.mutation(|ctx| {
let index = ctx.rng().gen_index(value.len()).unwrap();
value.remove(index);
Ok(())
})?;
}
if value.len() >= 2 {
c.mutation(|ctx| {
let i = ctx.rng().gen_index(value.len()).unwrap();
let j = ctx.rng().gen_index(value.len()).unwrap();
value.swap(i, j);
Ok(())
})?;
}
for x in value.iter_mut() {
self.mutator.mutate(c, x)?;
}
Ok(())
}
}
impl<M, T> Generate<alloc::collections::VecDeque<T>> for VecDeque<M>
where
M: Generate<T>,
{
#[inline]
fn generate(&mut self, ctx: &mut Context) -> Result<alloc::collections::VecDeque<T>> {
self.generate_via_mutate(ctx, 1)
}
}
impl<T> DefaultMutate for alloc::collections::VecDeque<T>
where
T: DefaultMutate,
T::DefaultMutate: Generate<T>,
{
type DefaultMutate = VecDeque<T::DefaultMutate>;
}