class DistanceMatrixTest(unittest.TestCase):
def setUp(self):
self.distances = DistanceMatrix()
def test_distance_for_same_system(self):
d = self.distances.getDistance('Arouca', 'Arouca')
self.assertIsNotNone(d)
self.assertEqual(d, 0)
def test_distance_for_different_system(self):
d = self.distances.getDistance('39 Tauri', '47 Ceti')
self.assertIsNotNone(d)
self.assertEqual(d, 99)
def test_find_optimal_systems(self):
o = self.distances.findOptimalSystems('Aerial')
self.assertIsNotNone(o)
def setUp(self):
self.distances = DistanceMatrix()