Exemplo n.º 1
0
    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)
Exemplo n.º 2
0
    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)
Exemplo n.º 3
0
    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)
Exemplo n.º 4
0
    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)