def test_binary(self):
args = [0,
np.zeros(2),
xray.Variable(['x'], [0, 0]),
xray.DataArray([0, 0], dims='x'),
xray.Dataset({'y': ('x', [0, 0])})]
for n, t1 in enumerate(args):
for t2 in args[n:]:
self.assertIdentical(t2 + 1, xu.maximum(t1, t2 + 1))
self.assertIdentical(t2 + 1, xu.maximum(t2, t1 + 1))
self.assertIdentical(t2 + 1, xu.maximum(t1 + 1, t2))
self.assertIdentical(t2 + 1, xu.maximum(t2 + 1, t1))
def test_binary(self):
args = [0,
np.zeros(2),
xray.Variable(['x'], [0, 0]),
xray.DataArray([0, 0], dims='x'),
xray.Dataset({'y': ('x', [0, 0])})]
for n, t1 in enumerate(args):
for t2 in args[n:]:
self.assertIdentical(t2 + 1, xu.maximum(t1, t2 + 1))
self.assertIdentical(t2 + 1, xu.maximum(t2, t1 + 1))
self.assertIdentical(t2 + 1, xu.maximum(t1 + 1, t2))
self.assertIdentical(t2 + 1, xu.maximum(t2 + 1, t1))
def test_groupby(self):
ds = xray.Dataset({'a': ('x', [0, 0, 0])}, {'c': ('x', [0, 0, 1])})
ds_grouped = ds.groupby('c')
group_mean = ds_grouped.mean('x')
arr_grouped = ds['a'].groupby('c')
self.assertIdentical(ds, xu.maximum(ds_grouped, group_mean))
self.assertIdentical(ds, xu.maximum(group_mean, ds_grouped))
self.assertIdentical(ds, xu.maximum(arr_grouped, group_mean))
self.assertIdentical(ds, xu.maximum(group_mean, arr_grouped))
self.assertIdentical(ds, xu.maximum(ds_grouped, group_mean['a']))
self.assertIdentical(ds, xu.maximum(group_mean['a'], ds_grouped))
self.assertIdentical(ds.a, xu.maximum(arr_grouped, group_mean.a))
self.assertIdentical(ds.a, xu.maximum(group_mean.a, arr_grouped))
with self.assertRaisesRegexp(TypeError, 'only support binary ops'):
xu.maximum(ds.a.variable, ds_grouped)
def test_groupby(self):
ds = xray.Dataset({'a': ('x', [0, 0, 0])}, {'c': ('x', [0, 0, 1])})
ds_grouped = ds.groupby('c')
group_mean = ds_grouped.mean('x')
arr_grouped = ds['a'].groupby('c')
self.assertIdentical(ds, xu.maximum(ds_grouped, group_mean))
self.assertIdentical(ds, xu.maximum(group_mean, ds_grouped))
self.assertIdentical(ds, xu.maximum(arr_grouped, group_mean))
self.assertIdentical(ds, xu.maximum(group_mean, arr_grouped))
self.assertIdentical(ds, xu.maximum(ds_grouped, group_mean['a']))
self.assertIdentical(ds, xu.maximum(group_mean['a'], ds_grouped))
self.assertIdentical(ds.a, xu.maximum(arr_grouped, group_mean.a))
self.assertIdentical(ds.a, xu.maximum(group_mean.a, arr_grouped))
with self.assertRaisesRegexp(TypeError, 'only support binary ops'):
xu.maximum(ds.a.variable, ds_grouped)
