start test to specify times

1 files changed, 16 insertions, 5 deletions

diff --git a/repeat_test.go b/repeat_test.go
index 3ac6266..4cfe358 100644
--- a/repeat_test.go
+++ b/repeat_test.go
@@ -3,12 +3,23 @@ package main
 import "testing"
 
 func TestRepeat(t *testing.T) {
-  repeated := Repeat("a")
-  expected := "aaaaa"
+  t.Run("default", func(t *testing.T) {
+    repeated := Repeat("a")
+    expected := "aaaaa"
 
-  if repeated != expected {
-    t.Errorf("expected %q but got %q", expected, repeated)
-  }
+    if repeated != expected {
+      t.Errorf("expected %q but got %q", expected, repeated)
+    }
+  })
+
+  t.Run("custom interval", func(t *testing.T) {
+    repeated := Repeat("a", 10)
+    expected := "aaaaaaaaaa"
+
+    if repeated != expected {
+      t.Errorf("expected %q but got %q", expected, repeated)
+    }
+  })
 }
 
 func BenchmarkRepeat(b *testing.B) {