Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-11-10 13:40:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

allow next to be compatible with cpython

Browse Source
This commit is contained in:
blueloveTH 2024-04-14 00:07:30 +08:00
parent 5acb5260e3
commit 29d0a6a932
2 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/vm.cpp
View File

@ -729,7 +729,7 @@ void VM::init_builtin_types(){
this->Ellipsis = heap._new<Dummy>(_new_type_object("ellipsis"));
this->True = heap._new<Dummy>(tp_bool);
this->False = heap._new<Dummy>(tp_bool);
this->StopIteration = heap._new<Dummy>(_new_type_object("StopIterationType"));
this->StopIteration = _all_types[_new_type_object("StopIteration", tp_exception)].obj;
this->builtins = new_module("builtins");

18
tests/28_iter.py
View File

@ -36,3 +36,21 @@ assert next(i) == 4
assert next(i) == 5
assert next(i) == StopIteration
assert next(i) == StopIteration
import builtins
def next(obj):
res = builtins.next(obj)
if res is StopIteration:
raise StopIteration
return res
a = iter([1])
assert next(a) == 1
try:
next(a)
exit(1)
except StopIteration:
pass