def test_get_january_with_existing_coord(self):
"""Test january extraction"""
iris.coord_categorisation.add_month_number(self.cube, 'time')
sliced = extract_month(self.cube, 1)
assert_array_equal(
np.array([1, 1]),
sliced.coord('month_number').points)
def test_bad_month_raises(self):
"""Test january extraction"""
with self.assertRaises(ValueError):
extract_month(self.cube, 13)
with self.assertRaises(ValueError):
extract_month(self.cube, -1)
def test_get_january(self):
"""Test january extraction"""
sliced = extract_month(self.cube, 1)
assert_array_equal(np.array([1, 1]),
sliced.coord('month_number').points)
