40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Check frozenset_vectorcall's C89 declarations and evaluation order."""
|
|
import re
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
source = Path(sys.argv[1]).read_text()
|
|
match = re.search(
|
|
r"static PyObject \*\nfrozenset_vectorcall\(PyObject \*type, PyObject \* const\*args,"
|
|
r"\s*size_t nargsf, PyObject \*kwnames\)\n\{(.*?)\n\}",
|
|
source,
|
|
re.S,
|
|
)
|
|
assert match, "frozenset_vectorcall not found"
|
|
body = match.group(1)
|
|
assert re.match(r"\s*Py_ssize_t nargs;\s*PyObject \*iterable;", body), (
|
|
"nargs and iterable must be declared without initializers in the initial block"
|
|
)
|
|
expected = """
|
|
Py_ssize_t nargs;
|
|
PyObject *iterable;
|
|
|
|
if (!_PyArg_NoKwnames("frozenset", kwnames)) {
|
|
return NULL;
|
|
}
|
|
|
|
nargs = PyVectorcall_NARGS(nargsf);
|
|
if (!_PyArg_CheckPositional("frozenset", nargs, 0, 1)) {
|
|
return NULL;
|
|
}
|
|
|
|
iterable = (nargs ? args[0] : NULL);
|
|
return make_new_frozenset(_PyType_CAST(type), iterable);
|
|
"""
|
|
assert re.sub(r"\s+", "", body) == re.sub(r"\s+", "", expected), (
|
|
"preserve keyword rejection, nargs evaluation, positional rejection, guarded "
|
|
"iterable evaluation, and construction in their original order"
|
|
)
|
|
print("frozenset_vectorcall C89 declarations and evaluation order: PASS")
|