Split RepeatFive, handle counts, add examples

iteration/iteration_test.go

@@ -1,11 +1,36 @@
 package iteration
 
-import "testing"
+import (
+	"fmt"
+	"testing"
+)
 
 func TestRepeat(t *testing.T) {
-	got := Repeat("a")
-	want := "aaaaa"
-	assertResult(t, got, want)
+	t.Run("RepeatFive a", func(t *testing.T) {
+		got := RepeatFive("a")
+		want := "aaaaa"
+		assertResult(t, got, want)
+	})
+	t.Run("RepeatFive z", func(t *testing.T) {
+		got := RepeatFive("z")
+		want := "zzzzz"
+		assertResult(t, got, want)
+	})
+	t.Run("repeat z, 11 times", func(t *testing.T) {
+		got := Repeat("z", 11)
+		want := "zzzzzzzzzzz"
+		assertResult(t, got, want)
+	})
+}
+
+func ExampleRepeatFive() {
+	fmt.Println(RepeatFive("y"))
+	// Output: yyyyy
+}
+
+func ExampleRepeat() {
+	fmt.Println(Repeat("t", 7))
+	// Output: ttttttt
 }
 
 func assertResult(t testing.TB, got, want string) {
@@ -17,6 +42,6 @@ func assertResult(t testing.TB, got, want string) {
 
 func BenchmarkRepeat(b *testing.B) {
 	for b.Loop() {
-		Repeat("a")
+		RepeatFive("a")
 	}
 }