def test_point(self): t_coord = mock.Mock() t_coord.has_bounds = mock.Mock(return_value=False) t_coord.points = [15] cube = mock.Mock() cube.coord = mock.Mock(return_value=t_coord) rt, rt_meaning, fp, fp_meaning = _missing_forecast_period(cube) t_coord.units.assert_has_call(mock.call.num2date(15)) self.assertEqual((rt_meaning, fp, fp_meaning), (2, 0, 1))
def test_bounds(self): t_coord = mock.Mock() t_coord.has_bounds = mock.Mock(return_value=True) t_coord.points = [15] t_coord.bounds = np.array([[10, 20]]) cube = mock.Mock() cube.coord = mock.Mock(return_value=t_coord) rt, rt_meaning, fp, fp_meaning = _missing_forecast_period(cube) t_coord.units.assert_has_call(mock.call.num2date(10)) self.assertEqual((rt_meaning, fp, fp_meaning), (2, 0, 1))