Function test_eval_data_variable_priority

xarray/tests/test_eval.py:259–304

Test that data variables take priority over builtin functions. Users may have data variables named 'sum', 'abs', 'min', etc. When they reference these in eval(), they should get their data, not the Python builtins. The builtins should still be accessible via the np namespace (np.sum, np

()

def test_eval_data_variable_priority() -> None:
    """Test that data variables take priority over builtin functions.

    Users may have data variables named 'sum', 'abs', 'min', etc. When they
    reference these in eval(), they should get their data, not the Python builtins.
    The builtins should still be accessible via the np namespace (np.sum, np.abs).
    """
    # Create dataset with data variables that shadow builtins
    ds = Dataset(
        {
            "sum": ("x", [10.0, 20.0, 30.0]),  # shadows builtin sum
            "abs": ("x", [1.0, 2.0, 3.0]),  # shadows builtin abs
            "min": ("x", [100.0, 200.0, 300.0]),  # shadows builtin min
            "other": ("x", [5.0, 10.0, 15.0]),
        }
    )

    # Data variables should take priority - user data wins
    result = ds.eval("sum + other")
    expected = ds["sum"] + ds["other"]
    assert_equal(result, expected)

    # Should get the data variable, not builtin sum applied to something
    result = ds.eval("sum * 2")
    expected = ds["sum"] * 2
    assert_equal(result, expected)

    # abs as data variable should work
    result = ds.eval("abs + 1")
    expected = ds["abs"] + 1
    assert_equal(result, expected)

    # min as data variable should work
    result = ds.eval("min - 50")
    expected = ds["min"] - 50
    assert_equal(result, expected)

    # np namespace should still provide access to actual functions
    result = ds.eval("np.abs(other - 10)")
    expected = abs(ds["other"] - 10)
    assert_equal(result, expected)

    # np.sum should work even when 'sum' is a data variable
    result = ds.eval("np.sum(other)")
    expected = np.sum(ds["other"])
    assert result == expected


def test_eval_coordinate_priority() -> None: