def display(self):
"""
陈列主算法
:return:
"""
begin_time = time.time()
print(self.shelf)
# 区域处理
levelid_to_displaygoods_list = {}
for level in self.shelf.levels:
levelid_to_displaygoods_list[
level.level_id] = level.display_goods_list
area_manager = AreaManager(self.shelf, levelid_to_displaygoods_list,
self.choose_goods_list)
self.best_candidate_shelf = area_manager.calculate_candidate_shelf()
end_time = time.time()
self.display_calculate_time = int(end_time - begin_time)
def test_special_level(self):
shelf = Shelf(1, 1800, 600, 300)
shelf.levels.append(Level(shelf, 0, 360, 300))
shelf.levels.append(Level(shelf, 1, 360, 300))
shelf.levels.append(Level(shelf, 2, 360, 300))
levelid_to_displaygoods_list = {
0: [
DisplayGoods(TestGoods('c2_1', 'c3_1', '1', 200, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_2', '2', 200, 80, 9, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '3', 200, 40, 8, 0, 0))
],
1: [
DisplayGoods(TestGoods('c2_4', 'c3_6', '6', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_3', 'c3_5', '5', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_4', '4', 150, 80, 10, 0, 0))
],
2: [
DisplayGoods(TestGoods('c2_4', 'c3_7', '7', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_5', 'c3_8', '8', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_6', 'c3_9', '9', 100, 80, 10, 0, 0))
],
}
shelf.levels[0].display_goods_list = levelid_to_displaygoods_list[0]
shelf.levels[1].display_goods_list = levelid_to_displaygoods_list[1]
shelf.levels[2].display_goods_list = levelid_to_displaygoods_list[2]
choose_goods_list = [
TestGoods('c2_1', 'c3_1', '101', 100, 80, 20, 1, 0), # 上架
]
area_manager = AreaManager(shelf, levelid_to_displaygoods_list,
choose_goods_list)
area_manager._first_born_areas()
self.assertEqual(len(area_manager.area_list), 9)
for area in area_manager.area_list:
print(area)
area_manager._second_combine_areas()
self.assertEqual(len(area_manager.area_list), 6)
# print('\n')
for area in area_manager.area_list:
print(area)
def test_bottom_level(self):
shelf = Shelf(1, 1800, 600, 300)
shelf.levels.append(Level(shelf, 0, 360, 300))
shelf.levels.append(Level(shelf, 1, 360, 300))
shelf.levels.append(Level(shelf, 2, 360, 300))
shelf.levels.append(Level(shelf, 3, 360, 300))
shelf.levels.append(Level(shelf, 4, 360, 300))
levelid_to_displaygoods_list = {
0: [
DisplayGoods(TestGoods('c2_1', 'c3_1', '1', 200, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_1', '2', 200, 80, 9, 0, 0),
face_num=2),
DisplayGoods(TestGoods('c2_1', 'c3_1', '3', 200, 40, 8, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_1', '4', 200, 40, 7, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_2', '5', 200, 20, 6, 0, 0),
face_num=2),
DisplayGoods(TestGoods('c2_1', 'c3_2', '6', 200, 40, 5, 0, 0),
face_num=3),
DisplayGoods(TestGoods('c2_1', 'c3_3', '7', 150, 40, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '8', 150, 40, 10, 0, 0))
],
1: [
DisplayGoods(TestGoods('c2_1', 'c3_3', '18', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '17', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '16', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '15', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '14', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '13', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '12', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '11', 150, 40, 10, 0,
0))
],
2: [
DisplayGoods(TestGoods('c2_1', 'c3_3', '21', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '22', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '23', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '24', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '25', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '26', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '27', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '28', 100, 40, 10, 0,
0))
],
3: [
DisplayGoods(TestGoods('c2_2', 'c3_6', '38', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '37', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '36', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '35', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '34', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '33', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '32', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '31', 100, 40, 10, 0,
0))
],
4: [
DisplayGoods(TestGoods('c2_2', 'c3_6', '41', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '42', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_7', '43', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_7', '44', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_7', '45', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_7', '46', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_7', '47', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_8', '48', 100, 40, 10, 0,
0))
],
}
shelf.levels[0].display_goods_list = levelid_to_displaygoods_list[0]
shelf.levels[1].display_goods_list = levelid_to_displaygoods_list[1]
shelf.levels[2].display_goods_list = levelid_to_displaygoods_list[2]
shelf.levels[3].display_goods_list = levelid_to_displaygoods_list[3]
shelf.levels[4].display_goods_list = levelid_to_displaygoods_list[4]
choose_goods_list = [
TestGoods('c2_1', 'c3_1', '101', 100, 80, 20, 1, 0), # 上架
TestGoods('c2_1', 'c3_1', '102', 100, 80, 20, 1, 0), # 上架
TestGoods('c2_1', 'c3_1', '103', 100, 160, 20, 1, 0), # 上架
TestGoods('c2_1', 'c3_1', '4', 100, 40, 7, 2, 0), # 下架
TestGoods('c2_1', 'c3_2', '5', 100, 40, 6, 2, 0), # 下架
]
area_manager = AreaManager(shelf, levelid_to_displaygoods_list,
choose_goods_list)
area_manager._first_born_areas()
self.assertEqual(len(area_manager.area_list), 8)
self.assertEqual(area_manager.area_list[0].start_level_id, 0)
self.assertEqual(area_manager.area_list[0].start_width, 0)
self.assertEqual(area_manager.area_list[0].end_level_id, 0)
self.assertEqual(area_manager.area_list[0].end_width, 320)
self.assertEqual(len(area_manager.area_list[0].child_area_list), 1)
self.assertEqual(area_manager.area_list[0].child_area_list[0].level_id,
0)
self.assertEqual(
len(area_manager.area_list[0].child_area_list[0].display_goods_list
), 4)
self.assertEqual(area_manager.area_list[1].start_level_id, 0)
self.assertEqual(area_manager.area_list[1].start_width, 320)
self.assertEqual(area_manager.area_list[1].end_level_id, 0)
self.assertEqual(area_manager.area_list[1].end_width, 480)
self.assertEqual(len(area_manager.area_list[1].child_area_list), 1)
self.assertEqual(area_manager.area_list[1].child_area_list[0].level_id,
0)
self.assertEqual(
len(area_manager.area_list[1].child_area_list[0].display_goods_list
), 2)
self.assertEqual(area_manager.area_list[2].start_level_id, 0)
self.assertEqual(area_manager.area_list[2].start_width, 480)
self.assertEqual(area_manager.area_list[2].end_level_id, 2)
self.assertEqual(area_manager.area_list[2].end_width, 160)
self.assertEqual(len(area_manager.area_list[2].child_area_list), 3)
self.assertEqual(area_manager.area_list[2].child_area_list[0].level_id,
0)
self.assertEqual(
len(area_manager.area_list[2].child_area_list[0].display_goods_list
), 2)
self.assertEqual(area_manager.area_list[2].child_area_list[1].level_id,
1)
self.assertEqual(
len(area_manager.area_list[2].child_area_list[1].display_goods_list
), 8)
self.assertEqual(area_manager.area_list[2].child_area_list[2].level_id,
2)
self.assertEqual(
len(area_manager.area_list[2].child_area_list[2].display_goods_list
), 2)
self.assertEqual(area_manager.area_list[3].start_level_id, 2)
self.assertEqual(area_manager.area_list[3].start_width, 160)
self.assertEqual(area_manager.area_list[3].end_level_id, 2)
self.assertEqual(area_manager.area_list[3].end_width, 480)
self.assertEqual(len(area_manager.area_list[3].child_area_list), 1)
self.assertEqual(area_manager.area_list[3].child_area_list[0].level_id,
2)
self.assertEqual(
len(area_manager.area_list[3].child_area_list[0].display_goods_list
), 4)
self.assertEqual(area_manager.area_list[4].start_level_id, 2)
self.assertEqual(area_manager.area_list[4].start_width, 480)
self.assertEqual(area_manager.area_list[4].end_level_id, 3)
self.assertEqual(area_manager.area_list[4].end_width, 120)
self.assertEqual(len(area_manager.area_list[4].child_area_list), 2)
self.assertEqual(area_manager.area_list[4].child_area_list[0].level_id,
2)
self.assertEqual(
len(area_manager.area_list[4].child_area_list[0].display_goods_list
), 2)
self.assertEqual(area_manager.area_list[4].child_area_list[1].level_id,
3)
self.assertEqual(
len(area_manager.area_list[4].child_area_list[1].display_goods_list
), 2)
self.assertEqual(area_manager.area_list[5].start_level_id, 3)
self.assertEqual(area_manager.area_list[5].start_width, 120)
self.assertEqual(area_manager.area_list[5].end_level_id, 4)
self.assertEqual(area_manager.area_list[5].end_width, 160)
self.assertEqual(len(area_manager.area_list[5].child_area_list), 2)
self.assertEqual(area_manager.area_list[5].child_area_list[0].level_id,
3)
self.assertEqual(
len(area_manager.area_list[5].child_area_list[0].display_goods_list
), 6)
self.assertEqual(area_manager.area_list[5].child_area_list[1].level_id,
4)
self.assertEqual(
len(area_manager.area_list[5].child_area_list[1].display_goods_list
), 2)
self.assertEqual(area_manager.area_list[6].start_level_id, 4)
self.assertEqual(area_manager.area_list[6].start_width, 160)
self.assertEqual(area_manager.area_list[6].end_level_id, 4)
self.assertEqual(area_manager.area_list[6].end_width, 560)
self.assertEqual(len(area_manager.area_list[6].child_area_list), 1)
self.assertEqual(area_manager.area_list[6].child_area_list[0].level_id,
4)
self.assertEqual(
len(area_manager.area_list[6].child_area_list[0].display_goods_list
), 5)
self.assertEqual(area_manager.area_list[7].start_level_id, 4)
self.assertEqual(area_manager.area_list[7].start_width, 560)
self.assertEqual(area_manager.area_list[7].end_level_id, 4)
self.assertEqual(area_manager.area_list[7].end_width, 600)
self.assertEqual(len(area_manager.area_list[7].child_area_list), 1)
self.assertEqual(area_manager.area_list[7].child_area_list[0].level_id,
4)
self.assertEqual(
len(area_manager.area_list[7].child_area_list[0].display_goods_list
), 1)
for area in area_manager.area_list:
print(area)
area_manager._second_combine_areas()
self.assertEqual(len(area_manager.area_list), 4)
self.assertEqual(area_manager.area_list[0].start_level_id, 0)
self.assertEqual(area_manager.area_list[0].start_width, 0)
self.assertEqual(area_manager.area_list[0].end_level_id, 0)
self.assertEqual(area_manager.area_list[0].end_width, 480)
self.assertEqual(len(area_manager.area_list[0].child_area_list), 2)
self.assertEqual(
len(area_manager.area_list[0].child_area_list[0].display_goods_list
), 4)
self.assertEqual(
len(area_manager.area_list[0].child_area_list[1].display_goods_list
), 2)
self.assertEqual(area_manager.area_list[1].start_level_id, 0)
self.assertEqual(area_manager.area_list[1].start_width, 480)
self.assertEqual(area_manager.area_list[1].end_level_id, 2)
self.assertEqual(area_manager.area_list[1].end_width, 160)
self.assertEqual(len(area_manager.area_list[1].child_area_list), 3)
self.assertEqual(
len(area_manager.area_list[1].child_area_list[0].display_goods_list
), 2)
self.assertEqual(
len(area_manager.area_list[1].child_area_list[1].display_goods_list
), 8)
self.assertEqual(
len(area_manager.area_list[1].child_area_list[2].display_goods_list
), 2)
self.assertEqual(area_manager.area_list[2].start_level_id, 2)
self.assertEqual(area_manager.area_list[2].start_width, 160)
self.assertEqual(area_manager.area_list[2].end_level_id, 3)
self.assertEqual(area_manager.area_list[2].end_width, 120)
self.assertEqual(len(area_manager.area_list[2].child_area_list), 3)
self.assertEqual(
len(area_manager.area_list[2].child_area_list[0].display_goods_list
), 4)
self.assertEqual(
len(area_manager.area_list[2].child_area_list[1].display_goods_list
), 2)
self.assertEqual(
len(area_manager.area_list[2].child_area_list[2].display_goods_list
), 2)
self.assertEqual(area_manager.area_list[3].start_level_id, 3)
self.assertEqual(area_manager.area_list[3].start_width, 120)
self.assertEqual(area_manager.area_list[3].end_level_id, 4)
self.assertEqual(area_manager.area_list[3].end_width, 600)
self.assertEqual(len(area_manager.area_list[3].child_area_list), 4)
self.assertEqual(
len(area_manager.area_list[3].child_area_list[0].display_goods_list
), 6)
self.assertEqual(
len(area_manager.area_list[3].child_area_list[1].display_goods_list
), 2)
self.assertEqual(
len(area_manager.area_list[3].child_area_list[2].display_goods_list
), 5)
self.assertEqual(
len(area_manager.area_list[3].child_area_list[3].display_goods_list
), 1)
# print('\n')
for area in area_manager.area_list:
print(area)
def test_no_two_bottom_level(self):
shelf = Shelf(1, 1800, 600, 300)
shelf.levels.append(Level(shelf, 0, 360, 300))
shelf.levels.append(Level(shelf, 1, 360, 300))
shelf.levels.append(Level(shelf, 2, 360, 300))
shelf.levels.append(Level(shelf, 3, 360, 300))
shelf.levels.append(Level(shelf, 4, 360, 300))
levelid_to_displaygoods_list = {
2: [DisplayGoods(TestGoods('c2_1', 'c3_3', '21', 150, 80, 10, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '22', 150, 80, 10, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '23', 100, 80, 10, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '24', 100, 80, 9, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '25', 100, 80, 8, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '26', 100, 80, 7, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '27', 100, 80, 6, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '28', 100, 40, 5, 0, 0))],
}
shelf.levels[2].display_goods_list = levelid_to_displaygoods_list[2]
choose_goods_list = []
area_manager = AreaManager(shelf, levelid_to_displaygoods_list, choose_goods_list)
area_manager._first_born_areas()
area_manager._second_combine_areas()
area_manager._prepare_area_base_data()
area_manager._prepare_area_calculate_data()
area_manager._print_choose_goods_info()
area_manager._arrange_areas()
best_candidate_shelf = area_manager.arrange_goods()
level_to_display_goods_name_str = get_level_to_display_goods_name(best_candidate_shelf)
self.assertEqual(level_to_display_goods_name_str[0], 'null,')
self.assertEqual(level_to_display_goods_name_str[1], 'null,')
self.assertEqual(level_to_display_goods_name_str[2], '21,22,23,24,25,26,27,28,')
print(shelf)
print(best_candidate_shelf)
def test_bottom_middle_level_same_psd_amount(self):
shelf = Shelf(1, 1800, 600, 300)
shelf.levels.append(Level(shelf, 0, 360, 300))
shelf.levels.append(Level(shelf, 1, 360, 300))
shelf.levels.append(Level(shelf, 2, 360, 300))
shelf.levels.append(Level(shelf, 3, 360, 300))
shelf.levels.append(Level(shelf, 4, 360, 300))
levelid_to_displaygoods_list = {
0: [
DisplayGoods(TestGoods('c2_1', 'c3_1', '1', 200, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_1', '2', 200, 80, 9, 0, 0),
face_num=2),
DisplayGoods(TestGoods('c2_1', 'c3_1', '3', 200, 40, 8, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_1', '4', 200, 40, 7, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_2', '5', 200, 20, 6, 0, 0),
face_num=2),
DisplayGoods(TestGoods('c2_1', 'c3_2', '6', 200, 40, 5, 0, 0),
face_num=3),
DisplayGoods(TestGoods('c2_1', 'c3_3', '7', 150, 40, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '8', 150, 40, 10, 0, 0))
],
1: [
DisplayGoods(TestGoods('c2_1', 'c3_3', '18', 150, 40, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '17', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '16', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '15', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '14', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '13', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '12', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '11', 150, 80, 10, 0,
0))
],
2: [
DisplayGoods(TestGoods('c2_1', 'c3_3', '21', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '22', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '23', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '24', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '25', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '26', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '27', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '28', 100, 40, 10, 0,
0))
],
3: [
DisplayGoods(TestGoods('c2_2', 'c3_6', '38', 100, 40, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '37', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '36', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '35', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '34', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '33', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '32', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '31', 100, 80, 10, 0,
0))
],
4: [
DisplayGoods(TestGoods('c2_2', 'c3_6', '41', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_6', '42', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_7', '43', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_7', '44', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_7', '45', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_7', '46', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_7', '47', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_8', '48', 100, 40, 10, 0,
0))
],
}
shelf.levels[0].display_goods_list = levelid_to_displaygoods_list[0]
shelf.levels[1].display_goods_list = levelid_to_displaygoods_list[1]
shelf.levels[2].display_goods_list = levelid_to_displaygoods_list[2]
shelf.levels[3].display_goods_list = levelid_to_displaygoods_list[3]
shelf.levels[4].display_goods_list = levelid_to_displaygoods_list[4]
choose_goods_list = [
TestGoods('c2_1', 'c3_1', '101', 100, 80, 20, 1, 0), # 上架
TestGoods('c2_1', 'c3_4', '102', 100, 80, 20, 1, 0), # 上架
TestGoods('c2_1', 'c3_5', '103', 100, 160, 20, 1, 0), # 上架
TestGoods('c2_1', 'c3_1', '4', 100, 40, 7, 2, 0), # 下架
TestGoods('c2_1', 'c3_4', '25', 100, 40, 8, 2, 0), # 下架
]
area_manager = AreaManager(shelf, levelid_to_displaygoods_list,
choose_goods_list)
area_manager._first_born_areas()
area_manager._second_combine_areas()
area_manager._prepare_area_base_data()
self.assertEqual(len(area_manager.area_list[0].choose_goods_list), 2)
self.assertEqual(len(area_manager.area_list[1].choose_goods_list), 0)
self.assertEqual(len(area_manager.area_list[2].choose_goods_list), 3)
self.assertEqual(len(area_manager.area_list[3].choose_goods_list), 0)
self.assertEqual(len(area_manager.area_list[0].up_choose_goods_list),
1)
self.assertEqual(
len(area_manager.area_list[0].down_display_goods_list), 1)
self.assertEqual(len(area_manager.area_list[2].up_choose_goods_list),
2)
self.assertEqual(
len(area_manager.area_list[2].down_display_goods_list), 1)
self.assertEqual(len(area_manager.down_display_goods_list), 2)
self.assertEqual(
len(area_manager.area_list[0].child_area_list[0].
down_display_goods_list), 1)
self.assertEqual(
len(area_manager.area_list[0].child_area_list[1].
down_display_goods_list), 0)
self.assertEqual(area_manager.area_list[0].levelid_to_goods_width[0],
480)
self.assertEqual(area_manager.area_list[0].levelid_to_remain_width[0],
int(480 / 600 * 40))
self.assertEqual(area_manager.area_list[1].levelid_to_goods_width[0],
80)
self.assertEqual(area_manager.area_list[1].levelid_to_remain_width[0],
int(80 / 600 * 40))
area_manager._prepare_area_calculate_data()
self.assertEqual(
len(area_manager.area_list[0].display_goods_to_reduce_face_num), 0)
self.assertEqual(
len(area_manager.area_list[0].second_down_display_goods_list), 0)
self.assertEqual(
len(area_manager.area_list[2].display_goods_to_reduce_face_num), 0)
self.assertEqual(
len(area_manager.area_list[2].second_down_display_goods_list), 2)
self.assertIn(levelid_to_displaygoods_list[2][2],
area_manager.area_list[2].second_down_display_goods_list)
self.assertIn(levelid_to_displaygoods_list[2][3],
area_manager.area_list[2].second_down_display_goods_list)
self.assertEqual(len(area_manager.down_display_goods_list), 4)
area_manager._arrange_areas()
def test_one_category3_not_down(self):
shelf = Shelf(1, 1800, 600, 300)
shelf.levels.append(Level(shelf, 0, 360, 300))
shelf.levels.append(Level(shelf, 1, 360, 300))
shelf.levels.append(Level(shelf, 2, 360, 300))
levelid_to_displaygoods_list = {
0: [
DisplayGoods(TestGoods('c2_1', 'c3_1', '1', 150, 40, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_1', '2', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_1', '3', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_1', '4', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_2', '5', 150, 80, 8, 0, 0)),
DisplayGoods(TestGoods('c2_1', 'c3_2', '6', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_2', '7', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_2', '8', 150, 80, 10, 0, 0))
],
1: [
DisplayGoods(TestGoods('c2_1', 'c3_3', '18', 80, 40, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '17', 80, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '16', 80, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_1', '1', 150, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '14', 80, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '13', 80, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '12', 80, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '11', 80, 80, 10, 0, 0))
],
2: [
DisplayGoods(TestGoods('c2_1', 'c3_3', '21', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_3', '22', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '23', 100, 80, 10, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '24', 100, 80, 9, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '25', 100, 80, 8, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_4', '26', 100, 80, 7, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '27', 100, 80, 6, 0,
0)),
DisplayGoods(TestGoods('c2_1', 'c3_5', '28', 100, 40, 5, 0, 0))
],
}
shelf.levels[0].display_goods_list = levelid_to_displaygoods_list[0]
shelf.levels[1].display_goods_list = levelid_to_displaygoods_list[1]
shelf.levels[2].display_goods_list = levelid_to_displaygoods_list[2]
choose_goods_list = [
TestGoods('c2_1', 'c3_1', '101', 100, 80, 20, 1, 0), # 上架
]
area_manager = AreaManager(shelf, levelid_to_displaygoods_list,
choose_goods_list)
area_manager._first_born_areas()
for area in area_manager.area_list:
print(area)
area_manager._second_combine_areas()
for area in area_manager.area_list:
print(area)
area_manager._prepare_area_base_data()
area_manager._prepare_area_calculate_data()
area_manager._print_choose_goods_info()
area_manager._arrange_areas()
best_candidate_shelf = area_manager.arrange_goods()
level_to_display_goods_name_str = get_level_to_display_goods_name(
best_candidate_shelf)
self.assertEqual(level_to_display_goods_name_str[0],
'1,2,3,4,101,6,7,8,')
self.assertEqual(level_to_display_goods_name_str[1],
'11,12,13,14,1,16,17,18,')
self.assertEqual(level_to_display_goods_name_str[2],
'21,22,23,24,25,26,27,28,')
print(shelf)
print(best_candidate_shelf)
def test_no_bottom_level(self):
shelf = Shelf(1, 510, 340, 300)
shelf.levels.append(Level(shelf, 0, 170, 300))
shelf.levels.append(Level(shelf, 1, 170, 300))
shelf.levels.append(Level(shelf, 2, 170, 300))
levelid_to_displaygoods_list = {
0: [
DisplayGoods(TestGoods('c2_1', 'c3_1', '1', 120, 58, 10, 0, 0),
face_num=2),
DisplayGoods(TestGoods('c2_1', 'c3_1', '1', 120, 58, 10, 0, 0),
face_num=2),
DisplayGoods(TestGoods('c2_1', 'c3_2', '2', 120, 52, 10, 0, 0))
],
1: [
DisplayGoods(TestGoods('c2_3', 'c3_4', '14', 120, 68, 10, 0,
0)),
DisplayGoods(TestGoods('c2_3', 'c3_4', '13', 120, 68, 10, 0,
0)),
DisplayGoods(TestGoods('c2_3', 'c3_4', '12', 120, 68, 10, 0,
0)),
DisplayGoods(TestGoods('c2_2', 'c3_3', '11', 120, 62, 10, 0,
0))
],
2: [
DisplayGoods(TestGoods('c2_3', 'c3_5', '21', 120, 70, 10, 0,
0)),
DisplayGoods(TestGoods('c2_3', 'c3_5', '22', 120, 70, 10, 0,
0)),
DisplayGoods(TestGoods('c2_4', 'c3_6', '23', 120, 68, 10, 0,
0)),
DisplayGoods(TestGoods('c2_4', 'c3_6', '24', 120, 68, 10, 0,
0))
],
}
shelf.levels[0].display_goods_list = levelid_to_displaygoods_list[0]
shelf.levels[1].display_goods_list = levelid_to_displaygoods_list[1]
shelf.levels[2].display_goods_list = levelid_to_displaygoods_list[2]
choose_goods_list = [
TestGoods('c2_1', 'c3_1', '101', 100, 50, 20, 1, 0), # 上架
TestGoods('c2_3', 'c3_4', '102', 100, 68, 20, 1, 0), # 上架
]
area_manager = AreaManager(shelf, levelid_to_displaygoods_list,
choose_goods_list)
area_manager._first_born_areas()
area_manager._second_combine_areas()
area_manager._prepare_area_base_data()
area_manager._prepare_area_calculate_data()
area_manager._arrange_areas()
area_manager._print_choose_goods_info()
best_candidate_shelf = area_manager.arrange_goods()
level_to_display_goods_name_str = get_level_to_display_goods_name(
best_candidate_shelf)
self.assertIn(level_to_display_goods_name_str[0],
['101,1,1,1,1,2,null,', '1,1,1,1,101,2,null,'])
self.assertEqual(level_to_display_goods_name_str[1],
'11,102,12,13,14,')
self.assertEqual(level_to_display_goods_name_str[2], '21,22,23,24,')
print(shelf)
print(best_candidate_shelf)