def test_direct_single(self):
direct(start_date=self.start_date, end_date=self.end_date)
obj = Rice.objects.filter(code='pt_1japt').first()
qs = DailyTran.objects.filter(product=obj,
date__range=(self.start_date, self.end_date))
self.assertEqual(qs.count(), 2)
def test_direct_multi(self):
start_date = datetime.date(year=2017, month=1, day=1)
end_date = datetime.date(year=2017, month=4, day=1)
direct(start_date=start_date, end_date=end_date)
qs = DailyTran.objects.filter(date__range=[start_date, end_date])
self.assertEquals(qs.count(), 1365)
def test_direct_delta(self):
direct(delta=-3)
end_date = datetime.date.today()
start_date = end_date + datetime.timedelta(-3)
count_1 = DailyTran.objects.filter(date__range=(start_date, end_date)).count()
direct(start_date=start_date, end_date=end_date)
count_2 = DailyTran.objects.filter(date__range=(start_date, end_date)).count()
self.assertEqual(count_1, count_2)
direct(start_date=start_date, end_date=end_date)