Implement Distribution<usize> for RangeTo(Inclusive)<usize>

Author: Ralith

src/distributions/uniform.rs

@@ -111,7 +111,7 @@
 use std::time::Duration;
 #[cfg(not(feature = "std"))]
 use core::time::Duration;
-use core::ops::{Range, RangeInclusive};
+use core::ops::{Range, RangeInclusive, RangeTo, RangeToInclusive};
 
 use crate::Rng;
 use crate::distributions::Distribution;
@@ -306,6 +306,18 @@ impl<T: SampleUniform> Distribution<T> for RangeInclusive<T> {
     }
 }
 
+impl Distribution<usize> for RangeTo<usize> {
+    fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> usize {
+        <usize as SampleUniform>::Sampler::sample_single(0, self.end, rng)
+    }
+}
+
+impl Distribution<usize> for RangeToInclusive<usize> {
+    fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> usize {
+        <usize as SampleUniform>::Sampler::sample_single_inclusive(0, self.end, rng)
+    }
+}
+
 /// Helper trait similar to [`Borrow`] but implemented
 /// only for SampleUniform and references to SampleUniform in
 /// order to resolve ambiguity issues.