Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2026-03-22 05:00:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
blueloveTH 2026-03-03 14:55:43 +08:00
parent 2d2fb620ca
commit 071ee89676
2 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/public/PyDict.c
View File

@ -568,22 +568,25 @@ static bool dict_pop(int argc, py_Ref argv) {
if(res == 0) py_assign(py_retval(), default_val); if(res == 0) py_assign(py_retval(), default_val);
return true; return true;
} }
static bool dict_popitem(int argc, py_Ref argv) { static bool dict_popitem(int argc, py_Ref argv) {
PY_CHECK_ARGC(1); PY_CHECK_ARGC(1);
Dict* self = py_touserdata(argv); Dict* self = py_touserdata(argv);
if(self->length == 0) return KeyError(py_None());
for(int i = self->entries.length - 1; i >= 0; i--) { for(int i = self->entries.length - 1; i >= 0; i--) {
DictEntry* entry = c11__at(DictEntry, &self->entries, i); DictEntry* entry = c11__at(DictEntry, &self->entries, i);
if(py_isnil(&entry->key)) continue; if(py_isnil(&entry->key)) continue;
py_Ref p = py_newtuple(py_retval(), 2); py_Ref p = py_newtuple(py_pushtmp(), 2);
p[0] = entry->key; p[0] = entry->key;
p[1] = entry->val; p[1] = entry->val;
py_TValue key_copy = entry->key; int res = Dict__pop(self, &p[0]);
Dict__pop(self, &key_copy); c11__rtassert(res == 1);
py_assign(py_retval(), py_peek(-1));
py_pop();
return true; return true;
} }
return KeyError(py_None()); return KeyError(py_None());
} }
static bool dict_keys(int argc, py_Ref argv) { static bool dict_keys(int argc, py_Ref argv) {
PY_CHECK_ARGC(1); PY_CHECK_ARGC(1);
Dict* self = py_touserdata(argv); Dict* self = py_touserdata(argv);

11
tests/080_dict.py
View File

@ -142,6 +142,17 @@ for i in range(n):
del a[str(i)] del a[str(i)]
assert len(a) == 0 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 # test del with int keys
if 0: if 0:
n = 2 ** 17 n = 2 ** 17