class TestLocalSearch(unittest.TestCase):
def setUp(self):
self.flow = [
[0, 1, 2],
[4, 0, 5],
[7, 2, 0]]
self.distance = [
[0, 5, 2],
[1, 0, 1],
[6, 2, 0]]
self.i = Instance(None, self.distance, self.flow)
self.ls_steepest = LocalSearch()
self.ls_greedy = LocalSearch(greedy=True)
self.startpoint = Solution((0, 1, 2))
def test_solve_steepest_with_startpoint(self):
expected = (2, 1, 0)
actual = self.ls_steepest.solve(self.i, self.startpoint).sequence
self.assertEqual(expected, actual)
def test_solve_greedy_with_startpoint(self):
expected = (2, 1, 0)
actual = self.ls_greedy.solve(self.i, self.startpoint).sequence
self.assertEqual(expected, actual)
