diff options
Diffstat (limited to 'vvs_efa/tests/__init__.py')
-rw-r--r-- | vvs_efa/tests/__init__.py | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/vvs_efa/tests/__init__.py b/vvs_efa/tests/__init__.py new file mode 100644 index 0000000..bbb1bcf --- /dev/null +++ b/vvs_efa/tests/__init__.py @@ -0,0 +1,39 @@ +# coding: utf-8 + +import unittest +import datetime as dt +from vvs_efa import VVS_EFA + +vvs_efa = VVS_EFA.VVS_EFA() + +class TestConvertNameToId(unittest.TestCase): + + def test_stadtbibliothek_name(self): + self.assertEqual(vvs_efa.convertNameToId("Stadtbibliothek"), "5006116") + + def test_hauptbahnhof_name(self): + self.assertEqual(vvs_efa.convertNameToId("Hauptbahnhof"), "5006118") + + def test_name_with_umlaut(self): + self.assertEqual(vvs_efa.convertNameToId("Möhringen"), "5006169") + + def test_name_with_scharfs(self): + self.assertEqual(vvs_efa.convertNameToId("Vaihinger Straße"), "5000170") + + def test_stadtbibliothek_name_mobile(self): + self.assertEqual(vvs_efa.convertNameToId("Stadtbibliothek", True), "5006116") + + def test_hauptbahnhof_name_mobile(self): + self.assertEqual(vvs_efa.convertNameToId("Hauptbahnhof", True), "5006118") + + def test_empty_name(self): + self.assertEqual(vvs_efa.convertNameToId(""), None) + +class TestGetNextConnections(unittest.TestCase): + + def test_invalid_origin(self): + with self.assertRaises(TypeError): + vvs_efa.getNextConnections("", "Feuersee", dt.datetime(2015, 7, 13, 7, 20), True) + +if __name__ == '__main__': + unittest.main() |