test_mask.py 文件源码

def test__mask_xarray_in_out_2D():
    # create xarray DataArray with 2D dims

    coords = {'lat_1D': [1, 2],
              'lon_1D': [1, 2],
              'lat_2D': (('lat_1D', 'lon_1D'), lat_2D),
              'lon_2D': (('lat_1D', 'lon_1D'), lon_2D)}

    d = np.random.rand(2, 2)

    data = xr.DataArray(d, coords = coords, dims=('lat_1D', 'lon_1D'))


    expected = expected_mask()
    result = r1.mask(data, lon_name='lon_2D', lat_name='lat_2D')

    assert isinstance(result, xr.DataArray)
    assert np.allclose(result, expected, equal_nan=True)
    assert np.allclose(result.lat_2D, lat_2D)
    assert np.allclose(result.lon_2D, lon_2D)

    assert np.allclose(result.lat_1D, [1, 2])
    assert np.allclose(result.lon_1D, [1, 2])