aboutsummaryrefslogtreecommitdiff
path: root/src/pkg/bytes/bytes_test.go
diff options
context:
space:
mode:
authorDavid G. Andersen <dave.andersen@gmail.com>2009-11-16 12:40:01 -0800
committerRuss Cox <rsc@golang.org>2009-11-16 12:40:01 -0800
commit37f71e8ad644c0e91bb4449882f60b95c7d4644a (patch)
tree105a92366d0c973cfc3c37374606eb2c8fcbefeb /src/pkg/bytes/bytes_test.go
parent11c1aa9f6d5d73ea75ad01eba054cee47dc7c621 (diff)
downloadgo-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.go27
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;
+ }
+ }
+}