rockstar

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 int: The 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 questions that don't end with '?'
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 operation mappings
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 (decimals not allowed per problem description)
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 if found_op:
75 continue
76
77 # If we reach here, we have an unknown operation or invalid character
78 # Edge Case: Unknown operations
79 raise ValueError("unknown operation")
80
81 # Edge Case: Empty tokens
82 if not tokens:
83 raise ValueError("syntax error")
84
85 # Edge Case: Expression starting with an operator
86 if isinstance(tokens[0], str) and tokens[0] in operations.values():
87 raise ValueError("syntax error")
88
89 # Evaluate the expression from left to right
90 # Edge Case: Single number
91 if len(tokens) == 1:
92 if isinstance(tokens[0], int):
93 return tokens[0]
94 else:
95 # Edge Case: Single token that is not a number
96 raise ValueError("syntax error")
97
98 # Edge Case: Even number of tokens (invalid syntax)
99 if len(tokens) % 2 == 0:
100 raise ValueError("syntax error")
101
102 result = tokens[0]
103
104 # Process operations in pairs
105 i = 1
106 while i < len(tokens):
107 # Edge Case: Missing operator
108 if i >= len(tokens) - 1:
109 raise ValueError("syntax error")
110
111 operator = tokens[i]
112 operand = tokens[i + 1]
113
114 # Edge Case: Non-integer operand
115 if not isinstance(operand, int):
116 raise ValueError("syntax error")
117
118 # Edge Case: Non-operator in operator position
119 if not isinstance(operator, str) or operator not in operations.values():
120 raise ValueError("syntax error")
121
122 if operator == "+":
123 result += operand
124 elif operator == "-":
125 result -= operand
126 elif operator == "*":
127 result *= operand
128 elif operator == "/":
129 # Edge Case: Division by zero
130 if operand == 0:
131 raise ValueError("syntax error")
132 result //= operand # Integer division as per examples
133
134 i += 2
135
136 return result
137 # Handled Edge Cases: Empty input, None input, non-math questions, malformed questions without '?',
138 # empty expressions, invalid number formats, unknown operations,
139 # expression starting with operator, single non-number token,
140 # even number of tokens, missing operators, non-integer operands,
141 # non-operators in operator positions, division by zero
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.