def __init__(self):
self.collector = Collector()
self.validator = Validator()
self.detector = Detector()
self.scanner = Scaner()
self.tentacle = Tentacle()
self.proxyList = Manager().list()
def test_generate_possible_valid_solutions_minimizing_waste_more_people_than_pizzas(
self):
pizzas = {
0: [0, 1, 2],
1: [3, 4, 5],
2: [6, 3, 1],
3: [4, 3, 5],
4: [6, 5]
}
pizzeria = Pizzeria(pizzas, 7)
competition = Competition(1, 2, 1)
num_pizzas = len(pizzeria.pizzas)
validator = Validator(competition, num_pizzas)
num_results = 10
results = generate_pop(num_results, competition, pizzeria)
assert len(results) == num_results
for result in results:
assert validator.validate(result)
remaining_pizzas = num_pizzas - sum([len(x[1:]) for x in result])
remaining_people = competition.total_people - sum(
[x[0] for x in result])
assert remaining_people > remaining_pizzas
class Workstation(object):
def __init__(self):
self.collector = Collector()
self.validator = Validator()
self.detector = Detector()
self.scanner = Scaner()
self.proxyList = Manager().list()
def run_validator(self, proxyList):
self.validator.run(proxyList)
def run_collector(self, proxyList):
self.collector.run(proxyList)
def run_detector(self, *params):
self.detector.run()
def run_scanner(self, *params):
self.scanner.run()
def work(self):
pool = Pool(5)
func = []
for i in MODE:
if MODE[i]:
func.append(eval('self.' + RUN_FUNC[i]))
[pool.apply_async(fun, args=(self.proxyList, )) for fun in func]
pool.close()
logger.info('Workstation process pool starting.....OK')
app.run()
def test_invalid_solution_with_different_competition(self):
num_pizzas = 5
competition = Competition(2, 2, 0)
validator = Validator(competition, num_pizzas)
solution_a = [[2, 2, 4], [2, 1, 3], [2, 0, 3]]
assert not validator.validate(solution_a)
class Workstation(object):
"""
整个项目的启动工作面板
"""
def __init__(self):
self.collector = Collector()
self.validator = Validator()
self.detector = Detector()
self.scanner = Scaner()
self.tentacle = Tentacle()
self.proxyList = Manager().list()
def run_validator(self, proxyList):
self.validator.run(proxyList)
def run_collector(self, proxyList):
self.collector.run(proxyList)
def run_detector(self, *params):
self.detector.run()
def run_scanner(self, *params):
self.scanner.run()
def run_tentacle(self, *params):
self.tentacle.run()
def work(self):
"""
项目启动，根据config中的MODE配置执行对应的部件
这样可以隔离部件功能，耦合性较低。异步多进程执行需要
共享变量，使用了multiprocessing的Manager来生成
共享List.
"""
pool = Pool(5)
func = []
for i in MODE:
if MODE[i]:
func.append(eval('self.' + RUN_FUNC[i]))
[pool.apply_async(fun, args=(self.proxyList, )) for fun in func]
pool.close()
app.run()
def test_mutate_without_available_pizzas(self):
num_pizzas = 4
pizzas = list(range(num_pizzas))
input_result = [[2, 0, 1], [2, 2, 3]]
possible_valid_outcomes = [[[2, 0, 3], [2, 2, 1]],
[[2, 0, 2], [2, 1, 3]],
[[2, 3, 1], [2, 2, 0]],
[[2, 2, 1], [2, 0, 3]]]
competition = Competition(2, 0, 0)
validator = Validator(competition, num_pizzas)
breeder = Breeder(pizzas, validator, competition, 100)
result = breeder.mutate(input_result)
assert result in possible_valid_outcomes
def test_breed_when_no_conflict_possible_and_no_mutation(self):
num_pizzas = 8
pizzas = list(range(num_pizzas))
parent_a = [[2, 0, 1], [2, 2, 3]]
parent_b = [[2, 4, 5], [2, 6, 7]]
expected_childs = [[[2, 0, 1], [2, 6, 7]], [[2, 4, 5], [2, 2, 3]],
[[2, 2, 3], [2, 4, 5]], [[2, 6, 7], [2, 0, 1]]]
competition = Competition(2, 0, 0)
validator = Validator(competition, num_pizzas)
breeder = Breeder(pizzas, validator, competition, 0)
result = breeder.breed(parent_a, parent_b)
assert len(result) == 4
for expected_child in expected_childs:
assert expected_child in result
def test_breed_when_conflict_possible_and_no_extra_pizzas_available(self):
num_pizzas = 5
pizzas = list(range(num_pizzas))
parent_a = [[2, 2, 4], [3, 0, 1, 3]]
parent_b = [[3, 4, 1, 2], [2, 0, 3]]
expected_childs = [[[2, 2, 4], [2, 0, 3]], [[3, 0, 1, 3]],
[[2, 0, 3], [2, 2, 4]], [[3, 4, 1, 2]]]
competition = Competition(2, 2, 0)
validator = Validator(competition, num_pizzas)
breeder = Breeder(pizzas, validator, competition, 0)
result = breeder.breed(parent_a, parent_b)
assert len(result) == 4
for expected_child in expected_childs:
assert expected_child in result
if p_id in available_pizzas.keys():
done = True
order.append(p_id)
del available_pizzas[p_id]
return order
if __name__ == '__main__':
args = parser.parse_args()
if not args.file:
print("Missing input file.")
sys.exit(1)
pizzeria, competition = read_file(args.file)
validator = Validator(competition, len(pizzeria.pizzas))
score_calculator = ScoreCalculator(pizzeria.pizzas)
first_team = args.team
if args.second_team:
second_team = args.second_team
else:
teams = [x for x in range(2, 5) if x != first_team]
second_team = sample(teams, k=1)[0]
teams_list = [first_team, second_team]
teams_list.extend([x for x in range(2, 5) if x not in teams_list])
orders_to_fill = []
pizzas_available = len(pizzeria.pizzas)
manager = multiprocessing.Manager()
pizza_lock = manager.Lock()
def validator(self):
num_pizzas = 30
competition = Competition(1, 2, 1)
validator = Validator(competition, num_pizzas)
yield validator
def parallel_generator(id_p: int, comp: Competition, pizz: Pizzeria,
result: dict) -> None:
random_solution = generate_pop(1, comp, pizz)[0]
result.update({id_p: random_solution})
if __name__ == '__main__':
args = parser.parse_args()
if not args.file:
print("Missing input file.")
sys.exit(1)
pizzeria, competition = read_file(args.file)
validator = Validator(competition, len(pizzeria.pizzas))
solution = []
best_score = args.score
print(f'Initial best score is {best_score} points')
score_calculator = ScoreCalculator(pizzeria.pizzas)
num_cores = multiprocessing.cpu_count()
manager = multiprocessing.Manager()
for i in range(args.max_loops):
print(f'iteration {i + 1} of {args.max_loops}')
result_dict = manager.dict()
jobs = []
def test_verify_and_correct_issues_with_teams(self):
competition_a = Competition(1, 1, 0)
validator_a = Validator(competition_a, 9)
breeder_a = Breeder(list(range(9)), validator_a, competition_a, 0)
solution_a = [[4, 0, 1, 2, 3], [3, 4, 5, 6], [2, 7, 8]]
result_a = breeder_a.verify_and_correct(solution_a)
assert validator_a.validate(result_a)
solution_b = [[3, 0, 1, 2], [3, 4, 5, 6], [2, 7, 8]]
result_b = breeder_a.verify_and_correct(solution_b)
assert validator_a.validate(result_b)
validator_c = Validator(competition_a, 14)
breeder_c = Breeder(list(range(14)), validator_c, competition_a, 0)
solution_c = [[4, 0, 1, 2, 3], [3, 4, 5, 6], [2, 7, 8], [3, 9, 10, 11],
[2, 12, 13]]
result_c = breeder_c.verify_and_correct(solution_c)
assert validator_c.validate(result_c)
competition_d = Competition(2, 3, 0)
validator_d = Validator(competition_d, 14)
breeder_d = Breeder(list(range(14)), validator_d, competition_d, 0)
solution_d = [[4, 0, 1, 2, 3], [3, 4, 5, 6], [2, 7, 8], [3, 9, 10, 11],
[2, 12, 13]]
result_d = breeder_d.verify_and_correct(solution_d)
assert validator_d.validate(result_d)
competition_e = Competition(2, 1, 0)
validator_e = Validator(competition_e, 9)
breeder_e = Breeder(list(range(9)), validator_e, competition_e, 0)
solution_e = [[4, 0, 1, 2, 3], [3, 4, 5, 6], [2, 7, 8]]
result_e = breeder_e.verify_and_correct(solution_e)
assert validator_e.validate(result_e)
competition_f = Competition(10, 0, 0)
validator_f = Validator(competition_d, 7)
breeder_f = Breeder(list(range(7)), validator_f, competition_f, 0)
solution_f = [[4, 0, 1, 2, 3], [3, 4, 5, 6]]
result_f = breeder_f.verify_and_correct(solution_f)
assert validator_f.validate(result_f)
def test_given_known_bad_results_verify_and_correct_provides_a_good_ones(
self):
num_pizzas = 500
pizzas = list(range(num_pizzas))
input_result_a = [[3, 331, 64, 242], [4, 25, 243, 282, 273],
[4, 345, 354, 89, 133], [2, 196, 337], [2, 126, 393],
[3, 434, 329, 312], [3, 438, 255, 37], [2, 73, 94],
[4, 395, 212, 357, 85], [4, 103, 347, 361, 420],
[4, 46, 370, 408, 355], [2, 189, 402], [2, 167, 58],
[3, 81, 314, 66], [3, 324, 235, 261], [2, 252, 16],
[2, 399, 88], [3, 61, 462, 13], [2, 175, 22],
[4, 123, 120, 436, 165], [4, 360, 43, 422, 161],
[2, 367, 431], [2, 225, 233], [2, 263, 130],
[3, 127, 149, 118], [4, 409, 452, 50, 100],
[3, 290, 96, 250], [4, 332, 170, 102, 144],
[4, 77, 135, 351, 137], [2, 311,
7], [3, 84, 456, 450],
[2, 202, 376], [3, 447, 268, 403],
[4, 405, 39, 365, 113], [2, 407, 483], [2, 97, 90],
[2, 117, 156], [3, 205, 301, 340],
[4, 141, 432, 421, 369], [4, 406, 496, 441, 396],
[4, 188, 234, 29, 230], [2, 33, 326],
[3, 249, 371, 404], [4, 315, 353, 426, 287],
[4, 169, 166, 76, 448], [4, 59, 251, 208, 323],
[4, 180, 139, 339, 286], [4, 163, 498, 150, 424],
[4, 283, 152, 442, 327], [3, 391, 151, 368],
[4, 423, 313, 98, 44], [2, 278, 143], [2, 3, 213],
[3, 494, 382, 453], [2, 404, 457], [2, 193, 462],
[2, 472, 127], [2, 104, 51], [3, 380, 270, 409],
[2, 103, 42], [4, 345, 172, 227, 442], [2, 78, 312],
[3, 14, 297, 436], [3, 182, 354, 333],
[4, 107, 491, 215, 108], [3, 339, 275, 490],
[4, 257, 175, 485, 13], [2, 340, 72],
[4, 40, 493, 1, 434], [2, 349, 263],
[3, 328, 402, 220], [3, 320, 212, 408],
[3, 374, 145, 285], [4, 62, 266, 386, 9],
[2, 425, 458], [4, 232, 357, 303, 396],
[2, 443, 336], [3, 298, 166, 300],
[4, 23, 76, 278, 383], [4, 153, 32, 289, 310],
[2, 317, 405], [4, 35, 149, 70, 96],
[4, 65, 331, 356, 467], [4, 15, 159, 377, 198],
[2, 146, 311], [3, 114, 423, 55], [2, 483, 307],
[2, 50, 188], [4, 242, 410, 0, 169],
[4, 375, 264, 6, 134], [3, 391, 238, 20],
[4, 91, 144, 422, 269], [2, 355, 248], [2, 481, 252],
[3, 97, 137, 118], [4, 283, 344, 478, 93],
[4, 459, 187, 274, 499], [4, 31, 260, 183, 7],
[2, 109, 316], [3, 203, 113, 75], [2, 370, 30],
[3, 217, 26, 262], [2, 497, 219], [2, 446, 305],
[3, 230, 60, 225], [3, 47, 58, 315],
[4, 382, 351, 272, 214], [4, 205, 251, 455, 90],
[2, 403, 111], [3, 49, 250, 265], [2, 138, 412],
[2, 430, 341], [4, 46, 57, 191, 249],
[4, 179, 447, 395, 329], [2, 268, 115], [2, 80, 489],
[4, 477, 99, 253, 234], [4, 475, 34, 206, 463],
[3, 125, 63, 98], [2, 417, 202],
[4, 296, 11, 148, 323], [4, 281, 261, 163, 243],
[2, 168, 84], [4, 470, 22, 29, 156],
[3, 200, 379, 439], [2, 19, 392],
[4, 284, 381, 54, 406], [4, 52, 433, 167, 338],
[2, 66, 304], [3, 494, 276, 176], [2, 165, 366],
[3, 407, 44, 259], [4, 369, 254, 3, 286],
[4, 143, 368, 197, 209], [2, 437, 233],
[3, 129, 290, 279], [4, 213, 74, 498, 309],
[3, 117, 372, 207], [2, 452, 461], [2, 385, 150],
[3, 362, 92, 8], [2, 152, 347],
[4, 43, 384, 387, 469], [4, 401, 449, 473, 236],
[4, 273, 390, 81, 210], [2, 480, 25],
[4, 465, 448, 394, 432], [3, 218, 221, 495],
[3, 431, 5, 364], [3, 293, 361, 294],
[4, 415, 352, 420, 178], [4, 128, 318, 271, 24],
[2, 73, 112], [2, 184, 123], [3, 440, 326, 350],
[4, 325, 343, 301, 267], [4, 208, 132, 195, 229],
[3, 241, 466, 216], [3, 314, 464, 426],
[3, 106, 171, 85], [3, 67, 419, 174],
[4, 288, 199, 239, 164], [2, 476, 414],
[4, 237, 474, 438, 486], [3, 363, 151, 157],
[3, 185, 397, 64], [3, 445, 133, 139],
[3, 131, 2, 228], [3, 95, 223, 94], [2, 196, 456]]
input_result_b = [[3, 485, 267, 353], [3, 454, 483, 464],
[2, 458, 350], [4, 449, 380, 448, 444],
[4, 275, 285, 78, 192], [4, 309, 423, 163, 439],
[4, 436, 17, 434, 290], [4, 34, 95, 388, 160],
[4, 481, 3, 203, 181], [2, 431, 71],
[4, 328, 430, 426, 424], [4, 395, 280, 248, 276],
[4, 26, 178, 55, 491], [2, 36, 223], [2, 255, 253],
[4, 35, 413, 183, 398], [3, 409, 142, 234],
[4, 310, 106, 453, 281], [2, 359, 408],
[4, 113, 497, 479, 297], [2, 342, 120],
[3, 264, 389, 9], [3, 272, 254, 446], [2, 488, 405],
[4, 364, 392, 401, 295], [4, 474, 484, 232, 130],
[4, 225, 222, 420, 399], [3, 383, 367, 482],
[3, 175, 437, 356], [3, 321, 206, 74],
[4, 30, 229, 351, 152], [3, 382, 468, 227],
[3, 33, 373, 102], [2, 63, 164],
[4, 372, 371, 370, 118], [2, 385,
185], [2, 365, 348],
[4, 217, 116, 418, 226], [2, 39, 394], [2, 425, 101],
[4, 352, 305, 347, 343], [4, 326, 68, 318, 221],
[2, 362, 136], [3, 80, 150, 317], [2, 414, 49],
[4, 231, 313, 170, 306], [3, 304, 301, 107],
[4, 325, 12, 62, 340], [4, 58, 358, 54, 296],
[2, 184, 440], [4, 291, 421, 4, 286], [2, 19, 133],
[3, 18, 278, 274], [2, 268, 7], [3, 260, 75, 360],
[3, 92, 43, 498], [2, 393, 461],
[4, 247, 256, 166, 128], [2, 243, 31], [2, 239, 122],
[2, 238, 314], [4, 357, 96, 104, 87],
[3, 237, 124, 279], [4, 230, 269, 82, 233],
[4, 162, 105, 228, 211], [2, 52, 157], [2, 417, 410],
[3, 208, 441, 378], [4, 200, 345, 199, 197],
[4, 195, 193, 215, 190], [2, 191, 419], [2, 189, 66],
[2, 287, 179], [4, 273, 196, 149,
186], [2, 213, 171], [2, 132, 245],
[4, 169, 2, 374, 159], [4, 218, 155, 246, 422],
[2, 151, 277], [4, 292, 294, 334, 320],
[4, 224, 146, 403, 144], [3, 153, 210, 143],
[4, 427, 67, 182, 400], [3, 140, 252, 270],
[3, 490, 156, 386], [2, 332, 316], [2, 134, 8],
[3, 375, 177, 344], [3, 411, 331, 443], [2, 44, 129],
[2, 127, 114], [2, 470, 257], [4, 97, 307, 93, 492],
[2, 219, 241], [2, 126,
377], [4, 119, 168, 336, 487],
[4, 475, 495, 123, 212], [3, 428, 387, 103],
[2, 117, 384], [2, 349, 249], [4, 429, 108, 476, 94],
[3, 11, 404, 98], [4, 91, 13, 415, 324],
[3, 89, 72, 396], [2, 473, 447], [3, 85, 1, 23],
[3, 83, 457, 242], [2, 61, 339], [2, 81, 354],
[2, 32, 79], [2, 77, 73], [2, 69, 282], [2, 65, 451],
[3, 60, 207, 188], [4, 258, 209, 329, 496],
[3, 56, 0, 368], [3, 311, 53, 259], [3, 50, 499, 90],
[3, 330, 299, 48], [4, 42, 137, 21, 14], [2, 70, 59],
[3, 366, 244, 493], [3, 465, 84, 456], [2, 37, 112],
[3, 38, 29, 433], [2, 361, 494],
[4, 412, 205, 315, 22], [2, 390, 139],
[3, 172, 300, 165], [3, 459, 452, 480],
[4, 76, 10, 204, 416], [4, 161, 131, 109, 47],
[3, 381, 407, 138], [3, 5, 289, 333],
[4, 363, 284, 40, 46], [2, 25, 235], [2, 158, 472],
[2, 154, 263], [4, 462, 355, 176, 327],
[3, 173, 298, 110], [2, 51, 341], [2, 214, 322],
[2, 99, 125], [4, 240, 261, 432, 88], [2, 460, 471],
[2, 28, 467], [3, 236, 141, 180], [3, 262, 100, 335],
[3, 111, 455, 469], [2, 466, 376], [2, 202, 450],
[2, 337, 167], [2, 201, 194], [2, 288, 478],
[2, 302, 220], [4, 24, 266, 369, 391], [2, 435, 265],
[3, 303, 148, 145], [4, 187, 402, 438, 216],
[4, 64, 477, 323, 174], [4, 312, 463, 489, 338],
[4, 293, 283, 346, 45], [2, 198, 121],
[4, 379, 486, 445, 442], [2, 27, 308],
[3, 251, 57, 147], [4, 250, 319, 271, 397],
[4, 135, 20, 86, 115], [3, 15, 41, 406]]
competition = Competition(65, 60, 60)
validator = Validator(competition, num_pizzas)
breeder = Breeder(pizzas, validator, competition, 0)
assert validator.validate(breeder.verify_and_correct(input_result_a))
assert validator.validate(breeder.verify_and_correct(input_result_b))