github/semantic
1
1
Fork 0
mirror of https://github.com/github/semantic.git synced 2024-12-18 04:11:48 +03:00
Code Issues Projects Releases Wiki Activity

Generalize generativeBenchmark over Gen a.

Browse Source
This commit is contained in:
Rob Rix 2016-06-06 17:15:51 -04:00
parent 19fc31747e
commit 2648406b34
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
bench/Main.hs
View File

@ -26,11 +26,14 @@ instance NFData a => NFData (Patch a)
-- | -- |
-- | The inputs’ sizes at each iteration are measured by a metric function, which gives the name of the benchmark. This makes it convenient to correlate a benchmark of some function over lists with e.g. input `length`. -- | The inputs’ sizes at each iteration are measured by a metric function, which gives the name of the benchmark. This makes it convenient to correlate a benchmark of some function over lists with e.g. input `length`.
generativeBenchmark :: (Arbitrary a, Show m, Ord m) => String -> Int -> (a -> m) -> (a -> Benchmarkable) -> IO Benchmark generativeBenchmark :: (Arbitrary a, Show m, Ord m) => String -> Int -> (a -> m) -> (a -> Benchmarkable) -> IO Benchmark
generativeBenchmark name n metric benchmark = do generativeBenchmark name n metric benchmark = generativeBenchmarkWith name n arbitrary metric benchmark
generativeBenchmarkWith :: (Show m, Ord m) => String -> Int -> Gen a -> (a -> m) -> (a -> Benchmarkable) -> IO Benchmark
generativeBenchmarkWith name n generator metric benchmark = do
benchmarks <- traverse measure (replicate n defaultSize) benchmarks <- traverse measure (replicate n defaultSize)
pure $! bgroup name (snd <$> (sortOn fst benchmarks)) pure $! bgroup name (snd <$> (sortOn fst benchmarks))
where measure n = do where measure n = do
input <- generate (resize n arbitrary) input <- generate (resize n generator)
let measurement = metric input let measurement = metric input
pure $! (measurement, bench (show measurement) (benchmark input)) pure $! (measurement, bench (show measurement) (benchmark input))
defaultSize = 100 defaultSize = 100