def test_get(self):
inv = Inventory(10)
inv.add(Good.grain, 5)
# get grain in inventory
self.assertEqual(inv.get(Good.grain).amount, 5)
# no iron ore in inventory
self.assertEqual(inv.get(Good.iron_ore), None)
def test_get(self):
inv = Inventory(10)
inv.add(Good.grain, 5)
# get grain in inventory
self.assertEqual(inv.get(Good.grain).amount, 5)
# no iron ore in inventory
self.assertEqual(inv.get(Good.iron_ore), None)
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)
def test_subtract(self):
inv = Inventory(10)
inv.add(Good.grain, 10)
self.assertTrue(inv.subtract(Good.grain, 1))
self.assertEqual(inv.get(Good.grain).amount, 9)
self.assertFalse(inv.subtract(Good.grain, 10))
def test_subtract(self):
inv = Inventory(10)
inv.add(Good.grain, 10)
self.assertTrue(inv.subtract(Good.grain, 1))
self.assertEqual(inv.get(Good.grain).amount, 9)
self.assertFalse(inv.subtract(Good.grain, 10))
