コード例 #1
0
class TestTransaction(TestCase):
    def setUp(self):
        self.test1 = Transaction()
        self.test1.description = 'Test1'
        self.test1.amount = to_decimal(-5, place='0.01')
        self.test1.save()
        self.test1.refresh_from_db()

        self.test2 = Transaction()
        self.test2.description = 'Test2'
        self.test2.amount = to_decimal(-10, place='0.01')
        self.test2.save()
        self.test2.refresh_from_db()

        self.test3 = Transaction()
        self.test3.description = 'Test3'
        self.test3.amount = to_decimal(1.2, place='0.01')
        self.test3.save()
        self.test3.refresh_from_db()

    def test_today_count(self):
        todays = Transaction.objects.today()
        self.assertEqual(todays.count(), 3)

    def test_today_total(self):
        total = Transaction.objects.today().total()
        self.assertEqual(total, to_decimal(-13.8, place='0.01'))
コード例 #2
0
ファイル: tests.py プロジェクト: knudsonb2/moolah
class TestTransaction(TestCase):
    def setUp(self):
        self.test1 = Transaction()
        self.test1.description = 'Test1'
        self.test1.amount = to_decimal(-5, place='0.01')
        self.test1.save()
        self.test1.refresh_from_db()

        self.test2 = Transaction()
        self.test2.description = 'Test2'
        self.test2.amount = to_decimal(-10, place='0.01')
        self.test2.save()
        self.test2.refresh_from_db()

        self.test3 = Transaction()
        self.test3.description = 'Test3'
        self.test3.amount = to_decimal(1.2, place='0.01')
        self.test3.save()
        self.test3.refresh_from_db()

    def test_today_count(self):
        todays = Transaction.objects.today()
        self.assertEqual(todays.count(), 3)

    def test_today_total(self):
        total = Transaction.objects.today().total()
        self.assertEqual(total, to_decimal(-13.8, place='0.01'))