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
