Add times parameter

2 files changed, 4 insertions, 4 deletions

diff --git a/repeat.go b/repeat.go
index ed390f4..c7a1468 100644
--- a/repeat.go
+++ b/repeat.go
@@ -1,8 +1,8 @@
 package main
 
-func Repeat(character string) string {
+func Repeat(character string, times int) string {
   var repeated string
-  for i := 0; i < 5; i++ {
+  for i := 0; i < times; i++ {
     repeated += character
   }
   return repeated
diff --git a/repeat_test.go b/repeat_test.go
index 4cfe358..ccb8d12 100644
--- a/repeat_test.go
+++ b/repeat_test.go
@@ -4,7 +4,7 @@ import "testing"
 
 func TestRepeat(t *testing.T) {
   t.Run("default", func(t *testing.T) {
-    repeated := Repeat("a")
+    repeated := Repeat("a", 5)
     expected := "aaaaa"
 
     if repeated != expected {
@@ -24,6 +24,6 @@ func TestRepeat(t *testing.T) {
 
 func BenchmarkRepeat(b *testing.B) {
   for i := 0; i < b.N; i++ {
-    Repeat("a")
+    Repeat("a", 5)
   }
 }