def test_calc_fitness(mocked_init, mocked_raise_on):
planner = GAPlanner(None, None, None)
planner._logger = ru.Logger('dummy')
planner._campaign = []
planner._population = [[1, 3, 5, 6, -1, 2, 7, -1, 4, 8, 9],
[1, 5, 6, -1, 3, 2, 7, -1, 4, 8, 9]]
planner._est_txs = [[10, 10, 10], [10, 10, 10], [10, 10, 10], [10, 10, 10],
[10, 10, 10], [10, 10, 10], [10, 10, 10], [10, 10, 10],
[10, 10, 10], [10, 10, 10]]
planner._abs_fitness_term = 30
planner._population_size = 2
planner._fitness = []
for i in range(9):
workflow = {
'description': 'W%s' % str(i + 1),
'id': i + 1,
'num_oper': 10,
'requirements': None
}
planner._campaign.append(workflow)
planner._resources = [{
'id': 1,
'performance': 1
}, {
'id': 2,
'performance': 1
}, {
'id': 3,
'performance': 1
}]
planner._calc_fitness()
assert planner._fitness[1] == 1
assert planner._fitness[0] == (1 / math.sqrt(200))
def test_selection(mocked_init, mocked_raise_on):
planner = GAPlanner(None, None, None)
planner._logger = ru.Logger('dummy')
planner._campaign = {'campaign': []}
planner._population = [[1, 3, 5, 6, -1, 2, 7, -1, 4, 8, 9],
[1, 5, 6, -1, 3, 2, 7, -1, 4, 8, 9]]
planner._population_size = 2
planner._fitness = [1, 0.5]
parents = planner._selection()
assert (parents[0] == [1, 3, 5, 6, -1, 2, 7, -1, 4, 8, 9]
or parents[0] == [1, 5, 6, -1, 3, 2, 7, -1, 4, 8, 9])
def test_initialize_population(mocked_init, mocked_raise_on):
population = [[3, -1, 1, 2, 4, 8, -1, 5, 6, 7, 9, 10],
[6, 8, -1, 1, 2, 3, 4, 5, 10, -1, 7, 9],
[2, 4, -1, 5, 9, -1, 1, 3, 6, 7, 8, 10],
[1, 2, 6, -1, 4, 7, 9, -1, 3, 5, 8, 10],
[3, 5, -1, 4, 7, 10, -1, 1, 2, 6, 8, 9]]
planner = GAPlanner(None, None, None)
planner._logger = ru.Logger('dummy')
planner._population = []
planner._population_size = 5
planner._est_txs = [[1, 0.5, 0.3], [2, 1, 0.7], [3, 1.5, 1], [4, 2, 1.3],
[5, 2.5, 1.7], [6, 3, 2], [7, 3.5, 2.3], [8, 4, 2.7],
[9, 4.5, 3], [10, 5, 3.3]]
workflows = [{
'description': 'W1',
'id': 1,
'num_oper': 1,
'requirements': None
}, {
'description': 'W2',
'id': 2,
'num_oper': 2,
'requirements': None
}, {
'description': 'W3',
'id': 3,
'num_oper': 3,
'requirements': None
}, {
'description': 'W4',
'id': 4,
'num_oper': 4,
'requirements': None
}, {
'description': 'W5',
'id': 5,
'num_oper': 5,
'requirements': None
}, {
'description': 'W6',
'id': 6,
'num_oper': 6,
'requirements': None
}, {
'description': 'W7',
'id': 7,
'num_oper': 7,
'requirements': None
}, {
'description': 'W8',
'id': 8,
'num_oper': 8,
'requirements': None
}, {
'description': 'W9',
'id': 9,
'num_oper': 9,
'requirements': None
}, {
'description': 'W10',
'id': 10,
'num_oper': 10,
'requirements': None
}]
resources = [{
'id': 1,
'performance': 1
}, {
'id': 2,
'performance': 2
}, {
'id': 3,
'performance': 3
}]
planner._initialize_population(workflows=workflows,
resources=resources,
random_init=0.5,
start_time=None)
assert planner._population == population