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()