Commit 861d75e

@@ -8,8 +8,8 @@ module Mathy
       def choose_operation(console)
         verifier = Verification.new
         console.operation?([
-          Operations::Addition.new(verifier),
-          Operations::Subtraction.new(verifier)
+          Operations::Addition.new(verifier, self),
+          Operations::Subtraction.new(verifier, self)
         ])
       end
     end

@@ -1,8 +1,8 @@
 module Mathy
   module Operations
     class Addition < Operation
-      def initialize(verification)
-        super(verification, "+")
+      def initialize(verification, difficulty)
+        super(verification, difficulty, "+")
       end
 
       def calculate(operands)

@@ -3,12 +3,14 @@ module Mathy
     class Operation
       attr_reader :key
 
-      def initialize(verification, key)
+      def initialize(verification, difficulty, key)
         @verifier = verification
+        @difficulty = difficulty
         @key = key
       end
 
-      def play_turn(operands)
+      def play_turn
+        operands = @difficulty.next_operands
         @verifier.check_answer("#{operands.join(" #{key} ")} = ", calculate(operands))
       end
 

@@ -1,8 +1,8 @@
 module Mathy
   module Operations
     class Subtraction < Operation
-      def initialize(verification)
-        super(verification, "-")
+      def initialize(verification, difficulty)
+        super(verification, difficulty, "-")
       end
 
       def calculate(operands)

@@ -10,7 +10,7 @@ module Mathy
       difficulty = console.difficulty?
       operation = difficulty.choose_operation(console)
       games_to_play.times do
-        @score += 1 if operation.play_turn(difficulty.next_operands)
+        @score += 1 if operation.play_turn
       end
 
       display_results(@score, games_to_play)