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