| 123456789101112131415161718192021222324252627282930313233343536373839404142434445 |
- import pytest
- from joblib._utils import eval_expr
- @pytest.mark.parametrize(
- "expr",
- [
- "exec('import os')",
- "print(1)",
- "import os",
- "1+1; import os",
- "1^1",
- "' ' * 10**10",
- "9. ** 10000.",
- ],
- )
- def test_eval_expr_invalid(expr):
- with pytest.raises(ValueError, match="is not a valid or supported arithmetic"):
- eval_expr(expr)
- def test_eval_expr_too_long():
- expr = "1" + "+1" * 50
- with pytest.raises(ValueError, match="is too long"):
- eval_expr(expr)
- @pytest.mark.parametrize("expr", ["1e7", "10**7", "9**9**9"])
- def test_eval_expr_too_large_literal(expr):
- with pytest.raises(ValueError, match="Numeric literal .* is too large"):
- eval_expr(expr)
- @pytest.mark.parametrize(
- "expr, result",
- [
- ("2*6", 12),
- ("2**6", 64),
- ("1 + 2*3**(4) / (6 + -7)", -161.0),
- ("(20 // 3) % 5", 1),
- ],
- )
- def test_eval_expr_valid(expr, result):
- assert eval_expr(expr) == result
|
