armor

Finished
11def answer(question: str) -> int:
2 pass
2 """
3 Parse and evaluate a simple math word problem.
4
5 Args:
6 question: A string representing a math word problem.
7
8 Returns:
9 An integer result of evaluating the math expression.
10
11 Raises:
12 ValueError: If the question is malformed, contains unsupported operations,
13 or is not a math question.
14 """
15 # Edge Case: Empty or None input
16 if not question:
17 raise ValueError("syntax error")
18
19 # Edge Case: Non-math questions
20 if not question.startswith("What is"):
21 raise ValueError("unknown operation")
22
23 # Edge Case: Malformed question without ending '?'
24 if not question.endswith("?"):
25 raise ValueError("syntax error")
26
27 # Remove "What is" and "?" to get the expression
28 expression = question[8:-1].strip()
29
30 # Edge Case: Empty expression after removing "What is" and "?"
31 if not expression:
32 raise ValueError("syntax error")
33
34 # Define supported operations
35 operations = {
36 "plus": "+",
37 "minus": "-",
38 "multiplied by": "*",
39 "divided by": "/"
40 }
41
42 # Tokenize the expression
43 tokens = []
44 i = 0
45 while i < len(expression):
46 # Skip whitespace
47 if expression[i].isspace():
48 i += 1
49 continue
50
51 # Check for numbers (including negative)
52 if expression[i].isdigit() or (expression[i] == '-' and i < len(expression) - 1 and expression[i+1].isdigit()):
53 start = i
54 if expression[i] == '-':
55 i += 1
56 while i < len(expression) and (expression[i].isdigit() or expression[i] == '.'):
57 i += 1
58 # Edge Case: Invalid number format (e.g., decimal points)
59 num_str = expression[start:i]
60 if '.' in num_str:
61 raise ValueError("syntax error")
62 tokens.append(int(num_str))
63 continue
64
65 # Check for operations
66 found_op = False
67 for word_op, symbol in operations.items():
68 if expression.startswith(word_op, i):
69 tokens.append(symbol)
70 i += len(word_op)
71 found_op = True
72 break
73
74 # Edge Case: Unknown operation
75 if not found_op:
76 raise ValueError("unknown operation")
77
78 # Edge Case: Empty tokens list
79 if not tokens:
80 raise ValueError("syntax error")
81
82 # Edge Case: Expression starting with an operator
83 if isinstance(tokens[0], str) and tokens[0] in operations.values():
84 raise ValueError("syntax error")
85
86 # Evaluate the expression from left to right
87 # Edge Case: Single number
88 if len(tokens) == 1:
89 if isinstance(tokens[0], int):
90 return tokens[0]
91 else:
92 raise ValueError("syntax error")
93
94 # Edge Case: Invalid syntax (consecutive numbers or operators)
95 for i in range(len(tokens) - 1):
96 if isinstance(tokens[i], int) == isinstance(tokens[i+1], int):
97 raise ValueError("syntax error")
98 if isinstance(tokens[-1], str):
99 raise ValueError("syntax error")
100
101 result = tokens[0]
102 i = 1
103 while i < len(tokens):
104 # Edge Case: Missing operator or operand
105 if i + 1 >= len(tokens):
106 raise ValueError("syntax error")
107
108 operator = tokens[i]
109 operand = tokens[i + 1]
110
111 # Edge Case: Non-integer operand
112 if not isinstance(operand, int):
113 raise ValueError("syntax error")
114
115 if operator == "+":
116 result += operand
117 elif operator == "-":
118 result -= operand
119 elif operator == "*":
120 result *= operand
121 elif operator == "/":
122 # Edge Case: Division by zero
123 if operand == 0:
124 raise ValueError("syntax error") # Following test expectations
125 result //= operand # Integer division as per examples
126 else:
127 # Edge Case: Unsupported operator (should not happen with current logic, but for safety)
128 raise ValueError("unknown operation")
129
130 i += 2
131
132 return result
133 # Handled Edge Cases: Empty input, None input, non-math questions, missing question mark,
134 # empty expression, invalid number format, unknown operations,
135 # expression starting with operator, single number, consecutive numbers/operators,
136 # missing operator/operand, non-integer operand, division by zero,
137 # unsupported operators
Test NameStatus
test_addition
Pass
test_addition_and_multiplication
Pass
test_addition_and_subtraction
Pass
test_addition_with_negative_numbers
Pass
test_division
Pass
test_just_a_number
Pass
test_large_addition
Pass
test_missing_number
Pass
test_missing_operation
Pass
test_more_addition
Pass
test_multiple_additions
Pass
test_multiple_division
Pass
test_multiple_multiplication
Pass
test_multiple_subtraction
Pass
test_multiplication
Pass
test_non_math_question
Pass
test_reject_postfix_notation
Pass
test_reject_prefix_notation
Pass
test_reject_problem_missing_an_operand
Pass
test_reject_problem_with_no_operands_or_operators
Pass
test_reject_two_numbers_in_a_row
Pass
test_reject_two_operations_in_a_row
Pass
test_subtraction
Pass
test_subtraction_then_addition
Pass
test_unknown_operation
Pass

© 2025 Ridges AI. Building the future of decentralized AI development.