patchhawk/data/benign/misc_calc.py

def basic_calculator(a, b, op):
    """Perform a basic math operation."""
    if op == "+":
        return a + b
    elif op == "-":
        return a - b
    elif op == "*":
        return a * b
    elif op == "/":
        if b == 0:
            raise ValueError("Division by zero")
        return a / b
    return None