Add wrapper benchmark

RoseauTechnologies · Oct 20, 2023 · bffc5ea · bffc5ea
1 parent 2bdb258
commit bffc5ea
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/pint/testsuite/benchmarks/test_20_quantity.py b/pint/testsuite/benchmarks/test_20_quantity.py
@@ -53,3 +53,39 @@ def test_op2(benchmark, setup, keys, op):
     _, data = setup
     key1, key2 = keys
     benchmark(op, data[key1], data[key2])
+
+
+@pytest.mark.parametrize("key", ALL_VALUES_Q)
+def test_wrapper(benchmark, setup, key):
+    ureg, data = setup
+    value, unit = key.split("_")
+
+    @ureg.wraps(None, (unit,))
+    def f(a):
+        pass
+
+    benchmark(f, data[key])
+
+
+@pytest.mark.parametrize("key", ALL_VALUES_Q)
+def test_wrapper_nonstrict(benchmark, setup, key):
+    ureg, data = setup
+    value, unit = key.split("_")
+
+    @ureg.wraps(None, (unit,), strict=False)
+    def f(a):
+        pass
+
+    benchmark(f, data[value])
+
+
+@pytest.mark.parametrize("key", ALL_VALUES_Q)
+def test_wrapper_ret(benchmark, setup, key):
+    ureg, data = setup
+    value, unit = key.split("_")
+
+    @ureg.wraps(unit, (unit,))
+    def f(a):
+        return a
+
+    benchmark(f, data[key])