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:*", "_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _*",