def test_fix_xarray():
    import flox.xarray
    import xarray as xr
    import numpy as np

    sa=10
    sb=13
    sc=3

    x=xr.Dataset({
        'v0':xr.DataArray(((np.arange(sa*sb*sc)/sa)%1).reshape((sa,sb,sc)),dims=('a','b','c')),
        'v1':xr.DataArray((np.arange(sa*sb)%3).reshape(sa,sb),dims=('a','b')),
    })

    r=flox.xarray.xarray_reduce(x['v0'],
                              x['v1'],x['v0'],
                              expected_groups=(np.arange(6),np.linspace(0,1,num=5)),
                              isbin=[False,True],
                              func='count',
                              dim='b',
                              fill_value=np.nan,
                             ).mean('a',skipna=True)
    assert len(r['v1'])==6
test_fix_xarray()