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}}])