def test_categorise_coord_function_time_year_month_day_hour_minute_second(self): delta = date_delta_creator(1, 3, 2, 4, 5, 6) result_function = categorise_coord_function(self.start, self.end, delta, True) expected = np.array([self.u.date2num(datetime.datetime(2000, 8, 16, 2, 2, 33)), self.u.date2num(datetime.datetime(2001, 11, 18, 6, 7, 39)), self.u.date2num(datetime.datetime(2003, 2, 20, 10, 12, 45))]) assert_equal(result_function(self.coord, self.u.date2num(datetime.datetime(2000, 1, 1, 0, 0, 0))), expected[0]) assert_equal(result_function(self.coord, self.u.date2num(datetime.datetime(2001, 7, 3, 0, 0, 0))), expected[1]) assert_equal(result_function(self.coord, self.u.date2num(datetime.datetime(2002, 9, 8, 0, 0, 0))), expected[2])
def test_categorise_coord_function_time_year_month(self): delta = date_delta_creator(1, 1) result_function = categorise_coord_function(self.start, self.end, delta, True) expected = np.array([self.u.date2num(datetime.datetime(2000, 7, 15, 0, 0, 0)), self.u.date2num(datetime.datetime(2001, 8, 15, 0, 0, 0)), self.u.date2num(datetime.datetime(2002, 9, 15, 0, 0, 0))]) assert_equal(result_function(self.coord, self.u.date2num(datetime.datetime(2000, 1, 1, 0, 0, 0))), expected[0]) assert_equal(result_function(self.coord, self.u.date2num(datetime.datetime(2001, 3, 3, 0, 0, 0))), expected[1]) assert_equal(result_function(self.coord, self.u.date2num(datetime.datetime(2002, 5, 8, 0, 0, 0))), expected[2])
def test_categorise_coord_function_with_month_going_past_december(self): start = datetime.datetime(2000, 11, 3) start = self.u.date2num(start) end = datetime.datetime(2004, 11, 3) end = self.u.date2num(end) delta = date_delta_creator(1, 2) result_function = categorise_coord_function(start, end, delta, True) expected = np.array([self.u.date2num(datetime.datetime(2001, 6, 3, 0, 0, 0)), self.u.date2num(datetime.datetime(2002, 8, 3, 0, 0, 0)), self.u.date2num(datetime.datetime(2003, 10, 3, 0, 0, 0))]) assert_equal(result_function(self.coord, self.u.date2num(datetime.datetime(1999, 1, 1, 0, 0, 0))), expected[0]) assert_equal(result_function(self.coord, self.u.date2num(datetime.datetime(2001, 3, 3, 0, 0, 0))), expected[0]) assert_equal(result_function(self.coord, self.u.date2num(datetime.datetime(2002, 3, 3, 0, 0, 0))), expected[1]) assert_equal(result_function(self.coord, self.u.date2num(datetime.datetime(2005, 5, 8, 0, 0, 0))), expected[2])