Merge pull request #79 from xfbs/proptest

Idea: use proptest additionally for unit testing
jeffparsons · Jan 31, 2024 · c50ed21 · c50ed21
2 parents c478b79 + 9564df0
commit c50ed21
Show file tree

Hide file tree

Showing 5 changed files with 102 additions and 2 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -30,6 +30,8 @@ rand = "0.8"
 chrono = "0.4" # For examples
 rustc_version = "0.4"
 serde_json = "1"
+proptest = "1.4.0"
+test-strategy = "0.3.1"
 
 [features]
 serde1 = ["serde"]

diff --git a/src/inclusive_map.rs b/src/inclusive_map.rs
@@ -840,7 +840,34 @@ where
 #[cfg(test)]
 mod tests {
     use super::*;
-    use alloc::{format, vec, vec::Vec};
+    use alloc::{format, vec, vec::Vec, string::String};
+    use test_strategy::proptest;
+    use alloc as std;
+
+    #[proptest]
+    fn test_arbitrary_map_u8(ranges: Vec<(RangeInclusive<u8>, String)>) {
+        let ranges: Vec<_> = ranges
+            .into_iter()
+            .filter(|(range, _value)| range.start() != range.end())
+            .collect();
+        let set = ranges
+            .iter()
+            .fold(RangeInclusiveMap::new(), |mut set, (range, value)| {
+                set.insert(range.clone(), value.clone());
+                set
+            });
+
+        for value in 0..u8::MAX {
+            assert_eq!(
+                set.get(&value),
+                ranges
+                    .iter()
+                    .rev()
+                    .find(|(range, _value)| range.contains(&value))
+                    .map(|(_range, value)| value)
+            );
+        }
+    }
 
     trait RangeInclusiveMapExt<K, V> {
         fn to_vec(&self) -> Vec<(RangeInclusive<K>, V)>;

diff --git a/src/inclusive_set.rs b/src/inclusive_set.rs
@@ -409,7 +409,30 @@ where
 #[cfg(test)]
 mod tests {
     use super::*;
+    use alloc as std;
     use alloc::{format, vec, vec::Vec};
+    use test_strategy::proptest;
+
+    #[proptest]
+    fn test_arbitrary_set_u8(ranges: Vec<RangeInclusive<u8>>) {
+        let ranges: Vec<_> = ranges
+            .into_iter()
+            .filter(|range| range.start() != range.end())
+            .collect();
+        let set = ranges
+            .iter()
+            .fold(RangeInclusiveSet::new(), |mut set, range| {
+                set.insert(range.clone());
+                set
+            });
+
+        for value in 0..u8::MAX {
+            assert_eq!(
+                set.contains(&value),
+                ranges.iter().any(|range| range.contains(&value))
+            );
+        }
+    }
 
     trait RangeInclusiveSetExt<T> {
         fn to_vec(&self) -> Vec<RangeInclusive<T>>;

diff --git a/src/map.rs b/src/map.rs
@@ -732,7 +732,34 @@ where
 #[cfg(test)]
 mod tests {
     use super::*;
-    use alloc::{format, vec, vec::Vec};
+    use alloc as std;
+    use alloc::{format, string::String, vec, vec::Vec};
+    use test_strategy::proptest;
+
+    #[proptest]
+    fn test_arbitrary_map_u8(ranges: Vec<(Range<u8>, String)>) {
+        let ranges: Vec<_> = ranges
+            .into_iter()
+            .filter(|(range, _value)| range.start != range.end)
+            .collect();
+        let set = ranges
+            .iter()
+            .fold(RangeMap::new(), |mut set, (range, value)| {
+                set.insert(range.clone(), value.clone());
+                set
+            });
+
+        for value in 0..u8::MAX {
+            assert_eq!(
+                set.get(&value),
+                ranges
+                    .iter()
+                    .rev()
+                    .find(|(range, _value)| range.contains(&value))
+                    .map(|(_range, value)| value)
+            );
+        }
+    }
 
     trait RangeMapExt<K, V> {
         fn to_vec(&self) -> Vec<(Range<K>, V)>;

diff --git a/src/set.rs b/src/set.rs
@@ -345,7 +345,28 @@ where
 #[cfg(test)]
 mod tests {
     use super::*;
+    use alloc as std;
     use alloc::{format, vec, vec::Vec};
+    use test_strategy::proptest;
+
+    #[proptest]
+    fn test_arbitrary_set_u8(ranges: Vec<Range<u8>>) {
+        let ranges: Vec<_> = ranges
+            .into_iter()
+            .filter(|range| range.start != range.end)
+            .collect();
+        let set = ranges.iter().fold(RangeSet::new(), |mut set, range| {
+            set.insert(range.clone());
+            set
+        });
+
+        for value in 0..u8::MAX {
+            assert_eq!(
+                set.contains(&value),
+                ranges.iter().any(|range| range.contains(&value))
+            );
+        }
+    }
 
     trait RangeSetExt<T> {
         fn to_vec(&self) -> Vec<Range<T>>;