Merge pull request #281 from mikerite/refactor-20221116

Simplify non_zero arbitrary impl
proptest-rs · Jan 6, 2023 · d666b7e · d666b7e
2 parents 932c1d6 + 2055d39
commit d666b7e
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/proptest/src/arbitrary/non_zero.rs b/proptest/src/arbitrary/non_zero.rs
@@ -25,13 +25,9 @@ macro_rules! non_zero_impl {
                 FilterMap<StrategyFor<$prim>, fn($prim) -> Option<Self>>;
 
             fn arbitrary_with((): Self::Parameters) -> Self::Strategy {
-                any::<$prim>().prop_filter_map(
-                    "must be non zero",
-                    |i| match i {
-                        0 => None,
-                        i => Some(Self::try_from(i).unwrap()),
-                    },
-                )
+                any::<$prim>().prop_filter_map("must be non zero", |i| {
+                    Self::try_from(i).ok()
+                })
             }
         }
     };