def test_lazy(self): lazy_data = COUNT.lazy_aggregate(self.lazy_cube.lazy_data(), axis=0, function=self.func) self.assertTrue(is_lazy_data(lazy_data))
def test_not_callable(self): with self.assertRaisesRegex(TypeError, 'function must be a callable'): COUNT.aggregate(self.cube.data, axis=0, function='wibble')
def test_lazy_not_callable(self): with self.assertRaisesRegex(TypeError, 'function must be a callable'): COUNT.lazy_aggregate(self.lazy_cube.lazy_data(), axis=0, function='wibble')
def test_name(self): self.assertEqual(COUNT.name(), 'count')
def test_no_function(self): exp_emsg = r"function must be a callable. Got <.* 'NoneType'>" with self.assertRaisesRegex(TypeError, exp_emsg): COUNT.lazy_aggregate(self.lazy_cube.lazy_data(), axis=0)
def test_ma(self): data = COUNT.aggregate(self.cube.data, axis=0, function=self.func) self.assertArrayEqual(data, [2])
def test_no_function(self): exp_emsg = r"function must be a callable. Got <.* 'NoneType'>" with self.assertRaisesRegexp(TypeError, exp_emsg): COUNT.lazy_aggregate(self.lazy_cube.lazy_data(), axis=0)
def test(self): self.assertIsNotNone(COUNT.units_func) new_units = COUNT.units_func(None) self.assertEqual(new_units, 1)
def test(self): self.assertEqual(COUNT.name(), 'count')
def test_lazy_collapse(self): lazy_data = COUNT.lazy_aggregate(self.lazy_cube.lazy_data(), axis=0, function=self.func) self.assertArrayEqual(lazy_data.compute(), [3])
def test_lazy_not_callable(self): with self.assertRaisesRegexp(TypeError, 'function must be a callable'): COUNT.lazy_aggregate(self.lazy_cube.lazy_data(), axis=0, function='wibble')
def test_not_callable(self): with self.assertRaisesRegexp(TypeError, 'function must be a callable'): COUNT.aggregate(self.cube.data, axis=0, function='wibble')
def test(self): shape = () kwargs = dict() self.assertTupleEqual(COUNT.aggregate_shape(**kwargs), shape) kwargs = dict(wibble='wobble') self.assertTupleEqual(COUNT.aggregate_shape(**kwargs), shape)
def test_ma(self): lazy_data = COUNT.lazy_aggregate(self.lazy_cube.lazy_data(), axis=0, function=self.func) self.assertTrue(is_lazy_data(lazy_data)) self.assertArrayEqual(lazy_data.compute(), [2])