Update unit tests

Author: ajnebro

src/jmetal/util/archive.py

@@ -1,3 +1,26 @@
+from typing import TypeVar, Generic, List, Optional
+from abc import ABC, abstractmethod
+from scipy.spatial import cKDTree
+
+S = TypeVar('S')
+
+class Archive(Generic[S], ABC):
+    def __init__(self):
+        self.solution_list: List[S] = []
+
+    @abstractmethod
+    def add(self, solution: S) -> bool:
+        pass
+
+    def get(self, index: int) -> S:
+        return self.solution_list[index]
+
+    def size(self) -> int:
+        return len(self.solution_list)
+
+    def get_name(self) -> str:
+        return self.__class__.__name__
+
 import copy
 import random
 import threading

tests/util/test_archive.py

@@ -1,3 +1,7 @@
+import unittest
+import time
+import random
+
 import unittest
 
 from jmetal.core.solution import Solution
@@ -113,6 +117,35 @@ def test_should_adding_three_solutions_work_properly_if_two_of_them_are_equal(se
         self.assertFalse(result)
         self.assertTrue(solution1 in self.archive.solution_list or solution3 in self.archive.solution_list)
 
+    def test_should_add_high_dimensional_solutions(self):
+        """Test behavior with solutions having more than 2 objectives. Only one solution should remain due to dominance logic."""
+        archive = NonDominatedSolutionsArchive()
+        s1 = Solution(1, 5)
+        s1.objectives = [0.0, 1.0, 2.0, 3.0, 4.0]
+        s2 = Solution(1, 5)
+        s2.objectives = [1.0, 2.0, 3.0, 4.0, 5.0]
+        s3 = Solution(1, 5)
+        s3.objectives = [0.5, 1.5, 2.5, 3.5, 4.5]
+        archive.add(s1)
+        archive.add(s2)
+        archive.add(s3)
+        # Only one solution should remain, as the dominance logic removes dominated solutions
+        self.assertEqual(1, archive.size())
+        self.assertTrue(s1 in archive.solution_list or s2 in archive.solution_list or s3 in archive.solution_list)
+
+    def test_should_add_with_numerical_tolerance(self):
+        """Test adding nearly identical solutions (numerical tolerance). Only one should be kept if they are equal within tolerance."""
+        archive = NonDominatedSolutionsArchive(objective_tolerance=1e-5)
+        s1 = Solution(1, 2)
+        s1.objectives = [1.000000, 2.000000]
+        s2 = Solution(1, 2)
+        s2.objectives = [1.000001, 2.000001]
+        archive.add(s1)
+        archive.add(s2)
+        # Only one solution should be kept, as they are equal within the tolerance
+        self.assertEqual(1, archive.size())
+        self.assertTrue(s1 in archive.solution_list or s2 in archive.solution_list)
+
 
 class CrowdingDistanceArchiveTestCases(unittest.TestCase):
     def setUp(self):