def test_returned(self):
lazy_data, weights = SUM.lazy_aggregate(
self.cube_2d.lazy_data(), axis=0, returned=True
)
self.assertTrue(is_lazy_data(lazy_data))
self.assertArrayEqual(lazy_data.compute(), [7, 9, 11, 13, 15])
self.assertArrayEqual(weights, [2, 2, 2, 2, 2])
def test_weights_and_returned(self):
data, weights = SUM.aggregate(self.cube_2d.data,
axis=0,
weights=self.weights,
returned=True)
self.assertArrayEqual(data, [14, 9, 11, 13, 15])
self.assertArrayEqual(weights, [4, 2, 2, 2, 2])
def test_weights(self):
data = SUM.aggregate(self.cube_2d.data, axis=0, weights=self.weights)
self.assertArrayEqual(data, [14, 9, 11, 13, 15])
def test_lazy_ma(self):
lazy_data = SUM.lazy_aggregate(self.cube.lazy_data(), axis=0)
self.assertTrue(is_lazy_data(lazy_data))
self.assertArrayEqual(lazy_data.compute(), [12])
def test_ma(self):
data = SUM.aggregate(self.cube.data, axis=0)
self.assertArrayEqual(data, [12])
def test_lazy_collapse(self):
lazy_data = SUM.lazy_aggregate(self.lazy_cube.lazy_data(), axis=0)
self.assertArrayEqual(lazy_data.compute(), [15])
def test_lazy(self):
lazy_data = SUM.lazy_aggregate(self.lazy_cube.lazy_data(), axis=0)
self.assertTrue(is_lazy_data(lazy_data))
def test_collapse(self):
data = SUM.aggregate(self.cube.data, axis=0)
self.assertArrayEqual(data, [15])
def test_name(self):
self.assertEqual(SUM.name(), "sum")
def test(self):
shape = ()
kwargs = dict()
self.assertTupleEqual(SUM.aggregate_shape(**kwargs), shape)
kwargs = dict(wibble="wobble")
self.assertTupleEqual(SUM.aggregate_shape(**kwargs), shape)
def test_weights(self):
lazy_data = SUM.lazy_aggregate(
self.cube_2d.lazy_data(), axis=0, weights=self.weights
)
self.assertTrue(is_lazy_data(lazy_data))
self.assertArrayEqual(lazy_data.compute(), [14, 9, 11, 13, 15])
