update brain_calc

Author: vdz

brain_games/games/brain_calc.py

@@ -8,18 +8,27 @@
 def calc():
     print('What is the result of the expression?')
     count = 0
+    def add(num1, num2):
+        return num1 + num2
+    
+    def sub(num1, num2):
+        return num1 - num2
+    
+    def mul(num1, num2):
+        return num1 * num2
+    
     operators = {
-            operator.add: '+',
-            operator.sub: '-',
-            operator.mul: '*',
-        }
+        '+': add,
+        '-': sub,
+        '*': mul,
+    }
 
     while count < 3:
-        rand_num1, rand_num2 = random.randint(1, 100), random.randint(1, 100)
+        num1, num2 = random.randint(1, 100), random.randint(1, 100)
         rand_operator = random.choice(list(operators))
         symbol = operators[rand_operator]
-        print(f'Question: {rand_num1} {symbol} {rand_num2}')
-        correct_answer = rand_operator(rand_num1, rand_num2)
+        print(f'Question: {num1} {rand_operator} {num2}')
+        correct_answer = symbol(num1, num2)
         user_answer = prompt.string('Your answer: ')
 
         if int(user_answer) == correct_answer: