Add unit tests

Author: githejie

pyproject.toml

@@ -1,12 +1,13 @@
 [project]
 name = "mcp-server-calculator"
-version = "0.1.0"
+version = "0.1.1"
 description = "A Model Context Protocol server for calculating"
 readme = "README.md"
 requires-python = ">=3.10"
 authors = [{ name = "He Jie", email = "[email protected]" }]
 keywords = ["mcp", "llm", "math", "calculator"]
 license = { text = "MIT" }
+urls = { Source = "https://github.com/githejie/mcp-server-calculator" }
 classifiers = [
     "Development Status :: 4 - Beta",
     "Intended Audience :: Developers",

src/mcp_server_calculator/calculator.py

@@ -24,7 +24,7 @@ def eval_expr(node):
                 return allowed_operators[type(node.op)](left, right)
         elif isinstance(node, ast.UnaryOp) and isinstance(node.op, ast.USub):
             return -eval_expr(node.operand)
-        raise ValueError("Unsupported operation")
+        raise ValueError(f"Unsupported operation: {ast.dump(node)}")
 
     parsed_expr = ast.parse(expression, mode='eval')
     result = eval_expr(parsed_expr.body)

tests/test_calculator.py

@@ -0,0 +1,80 @@
+import unittest
+from src.mcp_server_calculator.calculator import evaluate
+
+class TestCalculator(unittest.TestCase):
+    def test_addition(self):
+        self.assertEqual(evaluate("1 + 1"), "2")
+
+    def test_subtraction(self):
+        self.assertEqual(evaluate("5 - 3"), "2")
+
+    def test_multiplication(self):
+        self.assertEqual(evaluate("2 * 3"), "6")
+
+    def test_division(self):
+        self.assertEqual(evaluate("8 / 2"), "4.0")
+
+    def test_floor_division(self):
+        self.assertEqual(evaluate("7 // 2"), "3")
+
+    def test_modulus(self):
+        self.assertEqual(evaluate("10 % 3"), "1")
+
+    def test_power(self):
+        self.assertEqual(evaluate("2 ** 3"), "8")
+
+    def test_unary_minus(self):
+        self.assertEqual(evaluate("-5"), "-5")
+
+    def test_complex_expression(self):
+        self.assertEqual(evaluate("2 + 3 * (4 - 1) / 2 ** 2"), "4.25")
+
+    def test_parentheses_expression(self):
+        self.assertEqual(evaluate("(2 + 3) * 4"), "20")
+
+    def test_negative_numbers(self):
+        self.assertEqual(evaluate("-2 + 3"), "1")
+        self.assertEqual(evaluate("4 * -2"), "-8")
+        self.assertEqual(evaluate("-6 / 2"), "-3.0")
+
+    def test_floating_point_operations(self):
+        self.assertEqual(evaluate("0.5 + 0.25"), "0.75")
+        self.assertEqual(evaluate("2.5 * 2"), "5.0")
+        self.assertEqual(evaluate("5.0 / 2"), "2.5")
+
+    def test_large_numbers(self):
+        self.assertEqual(evaluate("123456789 * 987654321"), str(123456789 * 987654321))
+
+    def test_floating_point_precision(self):
+        self.assertAlmostEqual(float(evaluate("0.1 + 0.2")), 0.3, places=7)
+
+    def test_unsupported_operation(self):
+        with self.assertRaises(ValueError):
+            evaluate("unknown")
+
+    def test_empty_string(self):
+        with self.assertRaises(SyntaxError):
+            evaluate("")
+
+    def test_whitespace_string(self):
+        with self.assertRaises(SyntaxError):
+            evaluate("   ")
+
+    def test_invalid_expression(self):
+        with self.assertRaises(SyntaxError):
+            evaluate("2 +")
+
+    def test_division_by_zero(self):
+        with self.assertRaises(ZeroDivisionError):
+            evaluate("1 / 0")
+
+    def test_floor_division_by_zero(self):
+        with self.assertRaises(ZeroDivisionError):
+            evaluate("1 // 0")
+
+    def test_modulus_by_zero(self):
+        with self.assertRaises(ZeroDivisionError):
+            evaluate("1 % 0")
+
+if __name__ == '__main__':
+    unittest.main()