Function test_format_broadcast

xarray/tests/test_accessor_str.py:3581–3612

()

def test_format_broadcast() -> None:
    values = xr.DataArray(
        ["{}.{Y}.{ZZ}", "{},{},{X},{X}", "{X}-{Y}-{ZZ}"],
        dims=["X"],
    ).astype(np.str_)

    pos0 = 1
    pos1 = 1.2

    pos2 = xr.DataArray(
        ["2.3", "3.44444"],
        dims=["YY"],
    )

    X = "'test'"
    Y = "X"
    ZZ = None
    W = "NO!"

    expected = xr.DataArray(
        [
            ["1.X.None", "1.X.None"],
            ["1,1.2,'test','test'", "1,1.2,'test','test'"],
            ["'test'-X-None", "'test'-X-None"],
        ],
        dims=["X", "YY"],
    ).astype(np.str_)

    res = values.str.format(pos0, pos1, pos2, X=X, Y=Y, ZZ=ZZ, W=W)

    assert res.dtype == expected.dtype
    assert_equal(res, expected)


def test_mod_scalar() -> None: