def test_production(self): farmer = make_mock_pop(province, PopType.farmer) province.add_pops([farmer]) grain = farmer.inventory.get_amount(Good.grain) tools = farmer.inventory.get_amount(Good.tools) bread = farmer.inventory.get_amount(Good.bread) timber = farmer.inventory.get_amount(Good.timber) farmer.perform_production() self.assertEqual(farmer.inventory.get_amount(Good.grain), grain + 4) self.assertEqual(farmer.inventory.get_amount(Good.bread), bread - 1) self.assertEqual(farmer.inventory.get_amount(Good.timber), timber - 1) farmer.inventory.add(Good.timber, 1) farmer.inventory.add(Good.bread, 1) grain = farmer.inventory.get_amount(Good.grain) tools = farmer.inventory.get_amount(Good.tools) bread = farmer.inventory.get_amount(Good.bread) timber = farmer.inventory.get_amount(Good.timber) farmer.perform_production() self.assertEqual(farmer.inventory.get_amount(Good.grain), grain + 4) self.assertEqual(farmer.inventory.get_amount(Good.bread), bread - 1) self.assertEqual(farmer.inventory.get_amount(Good.timber), timber - 1)
def test_money(self): farmer = make_mock_pop(province, PopType.farmer) province.add_pops([farmer]) farmer.money += 10 self.assertEqual(farmer.money, 110) farmer.money -= 10 self.assertEqual(farmer.money, 100)
def test_idle_fee(self): farmer = make_mock_pop(province, PopType.farmer) province.add_pops([farmer]) farmer.inventory.set(Good.timber, 0) self.assertEqual(farmer.money, 100) farmer.perform_production() self.assertEqual(farmer.money, 99)
def test_idle_fee(self): farmer = make_mock_pop(province, PopJob.farmer) province.add_pops([farmer]) farmer.inventory.set(Good.timber, 0) self.assertEqual(farmer.money, 10) farmer.perform_logic() self.assertEqual(farmer.money, 8)