diff options
Diffstat (limited to 'benchmark/benchmark-pandoc.hs')
| -rw-r--r-- | benchmark/benchmark-pandoc.hs | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/benchmark/benchmark-pandoc.hs b/benchmark/benchmark-pandoc.hs index 90b0eee74..7dbd29bed 100644 --- a/benchmark/benchmark-pandoc.hs +++ b/benchmark/benchmark-pandoc.hs @@ -1,8 +1,11 @@ import Text.Pandoc import Text.Pandoc.Shared (readDataFile, normalize) import Criterion.Main +import Criterion.Config import Data.List (isSuffixOf) import Text.JSON.Generic +import System.Environment (getArgs) +import Data.Monoid readerBench :: Pandoc -> (String, ReaderOptions -> String -> Pandoc) @@ -37,12 +40,14 @@ normalizeBench doc = [ bench "normalize - with" $ nf (encodeJSON . normalize) do main :: IO () main = do + args <- getArgs + (conf,_) <- parseArgs defaultConfig{ cfgSamples = Last $ Just 20 } defaultOptions args inp <- readDataFile (Just ".") "README" inp2 <- readDataFile (Just ".") "tests/testsuite.txt" let opts = def{ readerSmart = True } let doc = readMarkdown opts $ inp ++ unlines (drop 3 $ lines inp2) let readerBs = map (readerBench doc) readers let writers' = [(n,w) | (n, PureStringWriter w) <- writers] - defaultMain $ + defaultMainWith conf (return ()) $ map (writerBench doc) writers' ++ readerBs ++ normalizeBench doc |