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