3 // Copyright 2009 The Go Authors. All rights reserved.
4 // Use of this source code is governed by a BSD-style
5 // license that can be found in the LICENSE file.
7 // Test that heavy recursion works. Simple torture test for
8 // segmented stacks: do math in unary by recursion.
16 // -------------------------------------
23 func is_zero(x *Number) bool {
27 func add1(x *Number) *Number {
33 func sub1(x *Number) *Number {
37 func add(x, y *Number) *Number {
42 return add(add1(x), sub1(y))
45 func mul(x, y *Number) *Number {
46 if is_zero(x) || is_zero(y) {
50 return add(mul(x, sub1(y)), x)
53 func fact(n *Number) *Number {
58 return mul(fact(sub1(n)), n)
61 // -------------------------------------
62 // Helpers to generate/count Peano integers
64 func gen(n int) *Number {
66 return add1(gen(n - 1))
72 func count(x *Number) int {
77 return count(sub1(x)) + 1
80 func check(x *Number, expected int) {
83 print("error: found ", c, "; expected ", expected, "\n")
88 // -------------------------------------
89 // Test basic functionality
93 check(add1(zero()), 1)
96 check(add(gen(3), zero()), 3)
97 check(add(zero(), gen(4)), 4)
98 check(add(gen(3), gen(4)), 7)
100 check(mul(zero(), zero()), 0)
101 check(mul(gen(3), zero()), 0)
102 check(mul(zero(), gen(4)), 0)
103 check(mul(gen(3), add1(zero())), 3)
104 check(mul(add1(zero()), gen(4)), 4)
105 check(mul(gen(3), gen(4)), 12)
107 check(fact(zero()), 1)
108 check(fact(add1(zero())), 1)
109 check(fact(gen(5)), 120)
112 // -------------------------------------
115 var results = [...]int{
116 1, 1, 2, 6, 24, 120, 720, 5040, 40320, 362880, 3628800,
122 if runtime.GOARCH == "wasm" {
123 max = 7 // stack size is limited
125 for i := 0; i <= max; i++ {
126 if f := count(fact(gen(i))); f != results[i] {
127 println("FAIL:", i, "!:", f, "!=", results[i])