diff --git a/.gitignore b/.gitignore index f1b916b5..8229cb75 100644 --- a/.gitignore +++ b/.gitignore @@ -22,6 +22,8 @@ __pycache__ pip-wheel-metadata +bench/*.svg + .benchmarks .DS_Store .idea diff --git a/bench/test_cost.py b/bench/test_cost.py index c9f82b65..682ac0ed 100644 --- a/bench/test_cost.py +++ b/bench/test_cost.py @@ -213,7 +213,8 @@ def run(): @pytest.mark.parametrize("n", N) @pytest.mark.parametrize("BatchMode", [False, True]) @pytest.mark.parametrize("NumCPU", [0, nb.get_num_threads()]) -def test_RooFit(benchmark, n, BatchMode, NumCPU): +@pytest.mark.parametrize("EvalBackend", ["legacy", "cpu"]) +def test_RooFit(benchmark, n, BatchMode, NumCPU, EvalBackend): import ROOT as R x = R.RooRealVar("x", "x", 0, 1) @@ -232,7 +233,11 @@ def run(): sigma.setVal(0.1) slope.setVal(1) z.setVal(0.5) - args = [R.RooFit.PrintLevel(-1), R.RooFit.BatchMode(BatchMode)] + args = [ + R.RooFit.PrintLevel(-1), + R.RooFit.BatchMode(BatchMode), + R.RooFit.EvalBackend(EvalBackend), + ] if NumCPU: args.append(R.RooFit.NumCPU(NumCPU)) pdf.fitTo(data, *args)