tests/test_lib_update_public_transport.py

   1 import unittest
   2 from datetime import date, timedelta
   3
   4 from wrpylib.lib_update_public_transport import default_query_date
   5
   6
   7 class TestLibUpdatePublicTransport(unittest.TestCase):
   8     def test_default_query_date(self):
   9         today_expected = [
  10             (date(2022, 1, 1), date(2022, 12, 9)),
  11             (date(2023, 6, 25), date(2023, 12, 11)),
  12             (date(2023, 12, 11), date(2023, 12, 11)),
  13             (date(2023, 12, 12), date(2024, 12, 11)),
  14             (date(2024, 1, 1), date(2024, 12, 11))
  15         ]
  16         for today, expected in today_expected:
  17             with self.subTest(today.isoformat()):
  18                 actual = default_query_date(today)
  19                 self.assertEqual(expected, actual)