def test_no_vertical(self, mock_set_long):
     grib = None
     cube = iris.cube.Cube([1, 2, 3, 4, 5])
     grib_save_rules.set_fixed_surfaces(cube, grib)
     mock_set_long.assert_any_call(grib, "typeOfFirstFixedSurface", 1)
     mock_set_long.assert_any_call(grib, "scaleFactorOfFirstFixedSurface", 0)
     mock_set_long.assert_any_call(grib, "scaledValueOfFirstFixedSurface", 0)
     mock_set_long.assert_any_call(grib, "typeOfSecondFixedSurface", -1)
     mock_set_long.assert_any_call(grib, "scaleFactorOfSecondFixedSurface", 255)
     mock_set_long.assert_any_call(grib, "scaledValueOfSecondFixedSurface", -1)
示例#2
0
 def test_no_vertical(self, mock_set):
     grib = None
     cube = iris.cube.Cube([1, 2, 3, 4, 5])
     grib_save_rules.set_fixed_surfaces(cube, grib)
     mock_set.assert_any_call(grib, "typeOfFirstFixedSurface", 1)
     mock_set.assert_any_call(grib, "scaleFactorOfFirstFixedSurface", 0)
     mock_set.assert_any_call(grib, "scaledValueOfFirstFixedSurface", 0)
     mock_set.assert_any_call(grib, "typeOfSecondFixedSurface", -1)
     mock_set.assert_any_call(grib, "scaleFactorOfSecondFixedSurface", 255)
     mock_set.assert_any_call(grib, "scaledValueOfSecondFixedSurface", -1)
    def test_height_point(self, mock_set_long):
        grib = None
        cube = iris.cube.Cube([1, 2, 3, 4, 5])
        cube.add_aux_coord(iris.coords.AuxCoord([12345], "height", units="m"))

        grib_save_rules.set_fixed_surfaces(cube, grib)

        mock_set_long.assert_any_call(grib, "typeOfFirstFixedSurface", 103)
        mock_set_long.assert_any_call(grib, "scaleFactorOfFirstFixedSurface", 0)
        mock_set_long.assert_any_call(grib, "scaledValueOfFirstFixedSurface", 12345)
        mock_set_long.assert_any_call(grib, "typeOfSecondFixedSurface", -1)
        mock_set_long.assert_any_call(grib, "scaleFactorOfSecondFixedSurface", 255)
        mock_set_long.assert_any_call(grib, "scaledValueOfSecondFixedSurface", -1)
示例#4
0
    def test_height_point(self, mock_set):
        grib = None
        cube = iris.cube.Cube([1, 2, 3, 4, 5])
        cube.add_aux_coord(iris.coords.AuxCoord([12345], "height", units="m"))

        grib_save_rules.set_fixed_surfaces(cube, grib)

        mock_set.assert_any_call(grib, "typeOfFirstFixedSurface", 103)
        mock_set.assert_any_call(grib, "scaleFactorOfFirstFixedSurface", 0)
        mock_set.assert_any_call(grib, "scaledValueOfFirstFixedSurface", 12345)
        mock_set.assert_any_call(grib, "typeOfSecondFixedSurface", -1)
        mock_set.assert_any_call(grib, "scaleFactorOfSecondFixedSurface", 255)
        mock_set.assert_any_call(grib, "scaledValueOfSecondFixedSurface", -1)
示例#5
0
 def test_bounded_altitude_feet(self):
     cube = iris.cube.Cube([0])
     cube.add_aux_coord(iris.coords.AuxCoord(
         1500.0, long_name='altitude', units='ft',
         bounds=np.array([1000.0, 2000.0])))
     grib = gribapi.grib_new_from_samples("GRIB2")
     set_fixed_surfaces(cube, grib)
     self.assertEqual(
         gribapi.grib_get_double(grib, "scaledValueOfFirstFixedSurface"),
         304.0)
     self.assertEqual(
         gribapi.grib_get_double(grib, "scaledValueOfSecondFixedSurface"),
         609.0)
示例#6
0
 def test_bounded_altitude_feet(self):
     cube = iris.cube.Cube([0])
     cube.add_aux_coord(
         iris.coords.AuxCoord(1500.0,
                              long_name='altitude',
                              units='ft',
                              bounds=np.array([1000.0, 2000.0])))
     grib = gribapi.grib_new_from_samples("GRIB2")
     set_fixed_surfaces(cube, grib)
     self.assertEqual(
         gribapi.grib_get_double(grib, "scaledValueOfFirstFixedSurface"),
         304.0)
     self.assertEqual(
         gribapi.grib_get_double(grib, "scaledValueOfSecondFixedSurface"),
         609.0)
     self.assertEqual(
         gribapi.grib_get_long(grib, "typeOfFirstFixedSurface"), 102)
     self.assertEqual(
         gribapi.grib_get_long(grib, "typeOfSecondFixedSurface"), 102)
示例#7
0
 def test_theta_level(self):
     cube = iris.cube.Cube([0])
     cube.add_aux_coord(iris.coords.AuxCoord(
         230.0, standard_name='air_potential_temperature',
         units='K', attributes={'positive': 'up'},
         bounds=np.array([220.0, 240.0])))
     grib = gribapi.grib_new_from_samples("GRIB2")
     set_fixed_surfaces(cube, grib)
     self.assertEqual(
         gribapi.grib_get_double(grib, "scaledValueOfFirstFixedSurface"),
         220.0)
     self.assertEqual(
         gribapi.grib_get_double(grib, "scaledValueOfSecondFixedSurface"),
         240.0)
     self.assertEqual(
         gribapi.grib_get_long(grib, "typeOfFirstFixedSurface"),
         107)
     self.assertEqual(
         gribapi.grib_get_long(grib, "typeOfSecondFixedSurface"),
         107)
示例#8
0
 def test_theta_level(self):
     cube = iris.cube.Cube([0])
     cube.add_aux_coord(
         iris.coords.AuxCoord(230.0,
                              standard_name='air_potential_temperature',
                              units='K',
                              attributes={'positive': 'up'},
                              bounds=np.array([220.0, 240.0])))
     grib = gribapi.grib_new_from_samples("GRIB2")
     set_fixed_surfaces(cube, grib)
     self.assertEqual(
         gribapi.grib_get_double(grib, "scaledValueOfFirstFixedSurface"),
         220.0)
     self.assertEqual(
         gribapi.grib_get_double(grib, "scaledValueOfSecondFixedSurface"),
         240.0)
     self.assertEqual(
         gribapi.grib_get_long(grib, "typeOfFirstFixedSurface"), 107)
     self.assertEqual(
         gribapi.grib_get_long(grib, "typeOfSecondFixedSurface"), 107)