blob: 84503f7d5b182751a825ff42dc97e1cb01583e74 [file] [log] [blame]
Russ Cox0b477ef2012-02-16 23:48:57 -05001// run
Russ Cox68796b02010-02-01 00:25:59 -08002
Emmanuel Odeke53fd5222016-04-10 14:32:26 -07003// Copyright 2010 The Go Authors. All rights reserved.
Russ Cox68796b02010-02-01 00:25:59 -08004// Use of this source code is governed by a BSD-style
5// license that can be found in the LICENSE file.
6
Rob Pike83976e32012-02-19 14:28:53 +11007// Test variadic functions and calls (dot-dot-dot).
8
Russ Cox68796b02010-02-01 00:25:59 -08009package main
10
11func sum(args ...int) int {
12 s := 0
13 for _, v := range args {
14 s += v
15 }
16 return s
17}
18
Russ Cox2ee420f2010-09-24 11:55:48 -040019func sumC(args ...int) int { return func() int { return sum(args...) }() }
Russ Coxbea730d2010-02-01 10:49:24 -080020
Russ Cox2ee420f2010-09-24 11:55:48 -040021var sumD = func(args ...int) int { return sum(args...) }
Russ Coxbea730d2010-02-01 10:49:24 -080022
Russ Cox2ee420f2010-09-24 11:55:48 -040023var sumE = func() func(...int) int { return func(args ...int) int { return sum(args...) } }()
Russ Coxbea730d2010-02-01 10:49:24 -080024
Russ Cox2ee420f2010-09-24 11:55:48 -040025var sumF = func(args ...int) func() int { return func() int { return sum(args...) } }
Russ Coxbea730d2010-02-01 10:49:24 -080026
Russ Cox68796b02010-02-01 00:25:59 -080027func sumA(args []int) int {
28 s := 0
29 for _, v := range args {
30 s += v
31 }
32 return s
33}
34
Russ Cox01385b42010-10-06 11:12:30 -040035func sumB(args []int) int { return sum(args...) }
36
Russ Cox2ee420f2010-09-24 11:55:48 -040037func sum2(args ...int) int { return 2 * sum(args...) }
Russ Cox68796b02010-02-01 00:25:59 -080038
39func sum3(args ...int) int { return 3 * sumA(args) }
40
Russ Cox01385b42010-10-06 11:12:30 -040041func sum4(args ...int) int { return 4 * sumB(args) }
42
Russ Cox68796b02010-02-01 00:25:59 -080043func intersum(args ...interface{}) int {
44 s := 0
45 for _, v := range args {
46 s += v.(int)
47 }
48 return s
49}
50
51type T []T
52
53func ln(args ...T) int { return len(args) }
54
Russ Cox2ee420f2010-09-24 11:55:48 -040055func ln2(args ...T) int { return 2 * ln(args...) }
Russ Cox68796b02010-02-01 00:25:59 -080056
Russ Cox2ee420f2010-09-24 11:55:48 -040057func (*T) Sum(args ...int) int { return sum(args...) }
Russ Coxbea730d2010-02-01 10:49:24 -080058
59type U struct {
60 *T
61}
62
Ian Lance Taylore54ad642012-03-07 08:24:10 -080063type I interface {
64 Sum(...int) int
65}
66
Russ Cox68796b02010-02-01 00:25:59 -080067func main() {
68 if x := sum(1, 2, 3); x != 6 {
Rob Pike325cf8e2010-03-24 16:46:53 -070069 println("sum 6", x)
70 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -080071 }
72 if x := sum(); x != 0 {
Rob Pike325cf8e2010-03-24 16:46:53 -070073 println("sum 0", x)
74 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -080075 }
76 if x := sum(10); x != 10 {
Rob Pike325cf8e2010-03-24 16:46:53 -070077 println("sum 10", x)
78 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -080079 }
80 if x := sum(1, 8); x != 9 {
Rob Pike325cf8e2010-03-24 16:46:53 -070081 println("sum 9", x)
82 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -080083 }
Russ Coxbea730d2010-02-01 10:49:24 -080084 if x := sumC(4, 5, 6); x != 15 {
Rob Pike325cf8e2010-03-24 16:46:53 -070085 println("sumC 15", x)
86 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -080087 }
Russ Coxbea730d2010-02-01 10:49:24 -080088 if x := sumD(4, 5, 7); x != 16 {
Rob Pike325cf8e2010-03-24 16:46:53 -070089 println("sumD 16", x)
90 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -080091 }
92 if x := sumE(4, 5, 8); x != 17 {
Rob Pike325cf8e2010-03-24 16:46:53 -070093 println("sumE 17", x)
94 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -080095 }
96 if x := sumF(4, 5, 9)(); x != 18 {
Rob Pike325cf8e2010-03-24 16:46:53 -070097 println("sumF 18", x)
98 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -080099 }
Russ Cox68796b02010-02-01 00:25:59 -0800100 if x := sum2(1, 2, 3); x != 2*6 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700101 println("sum 6", x)
102 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800103 }
104 if x := sum2(); x != 2*0 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700105 println("sum 0", x)
106 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800107 }
108 if x := sum2(10); x != 2*10 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700109 println("sum 10", x)
110 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800111 }
112 if x := sum2(1, 8); x != 2*9 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700113 println("sum 9", x)
114 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800115 }
116 if x := sum3(1, 2, 3); x != 3*6 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700117 println("sum 6", x)
118 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800119 }
120 if x := sum3(); x != 3*0 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700121 println("sum 0", x)
122 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800123 }
124 if x := sum3(10); x != 3*10 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700125 println("sum 10", x)
126 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800127 }
128 if x := sum3(1, 8); x != 3*9 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700129 println("sum 9", x)
130 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800131 }
Russ Cox01385b42010-10-06 11:12:30 -0400132 if x := sum4(1, 2, 3); x != 4*6 {
133 println("sum 6", x)
134 panic("fail")
135 }
136 if x := sum4(); x != 4*0 {
137 println("sum 0", x)
138 panic("fail")
139 }
140 if x := sum4(10); x != 4*10 {
141 println("sum 10", x)
142 panic("fail")
143 }
144 if x := sum4(1, 8); x != 4*9 {
145 println("sum 9", x)
146 panic("fail")
147 }
Russ Cox68796b02010-02-01 00:25:59 -0800148 if x := intersum(1, 2, 3); x != 6 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700149 println("intersum 6", x)
150 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800151 }
152 if x := intersum(); x != 0 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700153 println("intersum 0", x)
154 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800155 }
156 if x := intersum(10); x != 10 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700157 println("intersum 10", x)
158 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800159 }
160 if x := intersum(1, 8); x != 9 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700161 println("intersum 9", x)
162 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800163 }
164
165 if x := ln(nil, nil, nil); x != 3 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700166 println("ln 3", x)
167 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800168 }
169 if x := ln([]T{}); x != 1 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700170 println("ln 1", x)
171 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800172 }
173 if x := ln2(nil, nil, nil); x != 2*3 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700174 println("ln2 3", x)
175 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800176 }
177 if x := ln2([]T{}); x != 2*1 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700178 println("ln2 1", x)
179 panic("fail")
Russ Cox68796b02010-02-01 00:25:59 -0800180 }
Rob Pike325cf8e2010-03-24 16:46:53 -0700181 if x := ((*T)(nil)).Sum(1, 3, 5, 7); x != 16 {
182 println("(*T)(nil).Sum", x)
183 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -0800184 }
185 if x := (*T).Sum(nil, 1, 3, 5, 6); x != 15 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700186 println("(*T).Sum", x)
187 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -0800188 }
Rob Pike325cf8e2010-03-24 16:46:53 -0700189 if x := (&U{}).Sum(1, 3, 5, 5); x != 14 {
190 println("(&U{}).Sum", x)
191 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -0800192 }
193 var u U
Rob Pike325cf8e2010-03-24 16:46:53 -0700194 if x := u.Sum(1, 3, 5, 4); x != 13 {
195 println("u.Sum", x)
196 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -0800197 }
Rob Pike325cf8e2010-03-24 16:46:53 -0700198 if x := (&u).Sum(1, 3, 5, 3); x != 12 {
199 println("(&u).Sum", x)
200 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -0800201 }
Rob Pike325cf8e2010-03-24 16:46:53 -0700202 var i interface {
203 Sum(...int) int
204 } = &u
205 if x := i.Sum(2, 3, 5, 7); x != 17 {
206 println("i(=&u).Sum", x)
207 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -0800208 }
209 i = u
Rob Pike325cf8e2010-03-24 16:46:53 -0700210 if x := i.Sum(2, 3, 5, 6); x != 16 {
211 println("i(=u).Sum", x)
212 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -0800213 }
Ian Lance Taylore54ad642012-03-07 08:24:10 -0800214 var s struct {
215 I
216 }
217 s.I = &u
218 if x := s.Sum(2, 3, 5, 8); x != 18 {
219 println("s{&u}.Sum", x)
220 panic("fail")
221 }
Russ Coxbea730d2010-02-01 10:49:24 -0800222 if x := (*U).Sum(&U{}, 1, 3, 5, 2); x != 11 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700223 println("(*U).Sum", x)
224 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -0800225 }
226 if x := U.Sum(U{}, 1, 3, 5, 1); x != 10 {
Rob Pike325cf8e2010-03-24 16:46:53 -0700227 println("U.Sum", x)
228 panic("fail")
Russ Coxbea730d2010-02-01 10:49:24 -0800229 }
Russ Cox68796b02010-02-01 00:25:59 -0800230}