46 lines
1.4 KiB
Diff
46 lines
1.4 KiB
Diff
diff --git a/src/pytest_benchmark/compat.py b/src/pytest_benchmark/compat.py
|
|
index 63d01bd..9afecf2 100644
|
|
--- a/src/pytest_benchmark/compat.py
|
|
+++ b/src/pytest_benchmark/compat.py
|
|
@@ -1,3 +1,4 @@
|
|
import sys
|
|
|
|
PY38 = sys.version_info[0] == 3 and sys.version_info[1] >= 8
|
|
+PY311 = sys.version_info[0] == 3 and sys.version_info[1] >= 11
|
|
diff --git a/src/pytest_benchmark/utils.py b/src/pytest_benchmark/utils.py
|
|
index c80352a..e28c04e 100644
|
|
--- a/src/pytest_benchmark/utils.py
|
|
+++ b/src/pytest_benchmark/utils.py
|
|
@@ -26,7 +26,7 @@
|
|
|
|
import genericpath
|
|
|
|
-from .compat import PY38
|
|
+from .compat import PY38, PY311
|
|
|
|
# This is here (in the utils module) because it might be used by
|
|
# various other modules.
|
|
@@ -521,6 +521,10 @@ def clonefunc(f):
|
|
co.co_firstlineno, co.co_lnotab, co.co_freevars, co.co_cellvars]
|
|
if PY38:
|
|
args.insert(1, co.co_posonlyargcount)
|
|
+
|
|
+ if PY311:
|
|
+ args.insert(12, co.co_qualname)
|
|
+ args.insert(15, co.co_exceptiontable)
|
|
co2 = types.CodeType(*args)
|
|
#
|
|
# then, we clone the function itself, using the new co2
|
|
diff --git a/tests/test_benchmark.py b/tests/test_benchmark.py
|
|
index 280ce24..964787f 100644
|
|
--- a/tests/test_benchmark.py
|
|
+++ b/tests/test_benchmark.py
|
|
@@ -952,7 +952,6 @@ def result():
|
|
|
|
" def test_bad(benchmark):",
|
|
"? @benchmark",
|
|
- "? def result():",
|
|
|
|
"test_abort_broken.py:*",
|
|
"_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _*",
|