use super::{Generator, TestCase};
pub fn unit() -> JustGenerator<()> {
just(())
}
pub struct JustGenerator<T> {
value: T,
}
impl<T: Clone + Send + Sync> Generator<T> for JustGenerator<T> {
fn do_draw(&self, _tc: &TestCase) -> T {
self.value.clone()
}
}
pub fn just<T: Clone + Send + Sync>(value: T) -> JustGenerator<T> {
JustGenerator { value }
}
pub struct BoolGenerator {
p: f64,
}
impl Generator<bool> for BoolGenerator {
fn do_draw(&self, tc: &TestCase) -> bool {
tc.generate_boolean(self.p)
}
}
pub fn booleans() -> BoolGenerator {
BoolGenerator { p: 0.5 }
}
pub fn weighted_booleans(p: f64) -> BoolGenerator {
BoolGenerator { p }
}