def test_inventory_list(self):
inv = Inventory(10)
inv.set(Good.grain, 5, 1.5)
self.assertTrue(inv.add(Good.tools, 2, 3.25))
self.assertTrue(inv.add(Good.tools, 3, 4.25))
g1 = inv.get(Good.grain)
g2 = inv.get(Good.tools)
self.assertEqual(g1[0].amount, 5)
self.assertEqual(g1[0].price, 1.5)
self.assertEqual(g2[0].amount, 2)
self.assertEqual(g2[0].price, 3.25)
self.assertEqual(g2.amount, 5)
self.assertEqual(g2.price, 7.5)
inv.add(Good.iron_ore, 1)
inv.add(Good.iron_ore, 1)
self.assertEqual(len(inv.get(Good.iron_ore)), 1)
inv.add(Good.iron_ore, 1, 1)
inv.add(Good.iron_ore, 2, 2)
self.assertEqual(len(inv.get(Good.iron_ore)), 3)
self.assertEqual(inv.get_amount(Good.fish), 0)
def test_inventory_list(self):
inv = Inventory(10)
inv.set(Good.grain, 5, 1.5)
self.assertTrue(inv.add(Good.tools, 2, 3.25))
self.assertTrue(inv.add(Good.tools, 3, 4.25))
g1 = inv.get(Good.grain)
g2 = inv.get(Good.tools)
self.assertEqual(g1[0].amount, 5)
self.assertEqual(g1[0].price, 1.5)
self.assertEqual(g2[0].amount, 2)
self.assertEqual(g2[0].price, 3.25)
self.assertEqual(g2.amount, 5)
self.assertEqual(g2.price, 7.5)
inv.add(Good.iron_ore, 1)
inv.add(Good.iron_ore, 1)
self.assertEqual(len(inv.get(Good.iron_ore)), 1)
inv.add(Good.iron_ore, 1, 1)
inv.add(Good.iron_ore, 2, 2)
self.assertEqual(len(inv.get(Good.iron_ore)), 3)
self.assertEqual(inv.get_amount(Good.fabric), 0)
