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, day_of_time_table_change
   5
   6
   7 class TestLibUpdatePublicTransport(unittest.TestCase):
   8     def test_day_of_time_table_change(self):
   9         self.assertEqual(date(2021, 12, 12), day_of_time_table_change(2021))
  10         self.assertEqual(date(2022, 12, 11), day_of_time_table_change(2022))
  11         self.assertEqual(date(2023, 12, 10), day_of_time_table_change(2023))
  12
  13     def test_default_query_date(self):
  14         today_expected = [
  15             (date(2022, 1, 1), date(2022, 12, 9)),
  16             (date(2023, 6, 25), date(2023, 12, 7)),
  17             (date(2023, 12, 7), date(2023, 12, 7)),
  18             (date(2023, 12, 8), date(2024, 12, 6)),
  19             (date(2023, 12, 9), date(2024, 12, 6)),
  20             (date(2023, 12, 10), date(2024, 12, 6)),
  21             (date(2023, 12, 11), date(2024, 12, 6)),
  22             (date(2024, 1, 1), date(2024, 12, 6))
  23         ]
  24         for today, expected in today_expected:
  25             with self.subTest(today.isoformat()):
  26                 actual = default_query_date(today)
  27                 self.assertEqual(expected, actual)