def test_process_3d_xarray_dataset_with_coords(self): import pandas as pd data, x, y, by, groupby = process_xarray(data=self.ds, **self.default_kwargs) assert isinstance(data, pd.DataFrame) assert x == 'time' assert y == ['air'] assert not by assert groupby == ['lon', 'lat']
def test_process_2d_xarray_dataarray_with_no_coords(self): import xarray as xr import pandas as pd da = xr.DataArray(np.random.randn(4,5)) data, x, y, by, groupby = process_xarray(data=da, **self.default_kwargs) assert isinstance(data, pd.DataFrame) assert x == 'index' assert y == ['value'] assert not by assert not groupby
def test_process_3d_xarray_dataset_with_coords_as_gridded(self): import xarray as xr kwargs = self.default_kwargs kwargs.update(gridded=True, x='lon', y='lat') data, x, y, by, groupby = process_xarray(data=self.ds, **kwargs) assert isinstance(data, xr.Dataset) assert list(data.data_vars.keys()) == ['air'] assert x == 'lon' assert y == 'lat' assert by is None assert groupby == ['time']
def test_process_xarray_dataset_with_x_as_derived_datetime(self): import pandas as pd data = self.ds.mean(dim=['lat', 'lon']) kwargs = self.default_kwargs kwargs.update(gridded=False, y='air', x='time.dayofyear') data, x, y, by, groupby = process_xarray(data=data, **kwargs) assert isinstance(data, pd.DataFrame) assert x == 'time.dayofyear' assert y == 'air' assert not by assert not groupby
def test_process_2d_xarray_dataarray_with_no_coords_as_gridded(self): import xarray as xr da = xr.DataArray(np.random.randn(4, 5)) kwargs = self.default_kwargs kwargs.update(gridded=True) data, x, y, by, groupby = process_xarray(data=da, **kwargs) assert isinstance(data, xr.Dataset) assert list(data.data_vars.keys()) == ['value'] assert x == 'dim_1' assert y == 'dim_0' assert by is None assert groupby is None
def test_process_1d_xarray_dataarray_with_coords(self): import xarray as xr import pandas as pd da = xr.DataArray(data=[1, 2, 3], coords={'day': [5, 6, 7]}, dims=['day']) data, x, y, by, groupby = process_xarray(data=da, **self.default_kwargs) assert isinstance(data, pd.DataFrame) assert x == 'day' assert y == ['value'] assert by == [] assert groupby == []
def test_process_2d_xarray_dataarray_with_coords_as_gridded(self): import xarray as xr da = xr.DataArray(data=np.random.randn(4, 5), coords={'y': [3, 4, 5, 6, 7]}, dims=['x', 'y']) kwargs = self.default_kwargs kwargs.update(gridded=True) data, x, y, by, groupby = process_xarray(data=da, **kwargs) assert isinstance(data, xr.Dataset) assert list(data.data_vars.keys()) == ['value'] assert x == 'y' assert y == 'x' assert not by assert not groupby