diff --git a/src/pythonCalculator.py b/src/pythonCalculator.py index d2790f9..3bf2467 100644 --- a/src/pythonCalculator.py +++ b/src/pythonCalculator.py @@ -9,19 +9,18 @@ class Operation(Enum): def basicCalculator(operation, num1, num2): result = 0 - match operation: - #if operation = ADD, return the sum - case Operation.ADD: - result = num1 + num2 - case Operation.SUBTRACT: - result = num1 - num2 - case Operation.MULTIPLY: - result = num1 * num2 - case Operation.DIVIDE: - if num2 == 0: - print("Can't divide by 0") - else: - result = num1 / num2 + #if operation = ADD, return the sum + if operation == Operation.ADD: + result = num1 + num2 + elif operation == Operation.SUBTRACT: + result = num1 - num2 + elif operation == Operation.MULTIPLY: + result = num1 * num2 + elif operation == Operation.DIVIDE: + if num2 == 0: + print("Can't divide by 0") + else: + result = num1 / num2 return result