def test_aggregate_categorical_season_membership(self):
test_cube_a = common._generate_extended_cube()
test_cube_a = ch.aggregate_categorical(test_cube_a,
'season_membership')
self.assertIsNotNone(test_cube_a)
self.assertEqual(
test_cube_a.coord('season_membership').points[0], True)
def test_aggregate_categorical_compounds(self):
test_cube_a = common._generate_extended_cube()
test_cube_a = ch.aggregate_categorical(test_cube_a,
'annual_seasonal_mean')
self.assertIsNotNone(test_cube_a)
self.assertEqual(test_cube_a.coord('time').bounds[0][0], 0)
self.assertEqual(test_cube_a.coord('time').bounds[0][1], 58.0)
self.assertEqual(test_cube_a.coord('time').points[0], 29.0)
self.assertEqual(test_cube_a.coord('clim_season').points[0], 'djf')
self.assertEqual(test_cube_a.coord('season_year').points[0], 1970)
def test_aggregate_categorical_day_of_year(self):
test_cube_a = common._generate_extended_cube()
test_cube_a = ch.aggregate_categorical(test_cube_a, 'day_of_year')
self.assertEqual(test_cube_a.coord('day_of_year').points[0], 1)
self.assertEqual(test_cube_a.coord('day_of_year').points[47], 48)
self.assertEqual(test_cube_a.coord('day_of_year').points[133], 134)
self.assertEqual(test_cube_a.coord('day_of_year').points[260], 261)
self.assertEqual(test_cube_a.coord('time').points[0], 3470.0)
self.assertEqual(test_cube_a.coord('time').points[47], 3517.0)
self.assertEqual(test_cube_a.coord('time').points[133], 3420.0)
self.assertEqual(test_cube_a.coord('time').points[260], 3547.0)
def test_aggregate_categorical_season(self):
test_cube_a = common._generate_extended_cube()
test_cube_a = ch.aggregate_categorical(test_cube_a, 'season')
self.assertIsNotNone(test_cube_a)
self.assertEqual(test_cube_a.coord('season').points[0], 'djf')
self.assertEqual(test_cube_a.coord('season').points[1], 'mam')
self.assertEqual(test_cube_a.coord('season').points[2], 'jja')
self.assertEqual(test_cube_a.coord('season').points[3], 'son')
self.assertEqual(test_cube_a.coord('time').points[0], 3499.0)
self.assertEqual(test_cube_a.coord('time').points[1], 3529.0)
self.assertEqual(test_cube_a.coord('time').points[2], 3484.0)
self.assertEqual(test_cube_a.coord('time').points[3], 3575.5)
def test_aggregate_categorical_season_year(self):
test_cube_a = common._generate_extended_cube()
test_cube_a = ch.aggregate_categorical(test_cube_a, 'season_year')
self.assertIsNotNone(test_cube_a)
self.assertEqual(test_cube_a.coord('season_year').points[0], 1970)
self.assertEqual(test_cube_a.coord('season_year').points[4], 1974)
self.assertEqual(test_cube_a.coord('season_year').points[9], 1979)
self.assertEqual(test_cube_a.coord('season_year').points[14], 1984)
self.assertEqual(test_cube_a.coord('time').points[0], 166.5)
self.assertEqual(test_cube_a.coord('time').points[4], 1612.0)
self.assertEqual(test_cube_a.coord('time').points[9], 3438.0)
self.assertEqual(test_cube_a.coord('time').points[14], 5264.5)
def test_aggregate_categorical_year(self):
test_cube_a = common._generate_extended_cube()
test_cube_a = ch.aggregate_categorical(test_cube_a, 'year')
self.assertIsNotNone(test_cube_a)
self.assertEqual(test_cube_a.coord('year').points[0], 1970)
self.assertEqual(test_cube_a.coord('year').points[4], 1974)
self.assertEqual(test_cube_a.coord('year').points[9], 1979)
self.assertEqual(test_cube_a.coord('year').points[14], 1984)
self.assertEqual(test_cube_a.coord('time').points[0], 182.0)
self.assertEqual(test_cube_a.coord('time').points[4], 1643.0)
self.assertEqual(test_cube_a.coord('time').points[9], 3469.0)
self.assertEqual(test_cube_a.coord('time').points[14], 5295.5)
def test_aggregate_categorical_month_number(self):
test_cube_a = common._generate_extended_cube()
test_cube_a = ch.aggregate_categorical(test_cube_a, 'month_number')
self.assertIsNotNone(test_cube_a)
self.assertEqual(test_cube_a.coord('month_number').points[0], 1)
self.assertEqual(test_cube_a.coord('month_number').points[4], 5)
self.assertEqual(test_cube_a.coord('month_number').points[9], 10)
self.assertEqual(test_cube_a.coord('month_number').points[11], 12)
self.assertEqual(test_cube_a.coord('time').points[0], 3485.0)
self.assertEqual(test_cube_a.coord('time').points[4], 3422.5)
self.assertEqual(test_cube_a.coord('time').points[9], 3575.5)
self.assertEqual(test_cube_a.coord('time').points[11], 3636.5)
def test_aggregate_categorical_month(self):
test_cube_a = common._generate_extended_cube()
test_cube_a = ch.aggregate_categorical(test_cube_a, 'month')
self.assertIsNotNone(test_cube_a)
self.assertEqual(test_cube_a.coord('month').points[0], 'Jan')
self.assertEqual(test_cube_a.coord('month').points[4], 'May')
self.assertEqual(test_cube_a.coord('month').points[9], 'Oct')
self.assertEqual(test_cube_a.coord('month').points[11], 'Dec')
self.assertEqual(test_cube_a.coord('time').points[0], 3485.0)
self.assertEqual(test_cube_a.coord('time').points[4], 3422.5)
self.assertEqual(test_cube_a.coord('time').points[9], 3575.5)
self.assertEqual(test_cube_a.coord('time').points[11], 3636.5)
def test_aggregate_categorical_weekday_number(self):
test_cube_a = common._generate_extended_cube()
test_cube_a = ch.aggregate_categorical(test_cube_a, 'weekday_number')
self.assertIsNotNone(test_cube_a)
self.assertEqual(test_cube_a.coord('weekday_number').points[0], 3)
self.assertEqual(test_cube_a.coord('weekday_number').points[2], 5)
self.assertEqual(test_cube_a.coord('weekday_number').points[4], 0)
self.assertEqual(test_cube_a.coord('weekday_number').points[6], 2)
self.assertEqual(test_cube_a.coord('time').points[0], 3496.5)
self.assertEqual(test_cube_a.coord('time').points[2], 3498.5)
self.assertEqual(test_cube_a.coord('time').points[4], 3500.5)
self.assertEqual(test_cube_a.coord('time').points[6], 3502.5)
def test_aggregate_categorical_weekday(self):
test_cube_a = common._generate_extended_cube()
test_cube_a = ch.aggregate_categorical(test_cube_a, 'weekday')
self.assertIsNotNone(test_cube_a)
self.assertEqual(test_cube_a.coord('weekday').points[0], 'Thu')
self.assertEqual(test_cube_a.coord('weekday').points[2], 'Sat')
self.assertEqual(test_cube_a.coord('weekday').points[4], 'Mon')
self.assertEqual(test_cube_a.coord('weekday').points[6], 'Wed')
self.assertEqual(test_cube_a.coord('time').points[0], 3496.5)
self.assertEqual(test_cube_a.coord('time').points[2], 3498.5)
self.assertEqual(test_cube_a.coord('time').points[4], 3500.5)
self.assertEqual(test_cube_a.coord('time').points[6], 3502.5)
def test_aggregate_categorical_day_of_month(self):
test_cube_a = common._generate_extended_cube()
test_cube_a = ch.aggregate_categorical(test_cube_a, 'day_of_month')
self.assertEqual(test_cube_a.coord('day_of_month').points[0], 1)
self.assertEqual(test_cube_a.coord('day_of_month').points[5], 6)
self.assertEqual(test_cube_a.coord('day_of_month').points[10], 11)
self.assertEqual(test_cube_a.coord('day_of_month').points[15], 16)
self.assertEqual(test_cube_a.coord('day_of_month').points[20], 21)
self.assertEqual(test_cube_a.coord('day_of_month').points[25], 26)
self.assertEqual(test_cube_a.coord('time').points[0], 3499.5)
self.assertEqual(test_cube_a.coord('time').points[5], 3490.5)
self.assertEqual(test_cube_a.coord('time').points[10], 3495.5)
self.assertEqual(test_cube_a.coord('time').points[15], 3500.5)
