def test_get_son(self):
"""Test function for summer"""
sliced = extract_season(self.cube, 'son')
iris.coord_categorisation.add_month_number(sliced, 'time')
assert_array_equal(
np.array([9, 10, 11, 9, 10, 11]),
sliced.coord('month_number').points)
def test_get_mam(self):
"""Test function for spring"""
sliced = extract_season(self.cube, 'mam')
iris.coord_categorisation.add_month_number(sliced, 'time')
assert_array_equal(
np.array([3, 4, 5, 3, 4, 5]),
sliced.coord('month_number').points)
def test_get_djf_caps(self):
"""Test function works when season specified in caps"""
sliced = extract_season(self.cube, 'DJF')
iris.coord_categorisation.add_month_number(sliced, 'time')
assert_array_equal(
np.array([1, 2, 12, 1, 2, 12]),
sliced.coord('month_number').points)
def test_get_djf(self):
"""Test function for winter"""
sliced = extract_season(self.cube, 'djf')
iris.coord_categorisation.add_month_number(sliced, 'time')
assert_array_equal(
np.array([1, 2, 12, 1, 2, 12]),
sliced.coord('month_number').points)
def test_get_jja(self):
"""Test function for summer"""
sliced = extract_season(self.cube, 'jja')
assert_array_equal(np.array([6, 7, 8, 6, 7, 8]),
sliced.coord('month_number').points)
def test_get_djf(self):
"""Test function for winter"""
sliced = extract_season(self.cube, 'djf')
print(sliced)
assert_array_equal(np.array([1, 2, 12, 1, 2, 12]),
sliced.coord('month_number').points)
def test_get_son(self):
"""Test function for summer"""
sliced = extract_season(self.cube, 'son')
print(sliced)
assert_array_equal(np.array([9, 10, 11, 9, 10, 11]),
sliced.coord('month_number').points)
def test_get_mam(self):
"""Test function for spring"""
sliced = extract_season(self.cube, 'mam')
print(sliced)
assert_array_equal(np.array([3, 4, 5, 3, 4, 5]),
sliced.coord('month_number').points)
def test_get_djf_caps(self):
"""Test function works when season specified in caps"""
sliced = extract_season(self.cube, 'DJF')
print(sliced)
assert_array_equal(np.array([1, 2, 12, 1, 2, 12]),
sliced.coord('month_number').points)