Change distribution of smallDenominators.

nick8325 · Jan 8, 2025 · a1047d3 · a1047d3
1 parent ad62910
commit a1047d3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/Test/QuickCheck/Arbitrary.hs b/src/Test/QuickCheck/Arbitrary.hs
@@ -722,8 +722,8 @@ instance Arbitrary Float where
     ]
     where
       smallDenominators = sized $ \n -> do
-        i <- chooseInt (0, n)
-        pure (fromRational (streamNth (min i 256) rationalUniverse))
+        i <- chooseInt (0, min n 256)
+        pure (fromRational (streamNth i rationalUniverse))
 
       uniform = sized $ \n -> do
         let n' = toInteger n
@@ -748,8 +748,8 @@ instance Arbitrary Double where
     ]
     where
       smallDenominators = sized $ \n -> do
-        i <- chooseInt (0, n)
-        pure (fromRational (streamNth (min i 256) rationalUniverse))
+        i <- chooseInt (0, min n 256)
+        pure (fromRational (streamNth i rationalUniverse))
 
       uniform = sized $ \n -> do
         let n' = toInteger n