60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
"""Check type_new_set_module's module declaration and lookup timing."""
|
|
import re
|
|
from pathlib import Path
|
|
|
|
source = Path("cpython/Objects/typeobject.c").read_text()
|
|
match = re.search(
|
|
r"\nstatic int\ntype_new_set_module\(PyTypeObject \*type\)\n\{(.*?)\n\}",
|
|
source,
|
|
re.S,
|
|
)
|
|
assert match, "type_new_set_module definition missing"
|
|
body = re.sub(r"/\*.*?\*/|//[^\n]*", "", match.group(1), flags=re.S)
|
|
assert re.match(
|
|
r"\s*PyObject \*globals;"
|
|
r"\s*PyObject \*module;"
|
|
r"\s*int r = PyDict_Contains\(type->tp_dict, &_Py_ID\(__module__\)\);",
|
|
body,
|
|
), "type_new_set_module: module must be in the opening declaration block (C89)"
|
|
assert len(re.findall(r"\bPyObject\s*\*\s*module\b", body)) == 1, (
|
|
"declare module exactly once"
|
|
)
|
|
assert len(re.findall(r"\bmodule\s*=(?!=)", body)) == 1, (
|
|
"assign module exactly once"
|
|
)
|
|
expected = """
|
|
PyObject *globals;
|
|
PyObject *module;
|
|
int r = PyDict_Contains(type->tp_dict, &_Py_ID(__module__));
|
|
if (r < 0) {
|
|
return -1;
|
|
}
|
|
if (r > 0) {
|
|
return 0;
|
|
}
|
|
|
|
globals = PyEval_GetGlobals();
|
|
if (globals == NULL) {
|
|
return 0;
|
|
}
|
|
|
|
module = PyDict_GetItemWithError(globals, &_Py_ID(__name__));
|
|
if (module == NULL) {
|
|
if (PyErr_Occurred()) {
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
if (PyDict_SetItem(type->tp_dict, &_Py_ID(__module__), module) < 0) {
|
|
return -1;
|
|
}
|
|
return 0;
|
|
"""
|
|
assert body.split() == expected.split(), (
|
|
"type_new_set_module: preserve lookup/assignment order, early returns, "
|
|
"and module consumption"
|
|
)
|
|
print("type_new_set_module module C89 declaration and lookup timing: PASS")
|