瀏覽代碼

Merge branch 'main' of https://github.com/blueloveTH/pocketpy

blueloveTH 3 年之前
父節點
當前提交
b8f7491cd7
共有 2 個文件被更改,包括 9 次插入0 次删除
  1. 6 0
      src/builtins.h
  2. 3 0
      tests/_builtin_ty.py

+ 6 - 0
src/builtins.h

@@ -44,6 +44,12 @@ def any(iterable):
             return True
     return False
 
+def enumerate(iterable, start=0):
+    n = start
+    for elem in iterable:
+        yield n, elem
+        n += 1
+
 def sum(iterable):
     res = 0
     for i in iterable:

+ 3 - 0
tests/_builtin_ty.py

@@ -199,3 +199,6 @@ assert all([True, 1])
 assert not all([False])
 assert not all([True, False])
 assert not all([False, False])
+
+assert list(enumerate([1,2,3])) == [(0,1), (1,2), (2,3)]
+assert list(enumerate([1,2,3], 1)) == [(1,1), (2,2), (3,3)]