mirror of
https://github.com/nikdoof/dropbot.git
synced 2025-12-17 11:49:22 +00:00
60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
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]) |