Function test_eval_extended_builtins

xarray/tests/test_eval.py:196–256

Test extended builtins available in eval namespace. These builtins are safe (no I/O, no code execution) and commonly needed for typical xarray operations like slicing, type conversion, and iteration.

()

def test_eval_extended_builtins() -> None:
    """Test extended builtins available in eval namespace.

    These builtins are safe (no I/O, no code execution) and commonly needed
    for typical xarray operations like slicing, type conversion, and iteration.
    """
    ds = Dataset(
        {"a": ("x", [1.0, 2.0, 3.0, 4.0, 5.0])},
        coords={"time": pd.date_range("2019-01-01", periods=5)},
    )

    # slice - essential for .sel() with ranges
    result = ds.eval("a.sel(x=slice(1, 3))")
    expected = ds["a"].sel(x=slice(1, 3))
    assert_equal(result, expected)

    # str - type constructor
    result = ds.eval("str(int(a.mean()))")
    assert result == "3"

    # list, tuple - type constructors
    result = ds.eval("list(range(3))")
    assert result == [0, 1, 2]

    result = ds.eval("tuple(range(3))")
    assert result == (0, 1, 2)

    # dict, set - type constructors
    result = ds.eval("dict(x=1, y=2)")
    assert result == {"x": 1, "y": 2}

    result = ds.eval("set([1, 2, 2, 3])")
    assert result == {1, 2, 3}

    # range - iteration
    result = ds.eval("list(range(3))")
    assert result == [0, 1, 2]

    # zip, enumerate - iteration helpers
    result = ds.eval("list(zip([1, 2], [3, 4]))")
    assert result == [(1, 3), (2, 4)]

    result = ds.eval("list(enumerate(['a', 'b']))")
    assert result == [(0, "a"), (1, "b")]

    # map, filter - functional helpers
    result = ds.eval("list(map(abs, [-1, -2, 3]))")
    assert result == [1, 2, 3]

    result = ds.eval("list(filter(bool, [0, 1, 0, 2]))")
    assert result == [1, 2]

    # any, all - aggregation
    result = ds.eval("any([False, True, False])")
    assert result is True

    result = ds.eval("all([True, True, True])")
    assert result is True