def test_battles(self):
map = Map()
map.create(None, 3, 3)
world = World(map)
realm0 = world.create_realm('Realm0', 1.0, 1.0, 1.0)
realm1 = world.create_realm('Realm1', 1.0, 1.0, 1.0)
province = map.get_province(0, 0)
army0 = Army(0, realm0)
army0.size = 2
army1 = Army(0, realm1)
army1.size = 1
province.add_army(army0)
province.add_army(army1)
result = world.tick()
self.assertEqual(len(result.battles), 1)
battle = result.battles[0]
self.assertEqual(battle.army0, army0)
self.assertEqual(battle.army1, army1)
self.assertEqual(army0.size, 1)
self.assertEqual(army1.size, 0)
self.assertEqual(army0.province, province)
self.assertTrue(army0 in province.armies)
self.assertEqual(army1.province, None)
self.assertFalse(army1 in province.armies)
def test_remove(self):
army = Army()
province = Province(None, 1, 1)
province.add_army(army)
army.remove()
self.assertEqual(army.province, None)
self.assertFalse(army in province.armies)
def test_even(self):
realm0 = Realm()
realm1 = Realm()
army0 = Army(0, realm0)
army1 = Army(1, realm1)
army0.size = 5
army1.size = 5
province = Province(None, 1, 1)
province.add_army(army0)
province.add_army(army1)
battle = Battle(army0, army1)
self.assertEqual(army0.size, 0)
self.assertEqual(army1.size, 0)
self.assertEqual(army0.province, None)
self.assertFalse(army0 in province.armies)
self.assertEqual(army1.province, None)
self.assertFalse(army1 in province.armies)
def test_zero_wins(self):
realm0 = Realm()
realm1 = Realm()
army0 = Army(0, realm0)
army1 = Army(1, realm1)
army0.size = 10
army1.size = 4
province = Province(None, 1, 1)
province.add_army(army0)
province.add_army(army1)
battle = Battle(army0, army1)
self.assertEqual(army0.size, 6)
self.assertEqual(army1.size, 0)
self.assertEqual(army0.province, province)
self.assertTrue(army0 in province.armies)
self.assertEqual(army1.province, None)
self.assertFalse(army1 in province.armies)
