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 the 'for range' construct.
11 // test range over channels
13 func gen(c
chan int, lo
, hi
int) {
14 for i
:= lo
; i
<= hi
; i
++ {
20 func seq(lo
, hi
int) chan int {
26 const alphabet
= "abcdefghijklmnopqrstuvwxyz"
28 func testblankvars() {
34 println("for range: wrong count", n
, "want 26")
38 for _
= range alphabet
{
42 println("for _ = range: wrong count", n
, "want 26")
46 for _
, _
= range alphabet
{
50 println("for _, _ = range: wrong count", n
, "want 26")
54 for i
, _
:= range alphabet
{
58 println("for i, _ := range: wrong sum", s
, "want 325")
62 for _
, v
:= range alphabet
{
66 println("for _, v := range: wrong sum", r
, "want 2847")
73 for i
:= range seq('a', 'z') {
77 println("Wanted lowercase alphabet; got", s
)
81 for range seq('a', 'z') {
85 println("testchan wrong count", n
, "want 26")
90 // test that range over slice only evaluates
91 // the expression after "range" once.
95 func makeslice() []int {
97 return []int{1, 2, 3, 4, 5}
103 for _
, v
:= range makeslice() {
107 println("range called makeslice", nmake
, "times")
111 println("wrong sum ranging over makeslice", s
)
118 for i
, x
[i
] = range y
{
121 if i
!= 0 || x
[0] != 10 || x
[1] != 99 {
122 println("wrong parallel assignment", i
, x
[0], x
[1])
130 for i
:= range makeslice() {
134 println("range called makeslice", nmake
, "times")
138 println("wrong sum ranging over makeslice", s
)
146 for range makeslice() {
150 println("range called makeslice", nmake
, "times")
154 println("wrong count ranging over makeslice", n
)
159 // test that range over []byte(string) only evaluates
160 // the expression after "range" once.
162 func makenumstring() string {
164 return "\x01\x02\x03\x04\x05"
170 for _
, v
:= range []byte(makenumstring()) {
174 println("range called makenumstring", nmake
, "times")
178 println("wrong sum ranging over []byte(makenumstring)", s
)
183 // test that range over array only evaluates
184 // the expression after "range" once.
186 func makearray() [5]int {
188 return [5]int{1, 2, 3, 4, 5}
194 for _
, v
:= range makearray() {
198 println("range called makearray", nmake
, "times")
202 println("wrong sum ranging over makearray", s
)
210 for i
:= range makearray() {
214 println("range called makearray", nmake
, "times")
218 println("wrong sum ranging over makearray", s
)
226 for range makearray() {
230 println("range called makearray", nmake
, "times")
234 println("wrong count ranging over makearray", n
)
239 func makearrayptr() *[5]int {
241 return &[5]int{1, 2, 3, 4, 5}
244 func testarrayptr() {
246 x
:= len(makearrayptr())
247 if x
!= 5 || nmake
!= 1 {
248 println("len called makearrayptr", nmake
, "times and got len", x
)
252 x
= cap(makearrayptr())
253 if x
!= 5 || nmake
!= 1 {
254 println("cap called makearrayptr", nmake
, "times and got len", x
)
259 for _
, v
:= range makearrayptr() {
263 println("range called makearrayptr", nmake
, "times")
267 println("wrong sum ranging over makearrayptr", s
)
272 func testarrayptr1() {
275 for i
:= range makearrayptr() {
279 println("range called makearrayptr", nmake
, "times")
283 println("wrong sum ranging over makearrayptr", s
)
288 func testarrayptr2() {
291 for range makearrayptr() {
295 println("range called makearrayptr", nmake
, "times")
299 println("wrong count ranging over makearrayptr", n
)
304 // test that range over string only evaluates
305 // the expression after "range" once.
307 func makestring() string {
315 for _
, v
:= range makestring() {
319 println("range called makestring", nmake
, "times")
322 if s
!= 'a'+'b'+'c'+'d'+'☺' {
323 println("wrong sum ranging over makestring", s
)
327 x
:= []rune
{'a', 'b'}
329 for i
, x
[i
] = range "c" {
332 if i
!= 0 || x
[0] != 'a' || x
[1] != 'c' {
333 println("wrong parallel assignment", i
, x
[0], x
[1])
339 for y
[r
], r
= range "\x02" {
342 if r
!= 2 || y
[0] != 1 || y
[1] != 0 || y
[2] != 3 {
343 println("wrong parallel assignment", r
, y
[0], y
[1], y
[2])
351 for i
:= range makestring() {
355 println("range called makestring", nmake
, "times")
359 println("wrong sum ranging over makestring", s
)
367 for range makestring() {
371 println("range called makestring", nmake
, "times")
375 println("wrong count ranging over makestring", n
)
380 // test that range over map only evaluates
381 // the expression after "range" once.
383 func makemap() map[int]int {
385 return map[int]int{0: 'a', 1: 'b', 2: 'c', 3: 'd', 4: '☺'}
391 for _
, v
:= range makemap() {
395 println("range called makemap", nmake
, "times")
398 if s
!= 'a'+'b'+'c'+'d'+'☺' {
399 println("wrong sum ranging over makemap", s
)
407 for i
:= range makemap() {
411 println("range called makemap", nmake
, "times")
415 println("wrong sum ranging over makemap", s
)
423 for range makemap() {
427 println("range called makemap", nmake
, "times")
431 println("wrong count ranging over makemap", n
)
436 // test that range evaluates the index and value expressions
437 // exactly once per iteration.
441 func getvar(p
*int) *int {
450 for *getvar(&i
), *getvar(&v
) = range [2]int{1, 2} {
455 println("wrong number of calls:", ncalls
, "!= 4")
458 if si
!= 1 || sv
!= 3 {
459 println("wrong sum in testcalls", si
, sv
)
464 for *getvar(&i
), *getvar(&v
) = range [0]int{} {
465 println("loop ran on empty array")
469 println("wrong number of calls:", ncalls
, "!= 0")