Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2026-03-21 20:50:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: implement dict.popitem() method (#443) (#467)

Browse Source 
* Added popitem()

* fix

---------

Co-authored-by: blueloveTH <blueloveTH@foxmail.com>
This commit is contained in:
Tanisha Vasudeva 2026-03-03 12:25:56 +05:30 committed by GitHub
parent 5f9d44f451
commit 1d33267092
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/public/PyDict.c
View File

@ -569,6 +569,24 @@ static bool dict_pop(int argc, py_Ref argv) {
return true;
}
static bool dict_popitem(int argc, py_Ref argv) {
PY_CHECK_ARGC(1);
Dict* self = py_touserdata(argv);
for(int i = self->entries.length - 1; i >= 0; i--) {
DictEntry* entry = c11__at(DictEntry, &self->entries, i);
if(py_isnil(&entry->key)) continue;
py_Ref p = py_newtuple(py_pushtmp(), 2);
p[0] = entry->key;
p[1] = entry->val;
int res = Dict__pop(self, &p[0]);
c11__rtassert(res == 1);
py_assign(py_retval(), py_peek(-1));
py_pop();
return true;
}
return KeyError(py_None());
}
static bool dict_keys(int argc, py_Ref argv) {
PY_CHECK_ARGC(1);
Dict* self = py_touserdata(argv);
@ -616,6 +634,7 @@ py_Type pk_dict__register() {
py_bindmethod(type, "update", dict_update);
py_bindmethod(type, "get", dict_get);
py_bindmethod(type, "pop", dict_pop);
py_bindmethod(type, "popitem", dict_popitem);
py_bindmethod(type, "keys", dict_keys);
py_bindmethod(type, "values", dict_values);
py_bindmethod(type, "items", dict_items);

11
tests/080_dict.py
View File

@ -142,6 +142,17 @@ for i in range(n):
del a[str(i)]
assert len(a) == 0
# test popitem
n = 2 ** 17
a = {}
for i in range(n):
a[str(i)] = i
for i in range(n):
k, v = a.popitem()
assert k == str(n - 1 - i)
assert v == n - 1 - i
assert len(a) == 0
# test del with int keys
if 0:
n = 2 ** 17