Issue #27275: Fixed implementation of pop() and popitem() methods in
subclasses of accelerated OrderedDict.
This commit is contained in:
parent
7ff51bd2b8
commit
4832580596
@ -730,5 +730,64 @@ class CPythonSubclassMappingTests(mapping_tests.BasicTestMappingProtocol):
|
|||||||
self.assertRaises(KeyError, d.popitem)
|
self.assertRaises(KeyError, d.popitem)
|
||||||
|
|
||||||
|
|
||||||
|
class SimpleLRUCache:
|
||||||
|
|
||||||
|
def __init__(self, size):
|
||||||
|
super().__init__()
|
||||||
|
self.size = size
|
||||||
|
|
||||||
|
def __getitem__(self, item):
|
||||||
|
value = super().__getitem__(item)
|
||||||
|
self.move_to_end(item)
|
||||||
|
return value
|
||||||
|
|
||||||
|
def __setitem__(self, key, value):
|
||||||
|
while key not in self and len(self) >= self.size:
|
||||||
|
self.popitem(last=False)
|
||||||
|
super().__setitem__(key, value)
|
||||||
|
self.move_to_end(key)
|
||||||
|
|
||||||
|
|
||||||
|
class SimpleLRUCacheTests:
|
||||||
|
|
||||||
|
def test_add_after_full(self):
|
||||||
|
c = self.type2test(2)
|
||||||
|
c['t1'] = 1
|
||||||
|
c['t2'] = 2
|
||||||
|
c['t3'] = 3
|
||||||
|
self.assertEqual(list(c), ['t2', 't3'])
|
||||||
|
|
||||||
|
def test_popitem(self):
|
||||||
|
c = self.type2test(3)
|
||||||
|
for i in range(1, 4):
|
||||||
|
c[i] = i
|
||||||
|
self.assertEqual(c.popitem(last=False), (1, 1))
|
||||||
|
self.assertEqual(c.popitem(last=True), (3, 3))
|
||||||
|
|
||||||
|
def test_change_order_on_get(self):
|
||||||
|
c = self.type2test(3)
|
||||||
|
for i in range(1, 4):
|
||||||
|
c[i] = i
|
||||||
|
self.assertEqual(list(c), list(range(1, 4)))
|
||||||
|
self.assertEqual(c[2], 2)
|
||||||
|
self.assertEqual(list(c), [1, 3, 2])
|
||||||
|
|
||||||
|
|
||||||
|
class PySimpleLRUCacheTests(SimpleLRUCacheTests, unittest.TestCase):
|
||||||
|
|
||||||
|
class type2test(SimpleLRUCache, py_coll.OrderedDict):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@unittest.skipUnless(c_coll, 'requires the C version of the collections module')
|
||||||
|
class CSimpleLRUCacheTests(SimpleLRUCacheTests, unittest.TestCase):
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def setUpClass(cls):
|
||||||
|
class type2test(SimpleLRUCache, c_coll.OrderedDict):
|
||||||
|
pass
|
||||||
|
cls.type2test = type2test
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
@ -113,6 +113,9 @@ Core and Builtins
|
|||||||
Library
|
Library
|
||||||
-------
|
-------
|
||||||
|
|
||||||
|
- Issue #27275: Fixed implementation of pop() and popitem() methods in
|
||||||
|
subclasses of accelerated OrderedDict.
|
||||||
|
|
||||||
- Issue #28255: calendar.TextCalendar().prmonth() no longer prints a space
|
- Issue #28255: calendar.TextCalendar().prmonth() no longer prints a space
|
||||||
at the start of new line after printing a month's calendar. Patch by
|
at the start of new line after printing a month's calendar. Patch by
|
||||||
Xiang Zhang.
|
Xiang Zhang.
|
||||||
|
@ -1099,28 +1099,13 @@ _odict_popkey_hash(PyObject *od, PyObject *key, PyObject *failobj,
|
|||||||
}
|
}
|
||||||
|
|
||||||
/* Now delete the value from the dict. */
|
/* Now delete the value from the dict. */
|
||||||
if (PyODict_CheckExact(od)) {
|
if (node != NULL) {
|
||||||
if (node != NULL) {
|
value = _PyDict_GetItem_KnownHash(od, key, hash); /* borrowed */
|
||||||
value = _PyDict_GetItem_KnownHash(od, key, hash); /* borrowed */
|
if (value != NULL) {
|
||||||
if (value != NULL) {
|
Py_INCREF(value);
|
||||||
Py_INCREF(value);
|
if (_PyDict_DelItem_KnownHash(od, key, hash) < 0) {
|
||||||
if (_PyDict_DelItem_KnownHash(od, key, hash) < 0) {
|
Py_DECREF(value);
|
||||||
Py_DECREF(value);
|
return NULL;
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
int exists = PySequence_Contains(od, key);
|
|
||||||
if (exists < 0)
|
|
||||||
return NULL;
|
|
||||||
if (exists) {
|
|
||||||
value = PyObject_GetItem(od, key);
|
|
||||||
if (value != NULL) {
|
|
||||||
if (PyObject_DelItem(od, key) == -1) {
|
|
||||||
Py_CLEAR(value);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user