users/aspen/xanthous/bench/Xanthous/RandomBench.hs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

--------------------------------------------------------------------------------
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)
    ]
  ]