use crate::utils_test::test_utils;
use crate::{BaseCausaloidVec, IdentificationValue};
const SMALL: usize = 10;
const MEDIUM: usize = 1_000;
const LARGE: usize = 10_000;
pub fn get_small_collection_and_data() -> (BaseCausaloidVec<f64, bool>, [f64; SMALL]) {
(
build_causaloid_collection(SMALL),
test_utils::generate_sample_data(),
)
}
pub fn get_medium_collection_and_data() -> (BaseCausaloidVec<f64, bool>, [f64; MEDIUM]) {
(
build_causaloid_collection(MEDIUM),
test_utils::generate_sample_data(),
)
}
pub fn get_large_collection_and_data() -> (BaseCausaloidVec<f64, bool>, [f64; LARGE]) {
(
build_causaloid_collection(LARGE),
test_utils::generate_sample_data(),
)
}
fn build_causaloid_collection(k: usize) -> BaseCausaloidVec<f64, bool> {
let mut v = Vec::with_capacity(k);
for i in 0..k {
v.push(test_utils::get_test_causaloid(i as IdentificationValue));
}
v
}