Merge branch 'develop' into dependabot/github_actions/actions/checkout-4

scikit-hep · Dec 6, 2023 · 0fafeb2 · 0fafeb2
2 parents 1c8765b + 379f2b4
commit 0fafeb2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git 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
@@ -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)