From a3e6d0581c58604c4b16b4c873f22d34521c8ca3 Mon Sep 17 00:00:00 2001
From: Joshua Moerman <joshua@cs.rwth-aachen.de>
Date: Thu, 28 May 2020 09:21:42 +0200
Subject: [PATCH] simpler benchmarks

---
 bench/Bench.hs      | 13 ++++++++++---
 nominal-lstar.cabal |  2 +-
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/bench/Bench.hs b/bench/Bench.hs
index 3c723e2..70956b1 100644
--- a/bench/Bench.hs
+++ b/bench/Bench.hs
@@ -1,11 +1,18 @@
+{-# OPTIONS_GHC -Wno-missing-signatures #-}
 import Bollig
 import Examples
 import Teacher
 
-import Criterion.Main
+import Gauge.Main
+import Gauge.Main.Options
 
-main :: IO ()
-main = defaultMain [
+-- Run a single test since these are slow benchmarks
+myConfig = defaultConfig
+  { quickMode = True
+  , includeFirstIter = True
+  }
+
+main = defaultMainWith myConfig [
   bgroup "NomNLStar"
     [ bench "NFA1 -" $ whnf (learnBollig 0 0) (teacherWithTargetNonDet 2 Examples.exampleNFA1)
     , bench "NFA2 1" $ whnf (learnBollig 0 0) (teacherWithTargetNonDet 3 (Examples.exampleNFA2 1))
diff --git a/nominal-lstar.cabal b/nominal-lstar.cabal
index 87a0a91..3a98aaa 100644
--- a/nominal-lstar.cabal
+++ b/nominal-lstar.cabal
@@ -58,5 +58,5 @@ benchmark bench
   type: exitcode-stdio-1.0
   main-is: Bench.hs
   build-depends:
-    criterion,
+    gauge,
     nominal-lstar