blob: 3368614e52074728c89c0e8e30660b0bddf2b581 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
# Tests of Starlark recursion and while statement.
# This is a "chunked" file: each "---" effectively starts a new file.
# option:recursion
load("assert.star", "assert")
def sum(n):
r = 0
while n > 0:
r += n
n -= 1
return r
def fib(n):
if n <= 1:
return 1
return fib(n-1) + fib(n-2)
def while_break(n):
r = 0
while n > 0:
if n == 5:
break
r += n
n -= 1
return r
def while_continue(n):
r = 0
while n > 0:
if n % 2 == 0:
n -= 1
continue
r += n
n -= 1
return r
assert.eq(fib(5), 8)
assert.eq(sum(5), 5+4+3+2+1)
assert.eq(while_break(10), 40)
assert.eq(while_continue(10), 25)
|