Loop through rows to find winning player

stride-nyc · Apr 2, 2018 · 1dd3398 · 1dd3398
1 parent 91764b5
commit 1dd3398
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/tic-tac-toe/tic_tac_toe.rb b/tic-tac-toe/tic_tac_toe.rb
@@ -26,12 +26,8 @@ def winning_move?(square, player)
   end
 
   def winner
-    square = 0
-    return winning_player(square) if winning_player(square)
-    square = 3
-    return winning_player(square) if winning_player(square)
-    square = 6
-    return winning_player(square) if winning_player(square)
+    rows = [0, 3, 6]
+    rows.each { |row|  return winning_player(row) if winning_player(row) }
     return '-'
   end