tvl-depot/users/glittershark/xanthous/bench/Xanthous/RandomBench.hs
Griffin Smith 7252468b34 feat(xan): Add a benchmark suite
Change-Id: Id31960e7bc2243dfa53dc5e45b09d8253bdef852
Reviewed-on: https://cl.tvl.fyi/c/depot/+/727
Reviewed-by: glittershark <grfn@gws.fyi>
2020-06-28 23:38:48 +00:00

32 lines
1.1 KiB
Haskell

--------------------------------------------------------------------------------
module Xanthous.RandomBench (benchmark, main) where
--------------------------------------------------------------------------------
import Bench.Prelude
--------------------------------------------------------------------------------
import Control.Parallel.Strategies
import Control.Monad.Random
--------------------------------------------------------------------------------
import Xanthous.Random
--------------------------------------------------------------------------------
main :: IO ()
main = defaultMain [benchmark]
--------------------------------------------------------------------------------
benchmark :: Benchmark
benchmark = bgroup "Random"
[ bgroup "chooseSubset"
[ bench "serially" $
nf (evalRand $ chooseSubset (0.5 :: Double) [1 :: Int ..1000000])
(mkStdGen 1234)
]
, bgroup "choose weightedBy"
[ bench "serially" $
nf (evalRand
. choose
. weightedBy (\n -> product [n, pred n .. 1])
$ [1 :: Int ..1000000])
(mkStdGen 1234)
]
]