def testEmptyWarehouseOneOrderWithLeftovers(self):
orders = {'orange': 20}
warehouses = [{
'name': 'owd',
'inventory': {
'apple': 5,
'orange': 10
}
}, {
'name': 'dm',
'inventory': {
'banana': 5,
'orange': 10
}
}]
furthest = 1
cheapestShipment = []
inventoryAllocator = InventoryAllocator()
leftovers, cheapestShipment = \
inventoryAllocator.emptyWarehouse(orders,
warehouses,
furthest,
cheapestShipment)
self.assertEqual(leftovers, {'orange': 10})
self.assertEqual(cheapestShipment, [{'dm': {'orange': 10}}])
def testEmptyWarehouseOneOrder(self):
orders = {'apple': 1}
warehouses = [{'name': 'owd', 'inventory': {'apple': 1}}]
furthest = 0
cheapestShipment = []
inventoryAllocator = InventoryAllocator()
leftovers, cheapestShipment = \
inventoryAllocator.emptyWarehouse(orders,
warehouses,
furthest,
cheapestShipment)
self.assertEqual(leftovers, {'apple': 0})
self.assertEqual(cheapestShipment, [{'owd': {'apple': 1}}])