Test with flox (#339)

* Test with flox

* Actually test

* [test-upstream]

* [test-upstream]

* [test-upstream]

* Fix grouped subtraction test a bit

Author: dcherian

cf_xarray/tests/test_accessor.py

@@ -1318,8 +1318,12 @@ def test_groupby_special_ops():
         assert_identical(group, cfgroup)
 
     # arithmetic
-    expected = grouped - grouped.mean()
-    actual = grouped - cfgrouped.mean()
+    # TODO: Extremely buggy!
+    # 1. cfgrouped - cfgrouped.mean() raises RecursionError
+    # 2. use_flox=True doesn't preserve attributes for some reason
+    with xr.set_options(keep_attrs=True):
+        expected = grouped - grouped.mean()
+        actual = grouped - cfgrouped.mean()
     assert_identical(expected, actual)
 
 

ci/environment.yml

@@ -6,6 +6,7 @@ dependencies:
   - pytest
   - pytest-xdist
   - dask
+  - flox
   - matplotlib-base
   - netcdf4
   - pandas

ci/upstream-dev-env.yml

@@ -13,4 +13,5 @@ dependencies:
   - shapely
   - pip:
     - git+https://github.com/pydata/xarray
+    - git+https://github.com/dcherian/flox
     - git+https://github.com/hgrecco/pint