diff options
Diffstat (limited to 'src/libstd')
| -rw-r--r-- | src/libstd/rand/distributions.rs | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/src/libstd/rand/distributions.rs b/src/libstd/rand/distributions.rs index 845d8bbc952..0ee073a926b 100644 --- a/src/libstd/rand/distributions.rs +++ b/src/libstd/rand/distributions.rs @@ -41,6 +41,20 @@ pub trait IndependentSample<Support>: Sample<Support> { fn ind_sample<R: Rng>(&self, &mut R) -> Support; } +/// A wrapper for generating types that implement `Rand` via the +/// `Sample` & `IndependentSample` traits. +pub struct RandSample<Sup>; + +impl<Sup: Rand> Sample<Sup> for RandSample<Sup> { + fn sample<R: Rng>(&mut self, rng: &mut R) -> Sup { self.ind_sample(rng) } +} + +impl<Sup: Rand> IndependentSample<Sup> for RandSample<Sup> { + fn ind_sample<R: Rng>(&self, rng: &mut R) -> Sup { + rng.gen() + } +} + mod ziggurat_tables; // inlining should mean there is no performance penalty for this @@ -166,3 +180,24 @@ impl Rand for Exp1 { pdf, zero_case)) } } + +#[cfg(test)] +mod tests { + use rand::*; + use super::*; + + struct ConstRand(uint); + impl Rand for ConstRand { + fn rand<R: Rng>(_: &mut R) -> ConstRand { + ConstRand(0) + } + } + + #[test] + fn test_rand_sample() { + let mut rand_sample = RandSample::<ConstRand>; + + assert_eq!(*rand_sample.sample(task_rng()), 0); + assert_eq!(*rand_sample.ind_sample(task_rng()), 0); + } +} |
