diff options
author | John MacFarlane <jgm@berkeley.edu> | 2017-01-27 11:29:26 +0100 |
---|---|---|
committer | John MacFarlane <jgm@berkeley.edu> | 2017-01-27 11:29:26 +0100 |
commit | 86b9a51ee3df623b4f09588e2d07ce0e36e23f29 (patch) | |
tree | 40d480a09495493d2e9ba7bb210e54eaeea44dc5 /benchmark/benchmark-pandoc.hs | |
parent | 8a61d943f590f8ba1df8dbc5840301e26c9b4db9 (diff) | |
download | pandoc-86b9a51ee3df623b4f09588e2d07ce0e36e23f29.tar.gz |
benchmark: allow benchmark arguments.
These pattern match: so, '--benchmark-arguments "markdown reader"'
will only benchmark the markdown reader; with just "markdown" it
will do the writer too; with no arguments all benchmarks are run.
Diffstat (limited to 'benchmark/benchmark-pandoc.hs')
-rw-r--r-- | benchmark/benchmark-pandoc.hs | 20 |
1 files changed, 18 insertions, 2 deletions
diff --git a/benchmark/benchmark-pandoc.hs b/benchmark/benchmark-pandoc.hs index d5e3405be..c01750b6e 100644 --- a/benchmark/benchmark-pandoc.hs +++ b/benchmark/benchmark-pandoc.hs @@ -24,6 +24,7 @@ import Criterion.Main import Criterion.Types (Config(..)) import Data.Maybe (mapMaybe) import Debug.Trace (trace) +import System.Environment (getArgs) readerBench :: Pandoc -> (String, ReaderOptions -> String -> Pandoc) @@ -45,6 +46,21 @@ writerBench doc (name, writer) = bench (name ++ " writer") $ nf main :: IO () main = do + args <- getArgs + let matchReader (n, StringReader _) = + case args of + [] -> True + [x] -> x == n + (x:y:_) -> x == n && y == "reader" + matchReader (_, _) = False + let matchWriter (n, StringWriter _) = + case args of + [] -> True + [x] -> x == n + (x:y:_) -> x == n && y == "writer" + matchWriter (_, _) = False + let matchedReaders = filter matchReader readers + let matchedWriters = filter matchWriter writers inp <- readFile "tests/testsuite.txt" lalune <- B.readFile "tests/lalune.jpg" movie <- B.readFile "tests/movie.jpg" @@ -58,12 +74,12 @@ main = do let doc = either (error . show) id $ runPure $ readMarkdown opts inp let readers' = [(n, \o d -> either (error . show) id $ runPure $ r o d) - | (n, StringReader r) <- readers] + | (n, StringReader r) <- matchedReaders] let readerBs = mapMaybe (readerBench doc) $ filter (\(n,_) -> n /="haddock") readers' let writers' = [(n, \o d -> either (error . show) id $ runPure $ setupFakeFiles >> w o d) - | (n, StringWriter w) <- writers] + | (n, StringWriter w) <- matchedWriters] let writerBs = map (writerBench doc) $ writers' defaultMainWith defaultConfig{ timeLimit = 6.0 } |