def test_cell_method_coord_name_fail(self, mock_set):
cube = self.cube
cell_method = CellMethod(method='mean', coords=['season'])
cube.add_cell_method(cell_method)
with self.assertRaisesRegexp(
ValueError, "Expected a cell method with a coordinate "
"name of 'time'"):
_product_definition_template_8_and_11(cube, mock.sentinel.grib)
def test_unrecognised(self, mock_set):
cube = self.cube
cell_method = CellMethod(method='95th percentile', coords=['time'])
cube.add_cell_method(cell_method)
_product_definition_template_8_and_11(cube, mock.sentinel.grib)
mock_set.assert_any_call(mock.sentinel.grib,
"typeOfStatisticalProcessing", 255)
def test_multiple_cell_method_coords(self, mock_set):
cube = self.cube
cell_method = CellMethod(method='sum',
coords=['time', 'forecast_period'])
cube.add_cell_method(cell_method)
with self.assertRaisesRegexp(ValueError,
'Cannot handle multiple coordinate name'):
_product_definition_template_8_and_11(cube, mock.sentinel.grib)
def test_multiple_cell_method_coords(self, mock_set):
cube = self.cube
cell_method = CellMethod(method='sum',
coords=['time', 'forecast_period'])
cube.add_cell_method(cell_method)
with self.assertRaisesRegexp(ValueError,
'Cannot handle multiple coordinate name'):
_product_definition_template_8_and_11(cube, mock.sentinel.grib)
def test_cell_method_coord_name_fail(self, mock_set):
cube = self.cube
cell_method = CellMethod(method='mean', coords=['season'])
cube.add_cell_method(cell_method)
with self.assertRaisesRegexp(
ValueError, "Expected a cell method with a coordinate "
"name of 'time'"):
_product_definition_template_8_and_11(cube, mock.sentinel.grib)
def test_unrecognised(self, mock_set):
cube = self.cube
cell_method = CellMethod(method='95th percentile', coords=['time'])
cube.add_cell_method(cell_method)
_product_definition_template_8_and_11(cube, mock.sentinel.grib)
mock_set.assert_any_call(mock.sentinel.grib,
"typeOfStatisticalProcessing", 255)
def test_multiple_points(self, mock_set):
# Add time coord with multiple points.
coord = DimCoord([23, 24, 25], 'time',
bounds=[[22, 23], [23, 24], [24, 25]],
units=Unit('days since epoch', calendar='standard'))
self.cube.add_aux_coord(coord, 0)
with self.assertRaisesRegexp(
ValueError, 'Expected length one time coordinate'):
_product_definition_template_8_and_11(self.cube,
mock.sentinel.grib)
def test_more_than_two_bounds(self, mock_set):
# Add time coord with more than two bounds.
coord = DimCoord(23, 'time', bounds=[21, 22, 23],
units=Unit('days since epoch', calendar='standard'))
self.cube.add_aux_coord(coord)
with self.assertRaisesRegexp(
ValueError, 'Expected time coordinate with two bounds, '
'got 3 bounds'):
_product_definition_template_8_and_11(self.cube,
mock.sentinel.grib)
def test_multiple_points(self, mock_set):
# Add time coord with multiple points.
coord = DimCoord([23, 24, 25], 'time',
bounds=[[22, 23], [23, 24], [24, 25]],
units=Unit('days since epoch', calendar='standard'))
self.cube.add_aux_coord(coord, 0)
with self.assertRaisesRegexp(
ValueError, 'Expected length one time coordinate'):
_product_definition_template_8_and_11(self.cube,
mock.sentinel.grib)
def test_more_than_two_bounds(self, mock_set):
# Add time coord with more than two bounds.
coord = DimCoord(23, 'time', bounds=[21, 22, 23],
units=Unit('days since epoch', calendar='standard'))
self.cube.add_aux_coord(coord)
with self.assertRaisesRegexp(
ValueError, 'Expected time coordinate with two bounds, '
'got 3 bounds'):
_product_definition_template_8_and_11(self.cube,
mock.sentinel.grib)
def test_other_cell_methods(self, mock_set):
cube = stock.lat_lon_cube()
# Rename cube to avoid warning about unknown discipline/parameter.
cube.rename('air_temperature')
coord = DimCoord(23, 'time', bounds=[0, 24],
units=Unit('hours since epoch'))
cube.add_aux_coord(coord)
# Add one time cell method and another unrelated one.
cell_method = CellMethod(method='mean', coords=['elephants'])
cube.add_cell_method(cell_method)
cell_method = CellMethod(method='sum', coords=['time'])
cube.add_cell_method(cell_method)
_product_definition_template_8_and_11(cube, mock.sentinel.grib)
mock_set.assert_any_call(mock.sentinel.grib, 'numberOfTimeRange', 1)
def test_other_cell_methods(self, mock_set):
cube = stock.lat_lon_cube()
# Rename cube to avoid warning about unknown discipline/parameter.
cube.rename('air_temperature')
coord = DimCoord(23, 'time', bounds=[0, 24],
units=Unit('hours since epoch'))
cube.add_aux_coord(coord)
# Add one time cell method and another unrelated one.
cell_method = CellMethod(method='mean', coords=['elephants'])
cube.add_cell_method(cell_method)
cell_method = CellMethod(method='sum', coords=['time'])
cube.add_cell_method(cell_method)
_product_definition_template_8_and_11(cube, mock.sentinel.grib)
mock_set.assert_any_call(mock.sentinel.grib, 'numberOfTimeRange', 1)
def test_360_day_calendar(self, mock_set):
cube = self.cube
# End bound is 1972-05-07 10:27:07
coord = DimCoord(23.0,
'time',
bounds=[0.452, 20314.452],
units=Unit('hours since epoch', calendar='360_day'))
cube.add_aux_coord(coord)
grib = mock.sentinel.grib
_product_definition_template_8_and_11(cube, grib)
mock_set.assert_any_call(grib, "yearOfEndOfOverallTimeInterval", 1972)
mock_set.assert_any_call(grib, "monthOfEndOfOverallTimeInterval", 5)
mock_set.assert_any_call(grib, "dayOfEndOfOverallTimeInterval", 7)
mock_set.assert_any_call(grib, "hourOfEndOfOverallTimeInterval", 10)
mock_set.assert_any_call(grib, "minuteOfEndOfOverallTimeInterval", 27)
mock_set.assert_any_call(grib, "secondOfEndOfOverallTimeInterval", 7)
def test_360_day_calendar(self, mock_set):
cube = self.cube
# End bound is 1972-05-07 10:27:07
coord = DimCoord(23.0, 'time', bounds=[0.452, 20314.452],
units=Unit('hours since epoch', calendar='360_day'))
cube.add_aux_coord(coord)
grib = mock.sentinel.grib
_product_definition_template_8_and_11(cube, grib)
mock_set.assert_any_call(
grib, "yearOfEndOfOverallTimeInterval", 1972)
mock_set.assert_any_call(
grib, "monthOfEndOfOverallTimeInterval", 5)
mock_set.assert_any_call(
grib, "dayOfEndOfOverallTimeInterval", 7)
mock_set.assert_any_call(
grib, "hourOfEndOfOverallTimeInterval", 10)
mock_set.assert_any_call(
grib, "minuteOfEndOfOverallTimeInterval", 27)
mock_set.assert_any_call(
grib, "secondOfEndOfOverallTimeInterval", 7)
