def setUpClass(cls): start = date(2014, 1, 1) end = date(2014, 1, 31) cls.all_days_of_january = list(daterange(start, end)) end = date(2014, 1, 30) cls.some_days_of_january = list(daterange(start, end)) cls.missing_days_of_january = [date(2014, 1, 31)] cls.today = date.today()
def createTestMetrics(cls, metric_class): parameters = {'container': 1} if issubclass(metric_class, ContainerMetric): pass elif issubclass(metric_class, DomainMetric): parameters['domain'] = 1 else: raise TypeError('Cannot handle {class_name} class'.format( class_name=metric_class.__name__)) start = date(2010, 1, 1) end = date(2010, 12, 31) cls.test_metrics = [] for day in daterange(start, end): json = [] for i in xrange(5): json.append([random.randint(1, 100), random.randint(1, 100)]) parameters['day'] = day.day parameters['month'] = day.month parameters['year'] = day.year parameters['json'] = json test_metric = metric_class(**parameters) test_metric.save() cls.test_metrics.append(test_metric)
def test_all_days_handles_current_year(self): results = all_days_of(date.today().year) first_of_year = date(self.today.year, 1, 1) all_days_of_the_year_untill_today = list(daterange(first_of_year, self.today)) self.assertEqual(results, all_days_of_the_year_untill_today)
def test_all_days_handles_current_month(self): results = all_days_of(date.today().year, self.today.month) first_of_month = date(self.today.year, self.today.month, 1) all_days_of_the_month_untill_today = list(daterange(first_of_month, self.today)) self.assertEqual(results, all_days_of_the_month_untill_today)