import datetime
import unittest
import seepark_web


class SeeparkWebTestCase(unittest.TestCase):
    def test_first_of_month(self):
        for d in [1, 8, 25, 28]:
            date_a = datetime.date(2019, 2, d)
            date_b = seepark_web.first_of_month(date_a, 1)
            self.assertEqual(date_b, datetime.date(2019, 3, 1))

            date_b = seepark_web.first_of_month(date_a, -1)
            self.assertEqual(date_b, datetime.date(2019, 1, 1))

            date_b = seepark_web.first_of_month(date_a, 0)
            self.assertEqual(date_b, datetime.date(2019, 2, 1))