benchmark fixture?

lmfit · Dec 18, 2024 · f98349e · f98349e
1 parent 5f4d55a
commit f98349e
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/tests/test_performance.py b/tests/test_performance.py
@@ -2,7 +2,6 @@
 import time
 import timeit
 
-import pytest
 
 from uncertainties import ufloat
 
@@ -39,7 +38,7 @@ def test_complexity():
         assert 0.9 * log10(n / n0) < log10(t / t0) < 1.1 * log10(n / n0)
 
 
-@pytest.mark.parametrize("num", (10, 100, 1000, 10000, 100000))
-@pytest.mark.benchmark
-def test_speed(num):
-    time_ufloat_sum_benchmark(num)
+def test_speed(benchmark):
+    num_list = (10, 100, 1000, 10000, 100000)
+    for num in num_list:
+        benchmark(lambda: time_ufloat_sum_benchmark(num))