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)