use orx_iterable::*;
fn test_it(values: Vec<usize>, col: impl Iterable<Item = usize>) {
let sum = values.iter().sum::<usize>();
let count = values.len();
assert_eq!(col.iter().count(), count);
assert_eq!(col.iter().sum::<usize>(), sum);
}
#[test]
fn flat_mapped() {
test_it(
vec![1, 4, 7, 11, 3, 8],
vec![vec![0, 3], vec![6], vec![10, 2, 7]].flat_mapped(|x| x.iter().map(|x| x + 1)),
);
let data = vec![vec![1], vec![333], vec![4, 2], vec![8, 8, 3], vec![1000]];
let indices = vec![0, 2, 3];
assert_eq!(
indices
.flat_mapped(|idx| &data[*idx])
.iter()
.copied()
.collect::<Vec<_>>(),
vec![1, 4, 2, 8, 8, 3]
);
assert_eq!(
indices
.flat_mapped(|idx| &data[*idx])
.copied()
.iter()
.collect::<Vec<_>>(),
vec![1, 4, 2, 8, 8, 3]
);
}
#[cfg(feature = "std")]
mod object_safe {
use orx_iterable::{obj_safe::*, *};
fn obj_test_it(values: Vec<usize>, col: &dyn IterableObj<Item = usize>) {
let sum = values.iter().sum::<usize>();
let count = values.len();
assert_eq!(col.boxed_iter().count(), count);
assert_eq!(col.boxed_iter().sum::<usize>(), sum);
}
#[test]
fn obj_flat_mapped() {
obj_test_it(
vec![1, 4, 7, 11, 3, 8],
&vec![vec![0, 3], vec![6], vec![10, 2, 7]].flat_mapped(|x| x.iter().map(|x| x + 1)),
);
let data = vec![vec![1], vec![333], vec![4, 2], vec![8, 8, 3], vec![1000]];
let indices = vec![0, 2, 3];
assert_eq!(
indices
.flat_mapped(|idx| &data[*idx])
.boxed_iter()
.copied()
.collect::<Vec<_>>(),
vec![1, 4, 2, 8, 8, 3]
);
assert_eq!(
indices
.flat_mapped(|idx| &data[*idx])
.copied()
.boxed_iter()
.collect::<Vec<_>>(),
vec![1, 4, 2, 8, 8, 3]
);
}
}