arithmoi-0.13.4.0: benchmark/Math/NumberTheory/SmoothNumbersBench.hs

{-# OPTIONS_GHC -fno-warn-type-defaults #-}

module Math.NumberTheory.SmoothNumbersBench
  ( benchSuite
  ) where

import Test.Tasty.Bench

import Math.NumberTheory.Primes
import Math.NumberTheory.SmoothNumbers

doBench :: Int -> Int
doBench lim = sum $ take lim $ smoothOver $ fromList $ map unPrime [nextPrime 2 .. precPrime lim]

benchSuite :: Benchmark
benchSuite = bgroup "SmoothNumbers"
  [ bench "100"      $ nf doBench   100
  , bench "1000"     $ nf doBench  1000
  , bench "10000"    $ nf doBench 10000
  ]