diff options
| author | Robert Griesemer <gri@golang.org> | 2009-11-05 14:43:03 -0800 |
|---|---|---|
| committer | Robert Griesemer <gri@golang.org> | 2009-11-05 14:43:03 -0800 |
| commit | 91881826f7d90a6374b11300fea9fdf86b55a636 (patch) | |
| tree | feb68706550f2230acf4724be13ffcc5fa57cbe1 /src/pkg/exp/iterable/iterable_test.go | |
| parent | a78461e337f7ee26ee0b46e38444db7e2a177220 (diff) | |
| download | golang-91881826f7d90a6374b11300fea9fdf86b55a636.tar.gz | |
gofmt-ify 4s, iterable
R=rsc
http://go/go-review/1016055
Diffstat (limited to 'src/pkg/exp/iterable/iterable_test.go')
| -rw-r--r-- | src/pkg/exp/iterable/iterable_test.go | 89 |
1 files changed, 45 insertions, 44 deletions
diff --git a/src/pkg/exp/iterable/iterable_test.go b/src/pkg/exp/iterable/iterable_test.go index 28bdfb66f..bc66ff5dd 100644 --- a/src/pkg/exp/iterable/iterable_test.go +++ b/src/pkg/exp/iterable/iterable_test.go @@ -8,82 +8,83 @@ import ( "testing"; ) -type IntArray []int; +type IntArray []int -func (arr IntArray) Iter() <-chan interface {} { - ch := make(chan interface {}); +func (arr IntArray) Iter() <-chan interface{} { + ch := make(chan interface{}); go func() { for _, x := range arr { - ch <- x + ch <- x; } - close(ch) + close(ch); }(); - return ch + return ch; } -var oneToFive = IntArray{ 1, 2, 3, 4, 5 }; +var oneToFive = IntArray{1, 2, 3, 4, 5} -func isNegative(n interface {}) bool { - return n.(int) < 0 +func isNegative(n interface{}) bool { + return n.(int) < 0; } -func isPositive(n interface {}) bool { - return n.(int) > 0 +func isPositive(n interface{}) bool { + return n.(int) > 0; } -func isAbove3(n interface {}) bool { - return n.(int) > 3 +func isAbove3(n interface{}) bool { + return n.(int) > 3; } -func isEven(n interface {}) bool { - return n.(int) % 2 == 0 +func isEven(n interface{}) bool { + return n.(int) % 2 == 0; } -func doubler(n interface {}) interface {} { - return n.(int) * 2 +func doubler(n interface{}) interface{} { + return n.(int) * 2; } -func addOne(n interface {}) interface {} { - return n.(int) + 1 +func addOne(n interface{}) interface{} { + return n.(int) + 1; } -func adder(acc interface {}, n interface {}) interface {} { - return acc.(int) + n.(int) +func adder(acc interface{}, n interface{}) interface{} { + return acc.(int) + n.(int); } // A stream of the natural numbers: 0, 1, 2, 3, ... -type integerStream struct {} -func (i integerStream) Iter() <-chan interface {} { - ch := make(chan interface {}); - go func() { - for i := 0; ; i++ { - ch <- i - } - }(); - return ch +type integerStream struct{} + +func (i integerStream) Iter() <-chan interface{} { + ch := make(chan interface{}); + go func() { + for i := 0; ; i++ { + ch <- i; + } + }(); + return ch; } func TestAll(t *testing.T) { if !All(oneToFive, isPositive) { - t.Error("All(oneToFive, isPositive) == false") + t.Error("All(oneToFive, isPositive) == false"); } if All(oneToFive, isAbove3) { - t.Error("All(oneToFive, isAbove3) == true") + t.Error("All(oneToFive, isAbove3) == true"); } } func TestAny(t *testing.T) { if Any(oneToFive, isNegative) { - t.Error("Any(oneToFive, isNegative) == true") + t.Error("Any(oneToFive, isNegative) == true"); } if !Any(oneToFive, isEven) { - t.Error("Any(oneToFive, isEven) == false") + t.Error("Any(oneToFive, isEven) == false"); } } -func assertArraysAreEqual(t *testing.T, res []interface {}, expected []int) { +func assertArraysAreEqual(t *testing.T, res []interface{}, expected []int) { if len(res) != len(expected) { t.Errorf("len(res) = %v, want %v", len(res), len(expected)); - goto missing + goto missing; } for i := range res { if v := res[i].(int); v != expected[i] { t.Errorf("res[%v] = %v, want %v", i, v, expected[i]); - goto missing + goto missing; } } return; @@ -94,35 +95,35 @@ missing: func TestFilter(t *testing.T) { ints := integerStream{}; moreInts := Filter(ints, isAbove3).Iter(); - res := make([]interface {}, 3); + res := make([]interface{}, 3); for i := 0; i < 3; i++ { res[i] = <-moreInts; } - assertArraysAreEqual(t, res, []int{ 4, 5, 6 }) + assertArraysAreEqual(t, res, []int{4, 5, 6}); } func TestFind(t *testing.T) { ints := integerStream{}; first := Find(ints, isAbove3); if first.(int) != 4 { - t.Errorf("Find(ints, isAbove3) = %v, want 4", first) + t.Errorf("Find(ints, isAbove3) = %v, want 4", first); } } func TestInject(t *testing.T) { res := Inject(oneToFive, 0, adder); if res.(int) != 15 { - t.Errorf("Inject(oneToFive, 0, adder) = %v, want 15", res) + t.Errorf("Inject(oneToFive, 0, adder) = %v, want 15", res); } } func TestMap(t *testing.T) { res := Data(Map(Map(oneToFive, doubler), addOne)); - assertArraysAreEqual(t, res, []int{ 3, 5, 7, 9, 11 }) + assertArraysAreEqual(t, res, []int{3, 5, 7, 9, 11}); } func TestPartition(t *testing.T) { ti, fi := Partition(oneToFive, isEven); - assertArraysAreEqual(t, Data(ti), []int{ 2, 4 }); - assertArraysAreEqual(t, Data(fi), []int{ 1, 3, 5 }) + assertArraysAreEqual(t, Data(ti), []int{2, 4}); + assertArraysAreEqual(t, Data(fi), []int{1, 3, 5}); } |
