dependency-0.1.0.2: bench/Bench.hs

module Main where

import           Control.Arrow
import           Criterion.Main
import           Data.Dependency
import qualified Data.Map        as M
import qualified Data.Set        as S

microlens :: Dependency
microlens = Dependency "microlens" mempty mempty

free :: Dependency
free = Dependency "free" mempty mempty

comonad :: Dependency
comonad = Dependency "comonad" mempty mempty

bifunctors :: Dependency
bifunctors = Dependency "bifunctors" mempty ["comonad"]

lens :: Dependency
lens = Dependency "lens" mempty ["free", "comonad"]

deps :: [Dependency]
deps = [free, lens, comonad]

mapSingles :: [(d, b)] -> [(d, S.Set b)]
mapSingles = fmap (second S.singleton)

set :: PackageSet Dependency
set = PackageSet $ M.fromList (mapSingles [("lens", lens), ("comonad", comonad), ("free", free)])

main :: IO ()
main =
    defaultMain [ bgroup "buildSequence"
                      [ bench "3" $ nf buildSequence deps
                      , bench "6" $ nf buildSequence (bifunctors : microlens : deps)
                      ]
                , bgroup "resolveDependencies"
                      [ bench "3" $ nf (resolveDependencies set) (buildSequence [lens]) ]
                ]