{-# OPTIONS_GHC -Wno-missing-signatures #-}
import Bollig
import Examples
import Teacher

import Gauge.Main
import Gauge.Main.Options

-- Run a single test since these are slow benchmarks
myConfig = defaultConfig
  { quickMode = True
  , includeFirstIter = True
  , csvFile = Just "bench.csv"
  }

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))
    , bench "NFA2 2  " $ whnf (learnBollig 0 0) (teacherWithTargetNonDet 4 (Examples.exampleNFA2 2))
    , bench "Residual" $ whnf (learnBollig 1 0) (teacherWithTargetNonDet 2 Examples.exampleResidual1)
    ]
  ]