def test_usedbattalions(self):
paux = HumanPlayers("Pepe", 35, "orange", [])
country1 = Country("Europa del sur", paux)
country2 = Country("Francia", paux)
country3 = Country("Italia", paux)
country1.changebattalions(7)
country2.changebattalions(1)
country3.changebattalions(5)
paux.addconqueredcountry(country1)
paux.addconqueredcountry(country2)
paux.addconqueredcountry(country3)
self.assertEqual(paux.getusedbattalions(), 13)
def test_distributebattalions(self):
paux = HumanPlayers("Pepe", 35, "orange", [])
country1 = Country("Europa del sur", paux)
country2 = Country("Francia", paux)
country3 = Country("Italia", paux)
country1.changebattalions(7)
country2.changebattalions(1)
country3.changebattalions(5)
paux.addconqueredcountry(country1)
paux.addconqueredcountry(country2)
paux.addconqueredcountry(country3)
paux.distributebatallions()
self.assertEqual(paux.getusedbattalions(), 35)
conqueredCountries = paux.getconqueredcountries()
self.assertEqual(conqueredCountries[0].tostring(),
"Europa del sur 15:Pepe 35 orange")
self.assertEqual(conqueredCountries[1].tostring(),
"Francia 8:Pepe 35 orange")
self.assertEqual(conqueredCountries[2].tostring(),
"Italia 12:Pepe 35 orange")