def test_fill_board(self):
warehouse = Warehouse(True)
warehouse.board_width = 10
warehouse.board_height = 10
warehouse.init_board()
test_item = [123, 'apples', 1, 6, 6]
warehouse.add_item(test_item)
warehouse.fill_board(0, warehouse.new_item_list[0], 1)
self.assertEqual('1', warehouse.new_board[0][0], 'Passed')
def test_add_item(self):
warehouse = Warehouse(True)
test_list = [123, 'apples', 5, 6, 6]
warehouse.add_item(test_list)
self.assertEqual('apples', warehouse.new_item_list[0].name, 'Passed')
self.assertEqual(123, warehouse.new_item_list[0].item_id, 'Passed')
self.assertEqual(5, warehouse.new_item_list[0].quantity, 'Passed')
self.assertEqual(6, warehouse.new_item_list[0].length, 'Passed')
self.assertEqual(6, warehouse.new_item_list[0].width, 'Passed')
def test_packing(self):
warehouse = Warehouse(True)
warehouse.board_width = 10
warehouse.board_height = 10
warehouse.init_board()
test_item = [123, 'apples', 1, 6, 6]
warehouse.add_item(test_item)
self.assertEqual(0, len(warehouse.complete_list), 'Passed')
warehouse.packing(warehouse.new_item_list)
self.assertEqual(1, len(warehouse.complete_list), 'Passed')
def test_check_available_area(self):
warehouse = Warehouse(True)
test_list = [123, 'apples', 1, 6, 6]
warehouse.add_item(test_list)
test_list_2 = [456, 'oranges', 2, 2, 2]
warehouse.add_item(test_list_2)
test_true = warehouse.check_available_area(warehouse.new_item_list, 44)
test_false = warehouse.check_available_area(warehouse.new_item_list,
10)
self.assertEqual(True, test_true, 'Passed')
self.assertEqual(False, test_false, 'Passed')
class UserInterface:
def __init__(self):
self.warehouse = Warehouse()
self.sales = Sales()
self.finance = Finance()
def add_new_bike(self, model, price, colour, size, gender, quantity):
self.warehouse.add_item(model, price, colour, size, gender, quantity)
def display_stock(self):
self.warehouse.print_all_items()
# def add_bike_to_cart(self,model,quantity):
# if self.warehouse.can_buy(model,quantity) :
# self.sales.add_to_cart(model,quantity)
def add_to_cart(self):
bike_model = input("Enter bike model:")
bike_index = self.warehouse.find_bike_by_model(bike_model)
if (bike_index == -1):
print("We don't have that model")
else:
bike = self.warehouse.stock[bike_index]
amount = int(input("How many of bikes would you buy?"))
if (amount > bike.quantity):
print("We don't have enough of that model")
else:
self.sales.add_to_cart(bike, amount)
def make_purchase(self):
cart = self.sales.get_cart()
if (len(cart) == 0):
print("No items in cart!")
else:
for bike, amount in cart:
self.warehouse.decrease_quantity_by(bike.model, amount)
report = self.finance.create_report(cart)
print(report)
self.sales.make_purchase()
def display_cart(self):
self.sales.display_cart()
def display_reports(self):
self.finance.display_reports()
def empty_the_cart(self):
self.sales.empty_the_cart()
def test_create_rect(self):
warehouse = Warehouse(test_flag=True)
visualize = VisualizeWarehouse(test_flag=True)
visualize.screen_buffer = 10
visualize.width_segment = 40
visualize.height_segment = 40
test_board = [['0', '0', '1', '1', '0'], ['0', '0', '1', '1', '0'],
['0', '0', '1', '1', '0']]
test_item = [123, 'apples', 1, 6, 6]
warehouse.add_item(test_item)
result = visualize.create_item_rect(test_board,
warehouse.new_item_list)
self.assertEqual(1, len(visualize.rectangle_list), 'Passed')
def test_remove_item(self):
warehouse = Warehouse(True)
test_list = [123, 'apples', 5, 6, 6]
warehouse.add_item(test_list)
warehouse.remove_item(test_list[0])
self.assertEqual([], warehouse.item_list, 'Passed')
from warehouse import Warehouse, Printer
warehouse = Warehouse()
warehouse.add_item(Printer('HP', 'M15w', 'laser', 'LaserJet Pro'))
warehouse.add_item(Printer('Epson', 'M1100', 'solid_ink'))
warehouse.send_item_to_department(Printer, 1, 'Юридический отдел', {
'_type': 'solid_ink',
})
