refactoring state test

hantmac · Apr 6, 2016 · 98efd46 · 98efd46
1 parent 2f4f567
commit 98efd46
Showing 1 changed file with 8 additions and 16 deletions.
diff --git a/16_state/state_test.go b/16_state/state_test.go
@@ -2,22 +2,14 @@ package state
 
 func ExampleWeek() {
 	ctx := NewDayContext()
-	ctx.Today()
-	ctx.Next()
-	ctx.Today()
-	ctx.Next()
-	ctx.Today()
-	ctx.Next()
-	ctx.Today()
-	ctx.Next()
-	ctx.Today()
-	ctx.Next()
-	ctx.Today()
-	ctx.Next()
-	ctx.Today()
-	ctx.Next()
-	ctx.Today()
-	ctx.Next()
+	todayAndNext := func() {
+		ctx.Today()
+		ctx.Next()
+	}
+
+	for i := 0; i < 8; i++ {
+		todayAndNext()
+	}
 	// Output:
 	// Sunday
 	// Monday