just/src/range_ext.rs
Casey Rodarmor 3a287b864a
Housekeeping (#394)
- Upgrade to rust 2018
- Update dependencies
- Use BTree{Map,Set} instead of Map and Set
2019-04-11 15:23:14 -07:00

57 lines
1.3 KiB
Rust

use crate::common::*;
pub trait RangeExt<T> {
fn range_contains(&self, i: &T) -> bool;
}
impl<T> RangeExt<T> for Range<T>
where
T: PartialOrd,
{
fn range_contains(&self, i: &T) -> bool {
i >= &self.start && i < &self.end
}
}
impl<T> RangeExt<T> for RangeInclusive<T>
where
T: PartialOrd,
{
fn range_contains(&self, i: &T) -> bool {
i >= self.start() && i <= self.end()
}
}
#[cfg(test)]
mod test {
use super::*;
#[test]
fn exclusive() {
assert!(!(0..0).range_contains(&0));
assert!(!(0..0).range_contains(&0));
assert!(!(1..10).range_contains(&0));
assert!(!(1..10).range_contains(&10));
assert!(!(1..10).range_contains(&0));
assert!(!(1..10).range_contains(&10));
assert!((0..1).range_contains(&0));
assert!((0..1).range_contains(&0));
assert!((10..20).range_contains(&15));
assert!((10..20).range_contains(&15));
}
#[test]
fn inclusive() {
assert!(!(0..=10).range_contains(&11));
assert!(!(1..=10).range_contains(&0));
assert!(!(5..=10).range_contains(&4));
assert!((0..=0).range_contains(&0));
assert!((0..=1).range_contains(&0));
assert!((0..=10).range_contains(&0));
assert!((0..=10).range_contains(&10));
assert!((0..=10).range_contains(&7));
assert!((1..=10).range_contains(&10));
assert!((10..=20).range_contains(&15));
}
}