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     vao_ext_id_to_ifopt_stop_id
   6
   7
   8 class TestLibUpdatePublicTransport(unittest.TestCase):
   9     def test_day_of_time_table_change(self):
  10         self.assertEqual(date(2021, 12, 12), day_of_time_table_change(2021))
  11         self.assertEqual(date(2022, 12, 11), day_of_time_table_change(2022))
  12         self.assertEqual(date(2023, 12, 10), day_of_time_table_change(2023))
  13
  14     def test_default_query_date(self):
  15         today_expected = [
  16             (date(2022, 1, 1), date(2022, 12, 9)),
  17             (date(2023, 6, 25), date(2023, 12, 7)),
  18             (date(2023, 12, 7), date(2023, 12, 7)),
  19             (date(2023, 12, 8), date(2024, 12, 6)),
  20             (date(2023, 12, 9), date(2024, 12, 6)),
  21             (date(2023, 12, 10), date(2024, 12, 6)),
  22             (date(2023, 12, 11), date(2024, 12, 6)),
  23             (date(2024, 1, 1), date(2024, 12, 6))
  24         ]
  25         for today, expected in today_expected:
  26             with self.subTest(today.isoformat()):
  27                 actual = default_query_date(today)
  28                 self.assertEqual(expected, actual)
  29
  30     def test_vao_ext_id_to_ifopt_stop_id(self):
  31         self.assertEqual("at:47:61646", vao_ext_id_to_ifopt_stop_id("476164600"))
  32         self.assertEqual("at:48:1722", vao_ext_id_to_ifopt_stop_id("480172200"))