Issue #25421: __sizeof__ methods of builtin types now use dynamic basic size.
This allows sys.getsize() to work correctly with their subclasses with __slots__ defined.
This commit is contained in:
parent
efd7b34d7c
commit
5c4064e8bd
|
@ -1112,6 +1112,36 @@ class SizeofTest(unittest.TestCase):
|
||||||
# weakcallableproxy
|
# weakcallableproxy
|
||||||
check(weakref.proxy(int), size('2Pn2P'))
|
check(weakref.proxy(int), size('2Pn2P'))
|
||||||
|
|
||||||
|
def check_slots(self, obj, base, extra):
|
||||||
|
expected = sys.getsizeof(base) + struct.calcsize(extra)
|
||||||
|
if gc.is_tracked(obj) and not gc.is_tracked(base):
|
||||||
|
expected += struct.calcsize('2Pn') # PyGC_Head
|
||||||
|
self.assertEqual(sys.getsizeof(obj), expected)
|
||||||
|
|
||||||
|
def test_slots(self):
|
||||||
|
# check all subclassable types defined in Objects/ that allow
|
||||||
|
# non-empty __slots__
|
||||||
|
check = self.check_slots
|
||||||
|
class BA(bytearray):
|
||||||
|
__slots__ = 'a', 'b', 'c'
|
||||||
|
check(BA(), bytearray(), '3P')
|
||||||
|
class D(dict):
|
||||||
|
__slots__ = 'a', 'b', 'c'
|
||||||
|
check(D(x=[]), {'x': []}, '3P')
|
||||||
|
class L(list):
|
||||||
|
__slots__ = 'a', 'b', 'c'
|
||||||
|
check(L(), [], '3P')
|
||||||
|
class S(set):
|
||||||
|
__slots__ = 'a', 'b', 'c'
|
||||||
|
check(S(), set(), '3P')
|
||||||
|
class FS(frozenset):
|
||||||
|
__slots__ = 'a', 'b', 'c'
|
||||||
|
check(FS(), frozenset(), '3P')
|
||||||
|
from collections import OrderedDict
|
||||||
|
class OD(OrderedDict):
|
||||||
|
__slots__ = 'a', 'b', 'c'
|
||||||
|
check(OD(x=[]), OrderedDict(x=[]), '3P')
|
||||||
|
|
||||||
def test_pythontypes(self):
|
def test_pythontypes(self):
|
||||||
# check all types defined in Python/
|
# check all types defined in Python/
|
||||||
size = test.support.calcobjsize
|
size = test.support.calcobjsize
|
||||||
|
|
|
@ -10,6 +10,10 @@ Release date: tba
|
||||||
Core and Builtins
|
Core and Builtins
|
||||||
-----------------
|
-----------------
|
||||||
|
|
||||||
|
- Issue #25421: __sizeof__ methods of builtin types now use dynamic basic size.
|
||||||
|
This allows sys.getsize() to work correctly with their subclasses with
|
||||||
|
__slots__ defined.
|
||||||
|
|
||||||
- Issue #25709: Fixed problem with in-place string concatenation and utf-8 cache.
|
- Issue #25709: Fixed problem with in-place string concatenation and utf-8 cache.
|
||||||
|
|
||||||
- Issue #24097: Fixed crash in object.__reduce__() if slot name is freed inside
|
- Issue #24097: Fixed crash in object.__reduce__() if slot name is freed inside
|
||||||
|
|
|
@ -1443,7 +1443,7 @@ deque_sizeof(dequeobject *deque, void *unused)
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
Py_ssize_t blocks;
|
Py_ssize_t blocks;
|
||||||
|
|
||||||
res = sizeof(dequeobject);
|
res = _PyObject_SIZE(Py_TYPE(deque));
|
||||||
blocks = (deque->leftindex + Py_SIZE(deque) + BLOCKLEN - 1) / BLOCKLEN;
|
blocks = (deque->leftindex + Py_SIZE(deque) + BLOCKLEN - 1) / BLOCKLEN;
|
||||||
assert(deque->leftindex + Py_SIZE(deque) - 1 ==
|
assert(deque->leftindex + Py_SIZE(deque) - 1 ==
|
||||||
(blocks - 1) * BLOCKLEN + deque->rightindex);
|
(blocks - 1) * BLOCKLEN + deque->rightindex);
|
||||||
|
|
|
@ -4529,7 +4529,7 @@ dec_sizeof(PyObject *v, PyObject *dummy UNUSED)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(PyDecObject);
|
res = _PyObject_SIZE(Py_TYPE(v));
|
||||||
if (mpd_isdynamic_data(MPD(v))) {
|
if (mpd_isdynamic_data(MPD(v))) {
|
||||||
res += MPD(v)->alloc * sizeof(mpd_uint_t);
|
res += MPD(v)->alloc * sizeof(mpd_uint_t);
|
||||||
}
|
}
|
||||||
|
|
|
@ -847,7 +847,7 @@ static Py_ssize_t
|
||||||
_elementtree_Element___sizeof___impl(ElementObject *self)
|
_elementtree_Element___sizeof___impl(ElementObject *self)
|
||||||
/*[clinic end generated code: output=bf73867721008000 input=70f4b323d55a17c1]*/
|
/*[clinic end generated code: output=bf73867721008000 input=70f4b323d55a17c1]*/
|
||||||
{
|
{
|
||||||
Py_ssize_t result = sizeof(ElementObject);
|
Py_ssize_t result = _PyObject_SIZE(Py_TYPE(self));
|
||||||
if (self->extra) {
|
if (self->extra) {
|
||||||
result += sizeof(ElementObjectExtra);
|
result += sizeof(ElementObjectExtra);
|
||||||
if (self->extra->children != self->extra->_children)
|
if (self->extra->children != self->extra->_children)
|
||||||
|
|
|
@ -423,7 +423,7 @@ buffered_sizeof(buffered *self, void *unused)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(buffered);
|
res = _PyObject_SIZE(Py_TYPE(self));
|
||||||
if (self->buffer)
|
if (self->buffer)
|
||||||
res += self->buffer_size;
|
res += self->buffer_size;
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
|
|
|
@ -991,7 +991,7 @@ bytesio_sizeof(bytesio *self, void *unused)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(bytesio);
|
res = _PyObject_SIZE(Py_TYPE(self));
|
||||||
if (self->buf && !SHARED_BUF(self))
|
if (self->buf && !SHARED_BUF(self))
|
||||||
res += _PySys_GetSizeOf(self->buf);
|
res += _PySys_GetSizeOf(self->buf);
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
|
|
|
@ -4015,7 +4015,7 @@ _pickle_Pickler___sizeof___impl(PicklerObject *self)
|
||||||
{
|
{
|
||||||
Py_ssize_t res, s;
|
Py_ssize_t res, s;
|
||||||
|
|
||||||
res = sizeof(PicklerObject);
|
res = _PyObject_SIZE(Py_TYPE(self));
|
||||||
if (self->memo != NULL) {
|
if (self->memo != NULL) {
|
||||||
res += sizeof(PyMemoTable);
|
res += sizeof(PyMemoTable);
|
||||||
res += self->memo->mt_allocated * sizeof(PyMemoEntry);
|
res += self->memo->mt_allocated * sizeof(PyMemoEntry);
|
||||||
|
@ -6418,7 +6418,7 @@ _pickle_Unpickler___sizeof___impl(UnpicklerObject *self)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(UnpicklerObject);
|
res = _PyObject_SIZE(Py_TYPE(self));
|
||||||
if (self->memo != NULL)
|
if (self->memo != NULL)
|
||||||
res += self->memo_size * sizeof(PyObject *);
|
res += self->memo_size * sizeof(PyObject *);
|
||||||
if (self->marks != NULL)
|
if (self->marks != NULL)
|
||||||
|
|
|
@ -1924,7 +1924,7 @@ s_sizeof(PyStructObject *self, void *unused)
|
||||||
Py_ssize_t size;
|
Py_ssize_t size;
|
||||||
formatcode *code;
|
formatcode *code;
|
||||||
|
|
||||||
size = sizeof(PyStructObject) + sizeof(formatcode);
|
size = _PyObject_SIZE(Py_TYPE(self)) + sizeof(formatcode);
|
||||||
for (code = self->s_codes; code->fmtdef != NULL; code++)
|
for (code = self->s_codes; code->fmtdef != NULL; code++)
|
||||||
size += sizeof(formatcode);
|
size += sizeof(formatcode);
|
||||||
return PyLong_FromSsize_t(size);
|
return PyLong_FromSsize_t(size);
|
||||||
|
|
|
@ -1743,7 +1743,7 @@ array_array___sizeof___impl(arrayobject *self)
|
||||||
/*[clinic end generated code: output=d8e1c61ebbe3eaed input=805586565bf2b3c6]*/
|
/*[clinic end generated code: output=d8e1c61ebbe3eaed input=805586565bf2b3c6]*/
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
res = sizeof(arrayobject) + self->allocated * self->ob_descr->itemsize;
|
res = _PyObject_SIZE(Py_TYPE(self)) + self->allocated * self->ob_descr->itemsize;
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -2089,7 +2089,7 @@ product_sizeof(productobject *lz, void *unused)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(productobject);
|
res = _PyObject_SIZE(Py_TYPE(lz));
|
||||||
res += PyTuple_GET_SIZE(lz->pools) * sizeof(Py_ssize_t);
|
res += PyTuple_GET_SIZE(lz->pools) * sizeof(Py_ssize_t);
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
}
|
}
|
||||||
|
@ -2420,7 +2420,7 @@ combinations_sizeof(combinationsobject *co, void *unused)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(combinationsobject);
|
res = _PyObject_SIZE(Py_TYPE(co));
|
||||||
res += co->r * sizeof(Py_ssize_t);
|
res += co->r * sizeof(Py_ssize_t);
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
}
|
}
|
||||||
|
@ -2761,7 +2761,7 @@ cwr_sizeof(cwrobject *co, void *unused)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(cwrobject);
|
res = _PyObject_SIZE(Py_TYPE(co));
|
||||||
res += co->r * sizeof(Py_ssize_t);
|
res += co->r * sizeof(Py_ssize_t);
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
}
|
}
|
||||||
|
@ -3110,7 +3110,7 @@ permutations_sizeof(permutationsobject *po, void *unused)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(permutationsobject);
|
res = _PyObject_SIZE(Py_TYPE(po));
|
||||||
res += PyTuple_GET_SIZE(po->pool) * sizeof(Py_ssize_t);
|
res += PyTuple_GET_SIZE(po->pool) * sizeof(Py_ssize_t);
|
||||||
res += po->r * sizeof(Py_ssize_t);
|
res += po->r * sizeof(Py_ssize_t);
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
|
|
|
@ -719,7 +719,7 @@ mmap__sizeof__method(mmap_object *self, void *unused)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(mmap_object);
|
res = _PyObject_SIZE(Py_TYPE(self));
|
||||||
if (self->tagname)
|
if (self->tagname)
|
||||||
res += strlen(self->tagname) + 1;
|
res += strlen(self->tagname) + 1;
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
|
|
|
@ -397,7 +397,7 @@ parser_sizeof(PyST_Object *st, void *unused)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(PyST_Object) + _PyNode_SizeOf(st->st_node);
|
res = _PyObject_SIZE(Py_TYPE(st)) + _PyNode_SizeOf(st->st_node);
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -2974,7 +2974,7 @@ bytearray_sizeof_impl(PyByteArrayObject *self)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(PyByteArrayObject) + self->ob_alloc * sizeof(char);
|
res = _PyObject_SIZE(Py_TYPE(self)) + self->ob_alloc * sizeof(char);
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -384,7 +384,7 @@ code_sizeof(PyCodeObject *co, void *unused)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(PyCodeObject);
|
res = _PyObject_SIZE(Py_TYPE(co));
|
||||||
if (co->co_cell2arg != NULL && co->co_cellvars != NULL)
|
if (co->co_cell2arg != NULL && co->co_cellvars != NULL)
|
||||||
res += PyTuple_GET_SIZE(co->co_cellvars) * sizeof(unsigned char);
|
res += PyTuple_GET_SIZE(co->co_cellvars) * sizeof(unsigned char);
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
|
|
|
@ -2560,7 +2560,7 @@ _PyDict_SizeOf(PyDictObject *mp)
|
||||||
Py_ssize_t size, res;
|
Py_ssize_t size, res;
|
||||||
|
|
||||||
size = DK_SIZE(mp->ma_keys);
|
size = DK_SIZE(mp->ma_keys);
|
||||||
res = sizeof(PyDictObject);
|
res = _PyObject_SIZE(Py_TYPE(mp));
|
||||||
if (mp->ma_values)
|
if (mp->ma_values)
|
||||||
res += size * sizeof(PyObject*);
|
res += size * sizeof(PyObject*);
|
||||||
/* If the dictionary is split, the keys portion is accounted-for
|
/* If the dictionary is split, the keys portion is accounted-for
|
||||||
|
|
|
@ -2324,7 +2324,7 @@ list_sizeof(PyListObject *self)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(PyListObject) + self->allocated * sizeof(void*);
|
res = _PyObject_SIZE(Py_TYPE(self)) + self->allocated * sizeof(void*);
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -951,8 +951,6 @@ odict_sizeof(PyODictObject *od)
|
||||||
if (res == -1 && PyErr_Occurred())
|
if (res == -1 && PyErr_Occurred())
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
res += sizeof(PyODictObject) - sizeof(PyDictObject);
|
|
||||||
|
|
||||||
/* instance dict */
|
/* instance dict */
|
||||||
pylong = _PyDict_SizeOf((PyDictObject *)od->od_inst_dict);
|
pylong = _PyDict_SizeOf((PyDictObject *)od->od_inst_dict);
|
||||||
if (pylong == NULL)
|
if (pylong == NULL)
|
||||||
|
|
|
@ -1940,7 +1940,7 @@ set_sizeof(PySetObject *so)
|
||||||
{
|
{
|
||||||
Py_ssize_t res;
|
Py_ssize_t res;
|
||||||
|
|
||||||
res = sizeof(PySetObject);
|
res = _PyObject_SIZE(Py_TYPE(so));
|
||||||
if (so->table != so->smalltable)
|
if (so->table != so->smalltable)
|
||||||
res = res + (so->mask + 1) * sizeof(setentry);
|
res = res + (so->mask + 1) * sizeof(setentry);
|
||||||
return PyLong_FromSsize_t(res);
|
return PyLong_FromSsize_t(res);
|
||||||
|
|
Loading…
Reference in New Issue