class RmnkTest(unittest.TestCase):
"""Test the 'rmnk' problem."""
def setUp(self):
"""Init"""
project_path = os.path.dirname(os.path.abspath(__file__))
self.problem = Rmnk(project_path +
'/../data/instances/rmnk_0_2_100_1_0.dat')
def test_instance(self):
"""Test parameters"""
self.assertEqual(self.problem.rho, 0)
self.assertEqual(self.problem.m, 2)
self.assertEqual(self.problem.n, 100)
self.assertEqual(self.problem.k, 1)
def test_generate_random_solution(self):
"""Test the function 'generate_random_solution'"""
random_solution = self.problem.generate_random_solution()
self.assertEqual(len(random_solution.decision_vector), 100)
self.assertEqual(len(random_solution.F), 2)
for item in random_solution.decision_vector:
self.assertIn(item, [0, 1])
for function_i in random_solution.F:
self.assertLess(function_i, 0)
self.assertGreater(function_i, -1)
def test_generate_solution(self):
"""Test the function 'generate_solution'"""
array = []
for i in range(100):
array.append(random.randint(0, 1))
solution = self.problem.evaluate(array)
self.assertEqual(len(solution.decision_vector), 100)
self.assertEqual(len(solution.F), 2)
for item in solution.decision_vector:
self.assertIn(item, [0, 1])
for function_i in solution.F:
self.assertLess(function_i, 0)
self.assertGreater(function_i, -1)
def test_evaluation(self):
"""Test evaluation"""
array = []
for i in range(100):
array.append(1)
solution = self.problem.evaluate(array)
self.assertEqual(solution.F[0], -0.4884468640000001)
self.assertEqual(solution.F[1], -0.4930223456999998)
class ScalarizingTest(unittest.TestCase):
"""Test aggregation functions."""
def setUp(self):
"""Init"""
project_path = os.path.dirname(os.path.abspath(__file__))
self.problem = Rmnk(project_path + '/data/instances/rmnk_0_2_100_1_0.dat')
weight_file = project_path + "/data/weights/SOBOL-2objs-10wei.ws"
self.weights = generate_weight_vectors(weight_file, False)
array = []
for i in range(100):
array.append(1)
self.solution = self.problem.evaluate(array)
self.z = [1, 1]
def test_tchebycheff(self):
"""Test Tchebycheff"""
value1 = Tchebycheff().run(self.solution, 2, self.weights, 3, self.z)
value2 = Tchebycheff().run(self.solution, 2, self.weights, 8, self.z)
self.assertEqual(value1, 1.1197667592749998)
self.assertEqual(value2, 1.3023910060000001)
self.assertFalse(Tchebycheff().is_better(value1, value2))
self.assertTrue(Tchebycheff().is_better(value2, value1))
def test_weighted_sum(self):
"""Test Weighted Sum"""
value1 = WeightedSum().run(self.solution, 2, self.weights, 3, self.z)
value2 = WeightedSum().run(self.solution, 2, self.weights, 8, self.z)
self.assertEqual(value1, -0.4918784752749999)
self.assertEqual(value2, -0.4890187992125)
self.assertFalse(WeightedSum().is_better(value1, value2))
self.assertTrue(WeightedSum().is_better(value2, value1))