1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18

from datetime import date

from django.test import TestCase

from .utils import prev_month_range


class BudgetTestCase(TestCase):
    def test_prev_month_range(self):
        self.assertEqual(
            prev_month_range(1, 2025), (date(2024, 12, 1), date(2024, 12, 31))
        )
        self.assertEqual(
            prev_month_range(2, 2025), (date(2025, 1, 1), date(2025, 1, 31))
        )
        self.assertEqual(
            prev_month_range(12, 2025), (date(2025, 11, 1), date(2025, 11, 30))
        )