zip.rs - mozsearch

Enable keyboard shortcuts

use itertools::Itertools;

use itertools::EitherOrBoth::{Both, Left, Right};

use itertools::free::zip_eq;

use itertools::multizip;

#[test]

fn zip_longest_fused() {

    let a = [Some(1), None, Some(3), Some(4)];

    let b = [1, 2, 3];

    let unfused = a.iter().batching(|it| *it.next().unwrap())

        .zip_longest(b.iter().cloned());

    itertools::assert_equal(unfused,

                       vec![Both(1, 1), Right(2), Right(3)]);

#[test]

fn test_zip_longest_size_hint() {

    let c = (1..10).cycle();

    let v: &[_] = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9];

    let v2 = &[10, 11, 12];

    assert_eq!(c.zip_longest(v.iter()).size_hint(), (std::usize::MAX, None));

    assert_eq!(v.iter().zip_longest(v2.iter()).size_hint(), (10, Some(10)));

#[test]

fn test_double_ended_zip_longest() {

    let xs = [1, 2, 3, 4, 5, 6];

    let ys = [1, 2, 3, 7];

    let a = xs.iter().copied();

    let b = ys.iter().copied();

    let mut it = a.zip_longest(b);

    assert_eq!(it.next(), Some(Both(1, 1)));

    assert_eq!(it.next(), Some(Both(2, 2)));

    assert_eq!(it.next_back(), Some(Left(6)));

    assert_eq!(it.next_back(), Some(Left(5)));

    assert_eq!(it.next_back(), Some(Both(4, 7)));

    assert_eq!(it.next(), Some(Both(3, 3)));

    assert_eq!(it.next(), None);

#[test]

fn test_double_ended_zip() {

    let xs = [1, 2, 3, 4, 5, 6];

    let ys = [1, 2, 3, 7];

    let a = xs.iter().copied();

    let b = ys.iter().copied();

    let mut it = multizip((a, b));

    assert_eq!(it.next_back(), Some((4, 7)));

    assert_eq!(it.next_back(), Some((3, 3)));

    assert_eq!(it.next_back(), Some((2, 2)));

    assert_eq!(it.next_back(), Some((1, 1)));

    assert_eq!(it.next_back(), None);

#[should_panic]

#[test]

fn zip_eq_panic1()

    let a = [1, 2];

    let b = [1, 2, 3];

    zip_eq(&a, &b).count();

#[should_panic]

#[test]

fn zip_eq_panic2()

    let a: [i32; 0] = [];

    let b = [1, 2, 3];

    zip_eq(&a, &b).count();