Upload folder using huggingface_hub (part 5)

de3bbe0 verified 3 months ago

1.02 kB

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

	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", "107", "99**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 + 23*(4) / (6 + -7)", -161.0),
	("(20 // 3) % 5", 1),
	],
	)
	def test_eval_expr_valid(expr, result):
	assert eval_expr(expr) == result