Function test_cat_uniform

xarray/tests/test_accessor_str.py:3175–3221

(dtype)

def test_cat_uniform(dtype) -> None:
    values_1 = xr.DataArray(
        [["a", "bb", "cccc"], ["ddddd", "eeee", "fff"]],
        dims=["X", "Y"],
    ).astype(dtype)
    values_2 = xr.DataArray(
        [["11111", "222", "33"], ["4", "5555", "66"]],
        dims=["X", "Y"],
    )

    targ_blank = xr.DataArray(
        [["a11111", "bb222", "cccc33"], ["ddddd4", "eeee5555", "fff66"]],
        dims=["X", "Y"],
    ).astype(dtype)

    targ_space = xr.DataArray(
        [["a 11111", "bb 222", "cccc 33"], ["ddddd 4", "eeee 5555", "fff 66"]],
        dims=["X", "Y"],
    ).astype(dtype)

    targ_bars = xr.DataArray(
        [["a||11111", "bb||222", "cccc||33"], ["ddddd||4", "eeee||5555", "fff||66"]],
        dims=["X", "Y"],
    ).astype(dtype)

    targ_comma = xr.DataArray(
        [["a, 11111", "bb, 222", "cccc, 33"], ["ddddd, 4", "eeee, 5555", "fff, 66"]],
        dims=["X", "Y"],
    ).astype(dtype)

    res_blank = values_1.str.cat(values_2)
    res_add = values_1.str + values_2
    res_space = values_1.str.cat(values_2, sep=" ")
    res_bars = values_1.str.cat(values_2, sep="||")
    res_comma = values_1.str.cat(values_2, sep=", ")

    assert res_blank.dtype == targ_blank.dtype
    assert res_add.dtype == targ_blank.dtype
    assert res_space.dtype == targ_space.dtype
    assert res_bars.dtype == targ_bars.dtype
    assert res_comma.dtype == targ_comma.dtype

    assert_equal(res_blank, targ_blank)
    assert_equal(res_add, targ_blank)
    assert_equal(res_space, targ_space)
    assert_equal(res_bars, targ_bars)
    assert_equal(res_comma, targ_comma)


def test_cat_broadcast_right(dtype) -> None: