diff options
| -rw-r--r-- | test/Gen.hs | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/test/Gen.hs b/test/Gen.hs index b10e763..4f5fe96 100644 --- a/test/Gen.hs +++ b/test/Gen.hs @@ -78,9 +78,8 @@ genShRwithTarget targetMax sn = do dims <- genDims m (if dim == 0 then 0 else tgt `div` dim) return (dim :$: dims) dims <- genDims sn targetSize - let dimsL = toList dims - maxdim = maximum dimsL - cap = binarySearch (`div` 2) 1 maxdim (\cap' -> product (min cap' <$> dimsL) <= targetSize) + let maxdim = maximum dims + cap = binarySearch (`div` 2) 1 maxdim (\cap' -> shrSize (min cap' <$> dims) <= targetSize) shuffleShR (min cap <$> dims) -- | Example: given 3 and 7, might return: |