[skip-ci] Add alignment benchmarks

asv_bench/asv.conf.json

@@ -68,7 +68,8 @@
         "distributed": [""],
         "flox": [""],
         "numpy_groupies": [""],
-        "sparse": [""]
+        "sparse": [""],
+        "cftime": [""]
     },
 
 

asv_bench/benchmarks/alignment.py

@@ -0,0 +1,45 @@
+import numpy as np
+
+import xarray as xr
+
+from . import parameterized, requires_dask
+
+ntime = 365 * 30
+nx = 50
+ny = 50
+
+
+class Align:
+    def setup(self, *args, **kwargs):
+        data = np.random.RandomState(0).randn(ntime, nx, ny)
+        self.ds = xr.Dataset(
+            {"temperature": (("time", "x", "y"), data)},
+            coords={
+                "time": xr.date_range("2000", periods=ntime),
+                "x": np.arange(nx),
+                "y": np.arange(ny),
+            },
+        )
+        self.year = self.ds.time.dt.year
+
+    @parameterized(["join"], [("outer", "inner", "left", "right", "exact", "override")])
+    def time_already_aligned(self, join):
+        xr.align(self.ds, self.year, join=join)
+
+    @parameterized(["join"], [("outer", "inner", "left", "right")])
+    def time_not_aligned(self, join):
+        xr.align(self.ds, self.year[-100:], join=join)
+
+
+class AlignCFTime(Align):
+    def setup(self, *args, **kwargs):
+        super().setup()
+        self.ds["time"] = xr.date_range("2000", periods=ntime, calendar="noleap")
+        self.year = self.ds.time.dt.year
+
+
+class AlignDask(Align):
+    def setup(self, *args, **kwargs):
+        requires_dask()
+        super().setup()
+        self.ds = self.ds.chunk({"time": 100})