class TestVillage(unittest.TestCase): def setUp(self): self.world = 'pl100' self.village = Village(self.world) def test_build_normal_building(self): next_building = Building(self.world, 'main', 1) self.village.build(next_building) self.assertEqual(-90, self.village.wood) self.assertEqual(-80, self.village.stone) self.assertEqual(-70, self.village.iron) self.assertEqual(25, self.village.population) self.assertEqual(205, self.village.max_population) self.assertEqual(813, self.village.capacity) self.assertEqual(1, self.village.buildings['main'].level) self.assertEqual(10, self.village.points) def test_build_farm(self): next_building = Building(self.world, 'storage', 1) self.village.build(next_building) self.assertEqual(205, self.village.max_population) self.assertEqual(1000, self.village.capacity) self.assertEqual(6, self.village.points) def test_build_storage(self): next_building = Building(self.world, 'farm', 1) self.village.build(next_building) self.assertEqual(240, self.village.max_population) self.assertEqual(813, self.village.capacity) self.assertEqual(5, self.village.points) def test_add_resource(self): self.village.add_resource(10) self.assertEqual(0.07222222222222223, self.village.wood) self.assertEqual(0.07222222222222223, self.village.stone) self.assertEqual(0.07222222222222223, self.village.iron) def test_add_resource_max_capacity(self): self.village.add_resource(100000000000000) self.assertEqual(self.village.capacity, self.village.wood) self.assertEqual(self.village.capacity, self.village.stone) self.assertEqual(self.village.capacity, self.village.iron)