Basic unit tests, and wercker support.

2025-12-17 19:59:24 +00:00 · 2014-08-14 18:21:54 +01:00
parent de79c27765
commit 62d50cfbfe
6 changed files with 129 additions and 1 deletions
--- a/tests/test_map.py
+++ b/tests/test_map.py
@@ -0,0 +1,60 @@
+from unittest import TestCase
+from dropbot.map import Map
+import pkgutil
+
+class MapTestCase(TestCase):
+
+    def setUp(self):
+        self.map = Map.from_json(pkgutil.get_data('dropbot', 'data/map.json'))
+
+    def test_load_from_package_data(self):
+        m = Map.from_json(pkgutil.get_data('dropbot', 'data/map.json'))
+        self.assertIsNotNone(m)
+
+    def test_get_system_name(self):
+        self.assertEquals(self.map.get_system_name(123), None)
+        self.assertEqual(self.map.get_system_name(30000142), 'Jita')
+
+    def test_get_system_id(self):
+        self.assertEqual(self.map.get_system_id('Llamatron'), None)
+        self.assertEqual(self.map.get_system_id('Jita'), 30000142)
+
+    def test_get_systems(self):
+        self.assertEquals(len(self.map.get_systems('Jita')), 1)
+        self.assertEquals(len(self.map.get_systems('Ji')), 14)
+        self.assertEquals(len(self.map.get_systems('J')), 576)
+        self.assertEquals(len(self.map.get_systems('123435345345')), 0)
+
+    def test_system_distance(self):
+        self.assertEqual(self.map.system_distance(30000142, 30000144), 2.10268108033618)
+        self.assertEqual(self.map.system_distance(30000142, 30000222), 9.334275248404591)
+        self.assertEqual(self.map.system_distance(30000142, 30000536), 39.15289747780095)
+        self.assertRaises(Exception, self.map.system_distance, (1, 2))
+
+    def test_route_gate(self):
+        r = self.map.route_gate(30001161, 30001198)
+        self.assertEqual(len(r), 9)
+        self.assertListEqual(r, [30001161, 30001158, 30001160, 30001154, 30001157, 30001155, 30001156, 30001162, 30001198])
+
+    def test_route_jump(self):
+        pass
+
+    def test_route_jump_distance(self):
+        pass
+
+    def test_route_jump_isotopes(self):
+        pass
+
+    def test_neighbors_gate(self):
+        pass
+
+    def test_neighbors_jump(self):
+        pass
+
+    def test_jump_bridge_addition(self):
+        # HED-GP to GE-8
+        self.assertGreater(len(self.map.route_gate(30001161, 30001198)), 2)
+        self.map.add_jumpbridge(30001161, 30001198)
+        r = self.map.route_gate(30001161, 30001198)
+        self.assertEqual(len(r), 2)
+        self.assertListEqual(r, [30001161, 30001198])