test/ddd.go

   1 // run
   2
   3 // Copyright 2010 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.
   6
   7 // Test variadic functions and calls (dot-dot-dot).
   8
   9 package main
  10
  11 func sum(args ...int) int {
  12         s := 0
  13         for _, v := range args {
  14                 s += v
  15         }
  16         return s
  17 }
  18
  19 func sumC(args ...int) int { return func() int { return sum(args...) }() }
  20
  21 var sumD = func(args ...int) int { return sum(args...) }
  22
  23 var sumE = func() func(...int) int { return func(args ...int) int { return sum(args...) } }()
  24
  25 var sumF = func(args ...int) func() int { return func() int { return sum(args...) } }
  26
  27 func sumA(args []int) int {
  28         s := 0
  29         for _, v := range args {
  30                 s += v
  31         }
  32         return s
  33 }
  34
  35 func sumB(args []int) int { return sum(args...) }
  36
  37 func sum2(args ...int) int { return 2 * sum(args...) }
  38
  39 func sum3(args ...int) int { return 3 * sumA(args) }
  40
  41 func sum4(args ...int) int { return 4 * sumB(args) }
  42
  43 func intersum(args ...interface{}) int {
  44         s := 0
  45         for _, v := range args {
  46                 s += v.(int)
  47         }
  48         return s
  49 }
  50
  51 type T []T
  52
  53 func ln(args ...T) int { return len(args) }
  54
  55 func ln2(args ...T) int { return 2 * ln(args...) }
  56
  57 func (*T) Sum(args ...int) int { return sum(args...) }
  58
  59 type U struct {
  60         *T
  61 }
  62
  63 type I interface {
  64         Sum(...int) int
  65 }
  66
  67 func main() {
  68         if x := sum(1, 2, 3); x != 6 {
  69                 println("sum 6", x)
  70                 panic("fail")
  71         }
  72         if x := sum(); x != 0 {
  73                 println("sum 0", x)
  74                 panic("fail")
  75         }
  76         if x := sum(10); x != 10 {
  77                 println("sum 10", x)
  78                 panic("fail")
  79         }
  80         if x := sum(1, 8); x != 9 {
  81                 println("sum 9", x)
  82                 panic("fail")
  83         }
  84         if x := sumC(4, 5, 6); x != 15 {
  85                 println("sumC 15", x)
  86                 panic("fail")
  87         }
  88         if x := sumD(4, 5, 7); x != 16 {
  89                 println("sumD 16", x)
  90                 panic("fail")
  91         }
  92         if x := sumE(4, 5, 8); x != 17 {
  93                 println("sumE 17", x)
  94                 panic("fail")
  95         }
  96         if x := sumF(4, 5, 9)(); x != 18 {
  97                 println("sumF 18", x)
  98                 panic("fail")
  99         }
 100         if x := sum2(1, 2, 3); x != 2*6 {
 101                 println("sum 6", x)
 102                 panic("fail")
 103         }
 104         if x := sum2(); x != 2*0 {
 105                 println("sum 0", x)
 106                 panic("fail")
 107         }
 108         if x := sum2(10); x != 2*10 {
 109                 println("sum 10", x)
 110                 panic("fail")
 111         }
 112         if x := sum2(1, 8); x != 2*9 {
 113                 println("sum 9", x)
 114                 panic("fail")
 115         }
 116         if x := sum3(1, 2, 3); x != 3*6 {
 117                 println("sum 6", x)
 118                 panic("fail")
 119         }
 120         if x := sum3(); x != 3*0 {
 121                 println("sum 0", x)
 122                 panic("fail")
 123         }
 124         if x := sum3(10); x != 3*10 {
 125                 println("sum 10", x)
 126                 panic("fail")
 127         }
 128         if x := sum3(1, 8); x != 3*9 {
 129                 println("sum 9", x)
 130                 panic("fail")
 131         }
 132         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         }
 148         if x := intersum(1, 2, 3); x != 6 {
 149                 println("intersum 6", x)
 150                 panic("fail")
 151         }
 152         if x := intersum(); x != 0 {
 153                 println("intersum 0", x)
 154                 panic("fail")
 155         }
 156         if x := intersum(10); x != 10 {
 157                 println("intersum 10", x)
 158                 panic("fail")
 159         }
 160         if x := intersum(1, 8); x != 9 {
 161                 println("intersum 9", x)
 162                 panic("fail")
 163         }
 164
 165         if x := ln(nil, nil, nil); x != 3 {
 166                 println("ln 3", x)
 167                 panic("fail")
 168         }
 169         if x := ln([]T{}); x != 1 {
 170                 println("ln 1", x)
 171                 panic("fail")
 172         }
 173         if x := ln2(nil, nil, nil); x != 2*3 {
 174                 println("ln2 3", x)
 175                 panic("fail")
 176         }
 177         if x := ln2([]T{}); x != 2*1 {
 178                 println("ln2 1", x)
 179                 panic("fail")
 180         }
 181         if x := ((*T)(nil)).Sum(1, 3, 5, 7); x != 16 {
 182                 println("(*T)(nil).Sum", x)
 183                 panic("fail")
 184         }
 185         if x := (*T).Sum(nil, 1, 3, 5, 6); x != 15 {
 186                 println("(*T).Sum", x)
 187                 panic("fail")
 188         }
 189         if x := (&U{}).Sum(1, 3, 5, 5); x != 14 {
 190                 println("(&U{}).Sum", x)
 191                 panic("fail")
 192         }
 193         var u U
 194         if x := u.Sum(1, 3, 5, 4); x != 13 {
 195                 println("u.Sum", x)
 196                 panic("fail")
 197         }
 198         if x := (&u).Sum(1, 3, 5, 3); x != 12 {
 199                 println("(&u).Sum", x)
 200                 panic("fail")
 201         }
 202         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")
 208         }
 209         i = u
 210         if x := i.Sum(2, 3, 5, 6); x != 16 {
 211                 println("i(=u).Sum", x)
 212                 panic("fail")
 213         }
 214         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         }
 222         if x := (*U).Sum(&U{}, 1, 3, 5, 2); x != 11 {
 223                 println("(*U).Sum", x)
 224                 panic("fail")
 225         }
 226         if x := U.Sum(U{}, 1, 3, 5, 1); x != 10 {
 227                 println("U.Sum", x)
 228                 panic("fail")
 229         }
 230 }