diff options
| author | David G. Andersen <dave.andersen@gmail.com> | 2009-11-16 12:40:01 -0800 |
|---|---|---|
| committer | Russ Cox <rsc@golang.org> | 2009-11-16 12:40:01 -0800 |
| commit | 37f71e8ad644c0e91bb4449882f60b95c7d4644a (patch) | |
| tree | 105a92366d0c973cfc3c37374606eb2c8fcbefeb /src/pkg/bytes/bytes_test.go | |
| parent | 11c1aa9f6d5d73ea75ad01eba054cee47dc7c621 (diff) | |
| download | go-37f71e8ad644c0e91bb4449882f60b95c7d4644a.tar.xz | |
An asked-for-in #go-nuts extension to quickly create a repeated
copy of a string or a byte array.
strings.Repeat("-", 50)
bytes.Repeat(b, 99)
R=rsc
https://golang.org/cl/155063
Diffstat (limited to 'src/pkg/bytes/bytes_test.go')
| -rw-r--r-- | src/pkg/bytes/bytes_test.go | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/src/pkg/bytes/bytes_test.go b/src/pkg/bytes/bytes_test.go index dddaf5064e..3e737cb376 100644 --- a/src/pkg/bytes/bytes_test.go +++ b/src/pkg/bytes/bytes_test.go @@ -361,3 +361,30 @@ func TestAddByte(t *testing.T) { } } } + +type RepeatTest struct { + in, out string; + count int; +} + +var RepeatTests = []RepeatTest{ + RepeatTest{"", "", 0}, + RepeatTest{"", "", 1}, + RepeatTest{"", "", 2}, + RepeatTest{"-", "", 0}, + RepeatTest{"-", "-", 1}, + RepeatTest{"-", "----------", 10}, + RepeatTest{"abc ", "abc abc abc ", 3}, +} + +func TestRepeat(t *testing.T) { + for _, tt := range RepeatTests { + tin := strings.Bytes(tt.in); + tout := strings.Bytes(tt.out); + a := Repeat(tin, tt.count); + if !Equal(a, tout) { + t.Errorf("Repeat(%q, %d) = %q; want %q", tin, tt.count, a, tout); + continue; + } + } +} |
