def test_06_include_budget_with_records(self): budget = Budget( user=self.user, amount=100, tags_type='INCL', start_date=datetime.date(2016, 7, 1), ) budget.tags = ['books'] budget.save() # add record record = Record(user=self.user, transaction_type='EXP', amount='10') record.tags = ['books', 'cafe'] record.save() self.assertEqual(budget.spent, 10) self.assertEqual(budget.left, 90) self.assertEqual(budget.left_average_per_day, Decimal('4.50')) self.assertEqual(budget.average_per_day, Decimal('3.22'))