def test_twoWarehouse_perfectInventory(self):
order = {"apple":10, "banana":10, "cucumber":10, "mango":10, "pineapple":10}
warehouse = [{"name":"warehouse", "inventory":{"apple":5, "banana":5, "cucumber":5, "mango":5, "pineapple":5}},
{"name":"factory", "inventory":{"apple":5, "banana":5, "cucumber":5, "mango":5, "pineapple":5}}]
output = [{"warehouse":{"apple":5, "banana":5, "cucumber":5, "mango":5, "pineapple":5}},
{"factory":{"apple":5, "banana":5, "cucumber":5, "mango":5, "pineapple":5}}]
self.assertEqual(output, main.minimizeOrderCost(order, warehouse))
def test_shippingCost(self):
order = {"apple":10}
warehouse = [{"name":"factory", "inventory":{"apple":9}},
{ "name": "warehouse","inventory":{"apple": 15}}]
output = [{"factory":{"apple":9}}, {"warehouse":{"apple":1}}]
self.assertEqual(output, main.minimizeOrderCost(order, warehouse))
def test_oneWarehouse_badInventory(self):
order = {"apple":2, "banana":2, "cucumber":2, "mango":2, "pineapple":2}
warehouse = [{"name":"warehouse", "inventory":{"apple":1, "banana":1, "cucumber":1, "mango":1, "pineapple":1}}]
output = []
self.assertEqual(output, main.minimizeOrderCost(order, warehouse))
