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])