Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-22 04:20:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
blueloveTH 2023-09-24 19:44:15 +08:00
parent c0866022fb
commit d8f3a53220
2 changed files with 28 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/pocketpy.cpp
View File

@ -1188,10 +1188,19 @@ void init_builtins(VM* _vm) {
if(!ok) vm->KeyError(key);
});
_vm->bind_method<1>("dict", "pop", [](VM* vm, ArgsView args) {
_vm->bind_method<-1>("dict", "pop", [](VM* vm, ArgsView args) {
if(args.size() != 2 && args.size() != 3){
vm->TypeError("pop() expected 1 or 2 arguments");
return vm->None;
}
Dict& self = _CAST(Dict&, args[0]);
PyObject* value = self.try_get(args[1]);
if(value == nullptr) vm->KeyError(args[1]);
if(value == nullptr){
if(args.size() == 2) vm->KeyError(args[1]);
if(args.size() == 3){
return args[2];
}
}
self.erase(args[1]);
return value;
});

18
tests/07_dict.py
View File

@ -125,4 +125,20 @@ for i in range(len(data)):
b.append(z)
if i % 3 == 0:
y = b.pop()
del a[y]
del a[y]
a = {1: 2, 3: 4}
assert a.pop(1) == 2
try:
a.pop(1)
exit(1)
except KeyError:
pass
assert a.pop(1, None) is None
try:
a.pop(1, 2, 3)
exit(1)
except TypeError:
pass