Add test_vao_ext_id_to_ifopt_stop_id().

[philipp/winterrodeln/wrpylib.git] / tests / test_lib_update_public_transport.py

diff --git a/tests/test_lib_update_public_transport.py b/tests/test_lib_update_public_transport.py
index 53ce8197cd0969ac13646e37cecabbce45ec7d64..246fee0e0af9b16dbd315811bcbadfcf74285199 100644 (file)
--- a/tests/test_lib_update_public_transport.py
+++ b/tests/test_lib_update_public_transport.py
@@ -1,7 +1,8 @@
 import unittest
 from datetime import date, timedelta
 
-from wrpylib.lib_update_public_transport import default_query_date, day_of_time_table_change
+from wrpylib.lib_update_public_transport import default_query_date, day_of_time_table_change, \
+    vao_ext_id_to_ifopt_stop_id
 
 
 class TestLibUpdatePublicTransport(unittest.TestCase):
@@ -13,12 +14,19 @@ class TestLibUpdatePublicTransport(unittest.TestCase):
     def test_default_query_date(self):
         today_expected = [
             (date(2022, 1, 1), date(2022, 12, 9)),
-            (date(2023, 6, 25), date(2023, 12, 11)),
-            (date(2023, 12, 11), date(2023, 12, 11)),
-            (date(2023, 12, 12), date(2024, 12, 11)),
-            (date(2024, 1, 1), date(2024, 12, 11))
+            (date(2023, 6, 25), date(2023, 12, 7)),
+            (date(2023, 12, 7), date(2023, 12, 7)),
+            (date(2023, 12, 8), date(2024, 12, 6)),
+            (date(2023, 12, 9), date(2024, 12, 6)),
+            (date(2023, 12, 10), date(2024, 12, 6)),
+            (date(2023, 12, 11), date(2024, 12, 6)),
+            (date(2024, 1, 1), date(2024, 12, 6))
         ]
         for today, expected in today_expected:
             with self.subTest(today.isoformat()):
                 actual = default_query_date(today)
                 self.assertEqual(expected, actual)
+
+    def test_vao_ext_id_to_ifopt_stop_id(self):
+        self.assertEqual("at:47:61646", vao_ext_id_to_ifopt_stop_id("476164600"))
+        self.assertEqual("at:48:1722", vao_ext_id_to_ifopt_stop_id("480172200"))