def test_get_province(self): test_map = Map() test_map.create(None, 2, 4) for x in range(0, 2): for y in range(0, 4): self.assertEqual(test_map.get_province(x,y), test_map.provinces[(x, y)])
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_create(self): test_map = Map() test_map.create(1, 2, 4) self.assertEqual(test_map.width, 2) self.assertEqual(test_map.height, 4) self.assertEqual(len(test_map.provinces), 8) for x in range(0, 2): for y in range(0, 4): province = test_map.get_province(x,y) self.assertEqual(province.map, test_map) self.assertEqual(province.terrain, 1) self.assertEqual(province.x, x) self.assertEqual(province.y, y)
def test_outside(self): test_map = Map() test_map.create(None, 2, 4) self.assertEqual(test_map.get_province(-1,0), None) self.assertEqual(test_map.get_province(2,0), None) self.assertEqual(test_map.get_province(0,-1), None) self.assertEqual(test_map.get_province(0,4), None)
from model.world import World from utility.color import Color from view.mapview import MapView selected_army = None selected_province = None moved_army = None realm = None realm_index = 0 if __name__ == '__main__': window = pyglet.window.Window(800, 600, 'PyRealm 01') terrain0 = Terrain("Plain", Color(0.0, 1.0, 0.0)) gamemap = Map() gamemap.create(terrain0, 3, 3) world = World(gamemap) realm0 = world.create_realm('Askir', 1.0, 0.0, 0.0) realm0.add_province(gamemap.get_province(2, 2)) realm1 = world.create_realm('Thalak', 0.0, 0.0, 1.0) realm1.add_province(gamemap.get_province(0, 0)) realm = realm0 print('Realm={0:s}:'.format(realm.name)) view = MapView(gamemap, 50.0, 5.0, 5.0, 15.0)
def test_get_neighbour(self): test_map = Map() test_map.create(None, 3, 4) province = test_map.get_province(1, 1) self.assertEqual(test_map.get_province(2, 1), province.get_neighbour(0)) self.assertEqual(test_map.get_province(2, 2), province.get_neighbour(1)) self.assertEqual(test_map.get_province(1, 2), province.get_neighbour(2)) self.assertEqual(test_map.get_province(0, 1), province.get_neighbour(3)) self.assertEqual(test_map.get_province(1, 0), province.get_neighbour(4)) self.assertEqual(test_map.get_province(2, 0), province.get_neighbour(5)) province = test_map.get_province(1, 2) self.assertEqual(test_map.get_province(2, 2), province.get_neighbour(0)) self.assertEqual(test_map.get_province(1, 3), province.get_neighbour(1)) self.assertEqual(test_map.get_province(0, 3), province.get_neighbour(2)) self.assertEqual(test_map.get_province(0, 2), province.get_neighbour(3)) self.assertEqual(test_map.get_province(0, 1), province.get_neighbour(4)) self.assertEqual(test_map.get_province(1, 1), province.get_neighbour(5))