| """ |
| Tests for calculator tool (safe mathematical evaluation) |
| Author: @mangubee |
| Date: 2026-01-02 |
| |
| Tests cover: |
| - Basic arithmetic operations |
| - Mathematical functions |
| - Safety checks (no code execution, no imports, etc.) |
| - Timeout protection |
| - Complexity limits |
| - Error handling |
| """ |
|
|
| import pytest |
| from src.tools.calculator import safe_eval |
|
|
|
|
| |
| |
| |
|
|
| def test_addition(): |
| """Test basic addition""" |
| result = safe_eval("2 + 3") |
| assert result["result"] == 5 |
| assert result["success"] is True |
|
|
|
|
| def test_subtraction(): |
| """Test basic subtraction""" |
| result = safe_eval("10 - 4") |
| assert result["result"] == 6 |
|
|
|
|
| def test_multiplication(): |
| """Test basic multiplication""" |
| result = safe_eval("6 * 7") |
| assert result["result"] == 42 |
|
|
|
|
| def test_division(): |
| """Test basic division""" |
| result = safe_eval("15 / 3") |
| assert result["result"] == 5.0 |
|
|
|
|
| def test_floor_division(): |
| """Test floor division""" |
| result = safe_eval("17 // 5") |
| assert result["result"] == 3 |
|
|
|
|
| def test_modulo(): |
| """Test modulo operation""" |
| result = safe_eval("17 % 5") |
| assert result["result"] == 2 |
|
|
|
|
| def test_exponentiation(): |
| """Test exponentiation""" |
| result = safe_eval("2 ** 8") |
| assert result["result"] == 256 |
|
|
|
|
| def test_negative_numbers(): |
| """Test negative numbers""" |
| result = safe_eval("-5 + 3") |
| assert result["result"] == -2 |
|
|
|
|
| def test_complex_expression(): |
| """Test complex arithmetic expression""" |
| result = safe_eval("(2 + 3) * 4 - 10 / 2") |
| assert result["result"] == 15.0 |
|
|
|
|
| |
| |
| |
|
|
| def test_sqrt(): |
| """Test square root function""" |
| result = safe_eval("sqrt(16)") |
| assert result["result"] == 4.0 |
|
|
|
|
| def test_abs(): |
| """Test absolute value""" |
| result = safe_eval("abs(-42)") |
| assert result["result"] == 42 |
|
|
|
|
| def test_round(): |
| """Test rounding""" |
| result = safe_eval("round(3.7)") |
| assert result["result"] == 4 |
|
|
|
|
| def test_min(): |
| """Test min function""" |
| result = safe_eval("min(5, 2, 8, 1)") |
| assert result["result"] == 1 |
|
|
|
|
| def test_max(): |
| """Test max function""" |
| result = safe_eval("max(5, 2, 8, 1)") |
| assert result["result"] == 8 |
|
|
|
|
| def test_trigonometric(): |
| """Test trigonometric functions""" |
| result = safe_eval("sin(0)") |
| assert result["result"] == 0.0 |
|
|
| result = safe_eval("cos(0)") |
| assert result["result"] == 1.0 |
|
|
|
|
| def test_logarithm(): |
| """Test logarithmic functions""" |
| result = safe_eval("log10(100)") |
| assert result["result"] == 2.0 |
|
|
|
|
| def test_constants(): |
| """Test mathematical constants""" |
| result = safe_eval("pi") |
| assert abs(result["result"] - 3.14159) < 0.001 |
|
|
| result = safe_eval("e") |
| assert abs(result["result"] - 2.71828) < 0.001 |
|
|
|
|
| def test_factorial(): |
| """Test factorial function""" |
| result = safe_eval("factorial(5)") |
| assert result["result"] == 120 |
|
|
|
|
| def test_nested_functions(): |
| """Test nested function calls""" |
| result = safe_eval("sqrt(abs(-16))") |
| assert result["result"] == 4.0 |
|
|
|
|
| |
| |
| |
|
|
| def test_no_import(): |
| """Test that imports are blocked""" |
| with pytest.raises(SyntaxError): |
| safe_eval("import os") |
|
|
|
|
| def test_no_exec(): |
| """Test that exec is blocked""" |
| with pytest.raises((ValueError, SyntaxError)): |
| safe_eval("exec('print(1)')") |
|
|
|
|
| def test_no_eval(): |
| """Test that eval is blocked""" |
| with pytest.raises((ValueError, SyntaxError)): |
| safe_eval("eval('1+1')") |
|
|
|
|
| def test_no_lambda(): |
| """Test that lambda is blocked""" |
| with pytest.raises((ValueError, SyntaxError)): |
| safe_eval("lambda x: x + 1") |
|
|
|
|
| def test_no_attribute_access(): |
| """Test that attribute access is blocked""" |
| with pytest.raises(ValueError): |
| safe_eval("(1).__class__") |
|
|
|
|
| def test_no_list_comprehension(): |
| """Test that list comprehensions are blocked""" |
| with pytest.raises(ValueError): |
| safe_eval("[x for x in range(10)]") |
|
|
|
|
| def test_no_dict_access(): |
| """Test that dict operations are blocked""" |
| with pytest.raises((ValueError, SyntaxError)): |
| safe_eval("{'a': 1}") |
|
|
|
|
| def test_no_undefined_names(): |
| """Test that undefined variable names are blocked""" |
| with pytest.raises(ValueError, match="Undefined name"): |
| safe_eval("undefined_variable + 1") |
|
|
|
|
| def test_no_dangerous_functions(): |
| """Test that dangerous functions are blocked""" |
| with pytest.raises(ValueError, match="Unsupported function"): |
| safe_eval("open('file.txt')") |
|
|
|
|
| |
| |
| |
|
|
| def test_division_by_zero(): |
| """Test division by zero raises error""" |
| with pytest.raises(ZeroDivisionError): |
| safe_eval("10 / 0") |
|
|
|
|
| def test_invalid_syntax(): |
| """Test invalid syntax raises error""" |
| with pytest.raises(SyntaxError): |
| safe_eval("2 +* 3") |
|
|
|
|
| def test_empty_expression(): |
| """Test empty expression returns graceful error dict""" |
| result = safe_eval("") |
| assert result["success"] is False |
| assert "Empty expression" in result["error"] |
| assert result["result"] is None |
|
|
|
|
| def test_too_long_expression(): |
| """Test expression length limit returns graceful error dict""" |
| long_expr = "1 + " * 300 + "1" |
| result = safe_eval(long_expr) |
| assert result["success"] is False |
| assert "too long" in result["error"] |
| assert result["result"] is None |
|
|
|
|
| def test_huge_exponent(): |
| """Test that huge exponents are blocked""" |
| with pytest.raises(ValueError, match="Exponent too large"): |
| safe_eval("2 ** 10000") |
|
|
|
|
| def test_sqrt_negative(): |
| """Test sqrt of negative number raises error""" |
| with pytest.raises(ValueError): |
| safe_eval("sqrt(-1)") |
|
|
|
|
| def test_factorial_negative(): |
| """Test factorial of negative number raises error""" |
| with pytest.raises(ValueError): |
| safe_eval("factorial(-5)") |
|
|
|
|
| |
| |
| |
|
|
| def test_whitespace_handling(): |
| """Test that whitespace is handled correctly""" |
| result = safe_eval(" 2 + 3 ") |
| assert result["result"] == 5 |
|
|
|
|
| def test_floating_point(): |
| """Test floating point arithmetic""" |
| result = safe_eval("3.14 * 2") |
| assert abs(result["result"] - 6.28) < 0.01 |
|
|
|
|
| def test_very_small_numbers(): |
| """Test very small numbers""" |
| result = safe_eval("0.0001 + 0.0002") |
| assert abs(result["result"] - 0.0003) < 0.00001 |
|
|
|
|
| def test_scientific_notation(): |
| """Test scientific notation""" |
| result = safe_eval("1e3 + 2e2") |
| assert result["result"] == 1200.0 |
|
|
|
|
| def test_parentheses_precedence(): |
| """Test that parentheses affect precedence correctly""" |
| result1 = safe_eval("2 + 3 * 4") |
| assert result1["result"] == 14 |
|
|
| result2 = safe_eval("(2 + 3) * 4") |
| assert result2["result"] == 20 |
|
|
|
|
| def test_multiple_operations(): |
| """Test chaining multiple operations""" |
| result = safe_eval("10 + 20 - 5 * 2 / 2 + 3") |
| assert result["result"] == 28.0 |
|
|
