mirror of
https://github.com/nikdoof/dropbot.git
synced 2025-12-17 19:59:24 +00:00
226 lines
9.3 KiB
Python
226 lines
9.3 KiB
Python
from unittest import TestCase
|
|
from dropbot.map import Map, ship_class_to_range
|
|
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):
|
|
"""Check the package data can be correctly loaded into the map"""
|
|
m = Map.from_json(pkgutil.get_data('dropbot', 'data/map.json'))
|
|
self.assertIsNotNone(m)
|
|
|
|
def test_get_system_name(self):
|
|
"""Test looking up system names from IDs"""
|
|
self.assertEquals(self.map.get_system_name(123), None)
|
|
self.assertEqual(self.map.get_system_name(30000142), 'Jita')
|
|
|
|
def test_get_system_id(self):
|
|
"""Test looking up system ID by name"""
|
|
self.assertEqual(self.map.get_system_id('Llamatron'), None)
|
|
self.assertEqual(self.map.get_system_id('Jita'), 30000142)
|
|
|
|
def test_get_systems(self):
|
|
"""Check partial matching of system names works correctly"""
|
|
self.assertEquals(len(self.map.get_systems('Jita')), 1)
|
|
self.assertEquals(len(self.map.get_systems('Ji')), 7)
|
|
self.assertEquals(len(self.map.get_systems('J')), 2765)
|
|
self.assertEquals(len(self.map.get_systems('123435345345')), 0)
|
|
self.assertEquals(len(self.map.get_systems('jita')), 1)
|
|
self.assertEquals(len(self.map.get_systems('JITA')), 1)
|
|
self.assertEquals(len(self.map.get_systems('JiTa')), 1)
|
|
|
|
def test_system_distance(self):
|
|
"""Test the distance calculator"""
|
|
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):
|
|
"""Test the gate routing system"""
|
|
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_jump_fatigue_standard(self):
|
|
"""Test jump fatigue calculator, 4.119ly jump, no bonus"""
|
|
sys1 = self.map.get_system_id('U-HVIX')
|
|
sys2 = self.map.get_system_id('V-IUEL')
|
|
cooldown, new_fatigue = self.map.jump_fatigue(0, sys1, sys2)
|
|
self.assertEqual(cooldown, 5.12)
|
|
self.assertEqual(new_fatigue, 51.19)
|
|
|
|
def test_jump_fatigue_covertops(self):
|
|
"""Test jump fatigue calculator, 4.119ly jump, covertops being bridged by titan"""
|
|
sys1 = self.map.get_system_id('U-HVIX')
|
|
sys2 = self.map.get_system_id('V-IUEL')
|
|
cooldown, new_fatigue = self.map.jump_fatigue(0, sys1, sys2, ship_class='covertops')
|
|
self.assertEqual(cooldown, 5.12)
|
|
self.assertEqual(new_fatigue, 51.19)
|
|
|
|
def test_jump_fatigue_covertops_covertcyno(self):
|
|
"""Test jump fatigue calculator, 4.119ly jump, covertops being bridged by blops"""
|
|
sys1 = self.map.get_system_id('U-HVIX')
|
|
sys2 = self.map.get_system_id('V-IUEL')
|
|
cooldown, new_fatigue = self.map.jump_fatigue(0, sys1, sys2, ship_class='covertops', jump_type='covert')
|
|
self.assertEqual(cooldown, 3.06)
|
|
self.assertEqual(new_fatigue, 30.59)
|
|
|
|
def test_route_jump_fatigue_covertops_covertcyno(self):
|
|
"""Test jump fatigue calculator, 4.119ly jump, covertops being bridged by blops"""
|
|
sys1 = self.map.get_system_id('U-HVIX')
|
|
sys2 = self.map.get_system_id('V-IUEL')
|
|
res = self.map.route_jump_fatigue([sys1, sys2], 0, ship_class='covertops', jump_type='covert')
|
|
self.assertListEqual(res, [
|
|
{'source': sys1, 'target': sys2, 'cooldown': 3.06, 'fatigue': 30.59,}
|
|
])
|
|
|
|
def test_route_jump_fatigue_covertops_covertcyno_two_jumps(self):
|
|
"""Test route jump fatigue calculator, covertops being bridged by blops for two jumps"""
|
|
sys1 = self.map.get_system_id('U-HVIX')
|
|
sys2 = self.map.get_system_id('RMOC-W')
|
|
sys3 = self.map.get_system_id('Podion')
|
|
res = self.map.route_jump_fatigue([sys1, sys2, sys3], 0, ship_class='covertops', jump_type='covert')
|
|
self.assertListEqual(res, [
|
|
{'source': sys1, 'target': sys2, 'cooldown': 4.77, 'fatigue': 47.72, },
|
|
{'source': sys2, 'target': sys3, 'cooldown': 4.77, 'fatigue': 222.91, },
|
|
])
|
|
|
|
def test_route_jump_fatigue_covertops_covertcyno_three_jumps(self):
|
|
"""Test route jump fatigue calculator, covertops being bridged by blops for three jumps"""
|
|
sys1 = self.map.get_system_id('U-HVIX')
|
|
sys2 = self.map.get_system_id('RMOC-W')
|
|
sys3 = self.map.get_system_id('Podion')
|
|
sys4 = self.map.get_system_id('Hothomouh')
|
|
res = self.map.route_jump_fatigue([sys1, sys2, sys3, sys4], 0, ship_class='covertops', jump_type='covert')
|
|
self.assertListEqual(res, [
|
|
{'source': sys1, 'target': sys2, 'cooldown': 4.77, 'fatigue': 47.72, },
|
|
{'source': sys2, 'target': sys3, 'cooldown': 4.77, 'fatigue': 222.91, },
|
|
{'source': sys3, 'target': sys4, 'cooldown': 22.29, 'fatigue': 786.39, },
|
|
])
|
|
|
|
def test_neighbors_gate(self):
|
|
pass
|
|
|
|
def test_neighbors_jump(self):
|
|
pass
|
|
|
|
def test_jump_bridge_addition(self):
|
|
"""Test addition of a jump bridge"""
|
|
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])
|
|
|
|
# Jump ranges taken from devblog:
|
|
# http://community.eveonline.com/news/dev-blogs/phoebe-travel-change-update/
|
|
|
|
def test_jump_distance_titan(self):
|
|
"""
|
|
The maximum jump range of a titan is 5LY.
|
|
"""
|
|
self.assertEquals(ship_class_to_range('titan', 5), 5)
|
|
|
|
def test_jump_distance_supercarrier(self):
|
|
"""
|
|
The maximum jump range of a supercarrier is 5LY.
|
|
"""
|
|
self.assertEquals(ship_class_to_range('supercarrier', 5), 5)
|
|
|
|
def test_jump_distance_carrier(self):
|
|
"""
|
|
The maximum jump range of a carrier is 5LY.
|
|
"""
|
|
self.assertEquals(ship_class_to_range('carrier', 5), 5)
|
|
|
|
def test_jump_distance_dreadnought(self):
|
|
"""
|
|
The maximum jump range of a dread is 5LY.
|
|
"""
|
|
self.assertEquals(ship_class_to_range('dreadnought', 5), 5)
|
|
|
|
def test_jump_distance_industrial(self):
|
|
"""
|
|
The maximum jump range of a rorqual is 5LY.
|
|
"""
|
|
self.assertEquals(ship_class_to_range('industrial', 5), 5)
|
|
|
|
def test_jump_distance_jumpfreighter(self):
|
|
"""
|
|
The maximum jump range of a JF is 10LY.
|
|
"""
|
|
self.assertEquals(ship_class_to_range('jumpfreighter', 5), 10)
|
|
|
|
def test_jump_distance_blackops(self):
|
|
"""
|
|
The maximum jump range of a bloops is 8LY.
|
|
"""
|
|
self.assertEquals(ship_class_to_range('blackops', 5), 8)
|
|
|
|
def test_jump_distance_skills_titan(self):
|
|
"""
|
|
Test the correct range for titans for each JDC skill level
|
|
"""
|
|
ship_ranges = [2.5, 3.0, 3.5, 4.0, 4.5, 5.0]
|
|
for skill in range(0, 6):
|
|
jump_range = ship_ranges[skill]
|
|
self.assertEquals(ship_class_to_range('titan', skill), jump_range)
|
|
|
|
def test_jump_distance_skills_supercarrier(self):
|
|
"""
|
|
Test the correct range for titans for each JDC skill level
|
|
"""
|
|
ship_ranges = [2.5, 3.0, 3.5, 4.0, 4.5, 5.0]
|
|
for skill in range(0, 6):
|
|
jump_range = ship_ranges[skill]
|
|
self.assertEquals(ship_class_to_range('supercarrier', skill), jump_range)
|
|
|
|
def test_jump_distance_skills_carrier(self):
|
|
"""
|
|
Test the correct range for supercarriers for each JDC skill level
|
|
"""
|
|
ship_ranges = [2.5, 3.0, 3.5, 4.0, 4.5, 5.0]
|
|
for skill in range(0, 6):
|
|
jump_range = ship_ranges[skill]
|
|
self.assertEquals(ship_class_to_range('carrier', skill), jump_range)
|
|
|
|
def test_jump_distance_skills_dreadnought(self):
|
|
"""
|
|
Test the correct range for dreadnoughts for each JDC skill level
|
|
"""
|
|
ship_ranges = [2.5, 3.0, 3.5, 4.0, 4.5, 5.0]
|
|
for skill in range(0, 6):
|
|
jump_range = ship_ranges[skill]
|
|
self.assertEquals(ship_class_to_range('dreadnought', skill), jump_range)
|
|
|
|
def test_jump_distance_skills_industrial(self):
|
|
"""
|
|
Test the correct range for industrials for each JDC skill level
|
|
"""
|
|
ship_ranges = [2.5, 3.0, 3.5, 4.0, 4.5, 5.0]
|
|
for skill in range(0, 6):
|
|
jump_range = ship_ranges[skill]
|
|
self.assertEquals(ship_class_to_range('industrial', skill), jump_range)
|
|
|
|
def test_jump_distance_skills_jumpfreighter(self):
|
|
"""
|
|
Test the correct range for jump freighters for each JDC skill level
|
|
"""
|
|
ship_ranges = [5.0, 6.0, 7.0, 8.0, 9.0, 10.0]
|
|
for skill in range(0, 6):
|
|
jump_range = ship_ranges[skill]
|
|
self.assertEquals(ship_class_to_range('jumpfreighter', skill), jump_range)
|