def test_actions(self): world = World() realm = world.create_realm('Realm0', 1.0, 1.0, 1.0) province = Province(None, 0,0) action0 = Action() province.action = action0 realm.add_province(province) result = world.tick() self.assertEqual(world.turn, 1) self.assertEqual(action0.count, 1) self.assertEqual(province.action, None) self.assertEqual(len(result.battles), 0) army = world.create_army(realm) army.size = 1 province.add_army(army) action1 = Action() army.action = action1 result = world.tick() self.assertEqual(world.turn, 2) self.assertEqual(action0.count, 1) self.assertEqual(action1.count, 1) self.assertEqual(army.action, None) self.assertEqual(province.action, None) self.assertEqual(len(result.battles), 0)
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)
from model.world import World from view import console import time if __name__ == "__main__": world = World() while True: world.tick() console.overview(world) time.sleep(0.05)