* Included tests for "LocalSearchAlgorithm". #41

* Fixed bug related with exception handling.

jinete/algorithms/heuristics/insertion.py

@@ -7,7 +7,7 @@
     Planning,
 )
 from ...exceptions import (
-    NonFeasiblePlannedTripException,
+    StopPlannedTripIterationException,
 )
 from ..abc import (
     Algorithm,
@@ -46,7 +46,7 @@ def _optimize(self) -> Planning:
         while not crosser.completed:
             try:
                 planned_trip = next(crosser)
-            except NonFeasiblePlannedTripException:
+            except StopPlannedTripIterationException:
                 break
             route = planned_trip.route
             route.append_planned_trip(planned_trip)

tests/test_algorithms/test_heuristics/test_insertion.py

@@ -9,18 +9,33 @@
 
 
 class TestInsertionAlgorithm(unittest.TestCase):
+    job: jit.Job
+    fleet: jit.Fleet
+
+    @classmethod
+    def setUpClass(cls) -> None:
+        cls.job = jit.Job(generate_trips(10), objective_cls=jit.DialARideObjective)
+        cls.fleet = jit.Fleet(generate_vehicles(10))
 
     def test_creation(self):
-        job = jit.Job(generate_trips(10), objective_cls=jit.DialARideObjective)
-        fleet = jit.Fleet(generate_vehicles(10))
         algorithm = jit.InsertionAlgorithm(
             crosser_cls=jit.Crosser,
-            job=job,
-            fleet=fleet,
+            job=self.job,
+            fleet=self.fleet,
         )
         self.assertEqual(algorithm.crosser_cls, jit.Crosser)
-        self.assertEqual(algorithm.job, job)
-        self.assertEqual(algorithm.fleet, fleet)
+        self.assertEqual(algorithm.job, self.job)
+        self.assertEqual(algorithm.fleet, self.fleet)
+
+    def test_optimize(self):
+        algorithm = jit.InsertionAlgorithm(
+            job=self.job,
+            fleet=self.fleet,
+        )
+        result = algorithm.optimize()
+
+        self.assertIsNotNone(result)
+        self.assertIsInstance(result, jit.Result)
 
 
 if __name__ == '__main__':

tests/test_algorithms/test_heuristics/test_local_search.py

@@ -0,0 +1,48 @@
+import unittest
+
+import jinete as jit
+
+from tests.utils import (
+    generate_vehicles,
+    generate_trips,
+)
+
+
+class TestLocalSearchAlgorithm(unittest.TestCase):
+    job: jit.Job
+    fleet: jit.Fleet
+
+    @classmethod
+    def setUpClass(cls) -> None:
+        cls.job = jit.Job(generate_trips(10), objective_cls=jit.DialARideObjective)
+        cls.fleet = jit.Fleet(generate_vehicles(10))
+        algorithm = jit.InsertionAlgorithm(
+            job=cls.job,
+            fleet=cls.fleet,
+        )
+        cls.initial = algorithm.optimize()
+
+    def test_creation(self):
+        algorithm = jit.LocalSearchAlgorithm(
+            initial=self.initial,
+            job=self.job,
+            fleet=self.fleet,
+        )
+        self.assertEqual(algorithm.initial, self.initial)
+        self.assertEqual(algorithm.job, self.job)
+        self.assertEqual(algorithm.fleet, self.fleet)
+
+    def test_optimize(self):
+        algorithm = jit.LocalSearchAlgorithm(
+            initial=self.initial,
+            job=self.job,
+            fleet=self.fleet,
+        )
+        result = algorithm.optimize()
+
+        self.assertIsNotNone(result)
+        self.assertIsInstance(result, jit.Result)
+
+
+if __name__ == '__main__':
+    unittest.main()