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):
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"))