diff --git a/calc.py b/calc.py index ccecba4..2c1ed15 100644 --- a/calc.py +++ b/calc.py @@ -14,9 +14,7 @@ def calc(text): expression = "" should_calc = False for char in text: - if char.isdigit() or (should_calc and char in [".", " "]): - expression += char - elif expression and char == ".": + if char.isdigit() or (expression and char == ".") or (should_calc and char == " "): expression += char elif char in math_operators: # calculation must include at least 1 operator