55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
"""Check set_issuperset's C89 declarations and unchanged iterator ownership."""
|
|
import re
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
source = Path(sys.argv[1]).read_text()
|
|
match = re.search(
|
|
r"static PyObject \*\nset_issuperset\(PySetObject \*so, PyObject \*other\)"
|
|
r"\n\{(.*?)\n\}",
|
|
source,
|
|
re.S,
|
|
)
|
|
assert match, "set_issuperset not found"
|
|
body = match.group(1)
|
|
assert re.match(r"\s*PyObject \*key;\s*PyObject \*it;", body), (
|
|
"set_issuperset: key and it must be declared in the initial declaration "
|
|
"block; late PyObject declaration after PyAnySet_Check is not C89 compatible"
|
|
)
|
|
expected = """
|
|
PyObject *key;
|
|
PyObject *it;
|
|
|
|
if (PyAnySet_Check(other)) {
|
|
return set_issubset((PySetObject *)other, (PyObject *)so);
|
|
}
|
|
|
|
it = PyObject_GetIter(other);
|
|
if (it == NULL) {
|
|
return NULL;
|
|
}
|
|
while ((key = PyIter_Next(it)) != NULL) {
|
|
int rv = set_contains_key(so, key);
|
|
Py_DECREF(key);
|
|
if (rv < 0) {
|
|
Py_DECREF(it);
|
|
return NULL;
|
|
}
|
|
if (!rv) {
|
|
Py_DECREF(it);
|
|
Py_RETURN_FALSE;
|
|
}
|
|
}
|
|
Py_DECREF(it);
|
|
if (PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
Py_RETURN_TRUE;
|
|
"""
|
|
assert re.sub(r"\s+", "", body) == re.sub(r"\s+", "", expected), (
|
|
"preserve the set fast path, iterator acquisition, loop, reference ownership, "
|
|
"error handling, and return behavior"
|
|
)
|
|
print("set_issuperset C89 declarations and iterator ownership: PASS")
|