forked from Rockachopa/Timmy-time-dashboard
4e11dd2490
Move 97 test files from flat tests/ into 13 subdirectories: tests/dashboard/ (8 files — routes, mobile, mission control) tests/swarm/ (17 files — coordinator, docker, routing, tasks) tests/timmy/ (12 files — agent, backends, CLI, tools) tests/self_coding/ (14 files — git safety, indexer, self-modify) tests/lightning/ (3 files — L402, LND, interface) tests/creative/ (8 files — assembler, director, image/music/video) tests/integrations/ (10 files — chat bridge, telegram, voice, websocket) tests/mcp/ (4 files — bootstrap, discovery, executor) tests/spark/ (3 files — engine, tools, events) tests/hands/ (3 files — registry, oracle, phase5) tests/scripture/ (1 file) tests/infrastructure/ (3 files — router cascade, API) tests/security/ (3 files — XSS, regression) Fix Path(__file__) reference in test_mobile_scenarios.py for new depth. Add __init__.py to all test subdirectories. Tests: 1503 passed, 9 failed (pre-existing), 53 errors (pre-existing) https://claude.ai/code/session_019oMFNvD8uSGSSmBMGkBfQN
86 lines
1.9 KiB
Python
86 lines
1.9 KiB
Python
"""Tests for the calculator tool."""
|
|
|
|
from timmy.tools import calculator
|
|
|
|
|
|
def test_basic_multiplication():
|
|
assert calculator("347 * 829") == "287663"
|
|
|
|
|
|
def test_basic_addition():
|
|
assert calculator("100 + 200") == "300"
|
|
|
|
|
|
def test_basic_division():
|
|
assert calculator("100 / 4") == "25.0"
|
|
|
|
|
|
def test_integer_division():
|
|
assert calculator("100 // 3") == "33"
|
|
|
|
|
|
def test_exponentiation():
|
|
assert calculator("2 ** 10") == "1024"
|
|
|
|
|
|
def test_sqrt():
|
|
assert calculator("math.sqrt(17161)") == "131.0"
|
|
|
|
|
|
def test_sqrt_non_perfect():
|
|
result = float(calculator("math.sqrt(2)"))
|
|
assert abs(result - 1.4142135623730951) < 1e-10
|
|
|
|
|
|
def test_log_base_10():
|
|
result = float(calculator("math.log10(1000)"))
|
|
assert abs(result - 3.0) < 1e-10
|
|
|
|
|
|
def test_log_natural():
|
|
result = float(calculator("math.log(math.e)"))
|
|
assert abs(result - 1.0) < 1e-10
|
|
|
|
|
|
def test_trig_sin():
|
|
result = float(calculator("math.sin(math.pi / 2)"))
|
|
assert abs(result - 1.0) < 1e-10
|
|
|
|
|
|
def test_abs_builtin():
|
|
assert calculator("abs(-42)") == "42"
|
|
|
|
|
|
def test_round_builtin():
|
|
assert calculator("round(3.14159, 2)") == "3.14"
|
|
|
|
|
|
def test_min_max_builtins():
|
|
assert calculator("min(3, 7, 1)") == "1"
|
|
assert calculator("max(3, 7, 1)") == "7"
|
|
|
|
|
|
def test_complex_expression():
|
|
assert calculator("(347 * 829) + (100 / 4)") == "287688.0"
|
|
|
|
|
|
def test_invalid_expression_returns_error():
|
|
result = calculator("not a valid expression")
|
|
assert result.startswith("Error evaluating")
|
|
|
|
|
|
def test_no_builtins_access():
|
|
"""Ensure dangerous builtins like __import__ are blocked."""
|
|
result = calculator("__import__('os').system('echo pwned')")
|
|
assert result.startswith("Error evaluating")
|
|
|
|
|
|
def test_no_open_access():
|
|
result = calculator("open('/etc/passwd').read()")
|
|
assert result.startswith("Error evaluating")
|
|
|
|
|
|
def test_division_by_zero():
|
|
result = calculator("1 / 0")
|
|
assert result.startswith("Error evaluating")
|