class TestSettlement(unittest.TestCase):
def setUp(self):
self.dg=DummyGame()
self.p1=HumanPlayer(self.dg)
self.settlement = Settlement(population=1500, owner=self.p1, map=DummyMap())
def test_population(self):
self.assertEqual(1500, self.settlement.population, "Population initialized incorrectly")
self.settlement.build("farm")
self.assertEqual(1600, self.settlement.population, "Building a farm doesn't increase population")
def test_recruitment(self):
self.assertFalse(self.settlement.do_recruit(DummyUnit, self.p1, DummyTile(), 10000), "It's possible to recruit a unit too expensive")
self.assertTrue(self.settlement.do_recruit(DummyUnit, self.p1, DummyTile(), 500), "Recruitment doesn't work")
self.assertEqual(500, self.p1.treasury, "Recruitment doesn't affect owner's treasury")
def tearDown(self):
pass
