def test_cavalry_step2_fail(self): area1 = Area(id="A1", iso='UK', unit='cav') area2 = Area(id="A3", iso='UK') with self.assertRaises(MoveException) as context: movement.move_to(area1, area2) self.assertEqual(context.exception.reason, 'cant_move_there') self.assertEqual(area1.unit, 'cav') self.assertEqual(area2.unit, None)
def test_occupied_fail(self): area1 = Area(id="A1", iso='UK', unit='cav') area2 = Area(id="A2", iso='UK', unit='inf') with self.assertRaises(MoveException) as context: movement.move_to(area1, area2) self.assertEqual(context.exception.reason, 'not_enemy') self.assertEqual(area1.unit, 'cav') self.assertEqual(area2.unit, 'inf')
def test_nounit_fail(self): area1 = Area(id="A1", iso='UK') area2 = Area(id="A2", iso='FR') with self.assertRaises(MoveException) as context: movement.move_to(area1, area2) self.assertEqual(context.exception.reason, 'invalid_params') self.assertIsNone(area1.unit) self.assertIsNone(area2.unit)
def test_mount_exhaust_pass(self): area1 = Area(id="A1", iso='UK', unit='inf') area2 = Area(id="A2", iso='UK', tile='mount') try: movement.move_to(area1, area2) except MoveException as e: self.fail("test_mount_exhaust_pass raised MoveException({}) unexpectedly!".format(e.reason)) self.assertEqual(area2.exhaust, 3)
def test_inf_attack2_fail(self): area1 = Area(id="A1", iso='UK', unit='inf') area2 = Area(id="A3", iso='FR', unit='cav') with self.assertRaises(MoveException) as context: movement.move_to(area1, area2) self.assertEqual(context.exception.reason, 'cant_attack_there') self.assertEqual(area1.unit, 'inf') self.assertEqual(area2.unit, 'cav') self.assertEqual(area2.iso, 'FR')
def test_move_pass(self): area1 = Area(id="A1", iso='UK', unit='inf') area2 = Area(id="A2", iso='FR') try: movement.move_to(area1, area2) except MoveException as e: self.fail("test_move_pass raised MoveException({}) unexpectedly!".format(e.reason)) self.assertEqual(area1.unit, None) self.assertEqual(area1.iso, 'UK') self.assertEqual(area2.unit, 'inf') self.assertEqual(area2.iso, 'UK')
def test_occupied_fail(self): country = Country(iso='UK', gold=20, pop=1) area = Area(iso='UK', tile='city', unit='inf') with self.assertRaises(BuyException) as context: service.buy_item(area, country, 'inf') self.assertEqual(context.exception.reason, 'item_exists')
def test_money_fail(self): country = Country(iso='UK', gold=19, pop=1) area = Area(iso='UK', tile='city') with self.assertRaises(BuyException) as context: service.buy_item(area, country, 'inf') self.assertEqual(context.exception.reason, 'not_enough_gold') self.assertIsNone(area.unit)
def test_not_city_fail(self): country = Country(iso='UK', gold=20, pop=1) area = Area(iso='UK') with self.assertRaises(BuyException) as context: service.buy_item(area, country, 'inf') self.assertEqual(context.exception.reason, 'missing_city') self.assertIsNone(area.unit)
def test_pesto_sacrifice(self): country = Country(iso='UK', gold=20, pop=0, shields=2) area = Area(iso='UK', tile='city') economy.sacrifice_shield(country, area, 'cav') self.assertEqual(country.gold, 20) self.assertEqual(country.pop, 0) self.assertEqual(area.unit, 'cav')
def test_pesto_notcity_fail(self): country = Country(iso='UK', gold=0, pop=0, shields=2) area = Area(iso='UK') with self.assertRaises(BuyException) as context: economy.sacrifice_shield(country, area, 'art') self.assertEqual(context.exception.reason, 'missing_city') self.assertIsNone(area.unit)
def test_cavalry_forest_fail(self): area1 = Area(id="A1", iso='UK', unit='cav') area2 = Area(id="A2", iso='FR', unit='inf', tile='forest') with self.assertRaises(MoveException) as context: movement.move_to(area1, area2) self.assertEqual(context.exception.reason, 'cant_attack_cavalry') self.assertEqual(area1.unit, 'cav') self.assertEqual(area2.unit, 'inf') self.assertEqual(area2.iso, 'FR') # infantry can, however area3 = Area(id="A3", iso='UK', unit='inf') try: movement.move_to(area3, area2) except MoveException as e: self.fail("test_cavalry_forest_fail raised MoveException({}) unexpectedly!".format(e.reason)) self.assertEqual(area3.unit, None) self.assertEqual(area2.unit, 'inf') self.assertEqual(area2.iso, 'UK')
def test_buy_pass(self): country = Country(iso='UK', gold=20 + 70 + 120, pop=3) try: area = Area(iso='UK', tile='city') service.buy_item(area, country, 'inf') self.assertEqual(area.unit, 'inf') self.assertEqual(country.gold, 70 + 120) self.assertEqual(country.pop, 2) except BuyException as e: self.fail( "test_occupied_fail raised BuyException({}) unexpectedly!". format(e.reason)) try: area = Area(iso='UK', tile='city') service.buy_item(area, country, 'cav') self.assertEqual(area.unit, 'cav') self.assertEqual(country.gold, 120) self.assertEqual(country.pop, 1) except BuyException as e: self.fail( "test_occupied_fail raised BuyException({}) unexpectedly!". format(e.reason)) try: area = Area(iso='UK', tile='city') service.buy_item(area, country, 'art') self.assertEqual(area.unit, 'art') self.assertEqual(country.gold, 0) self.assertEqual(country.pop, 0) except BuyException as e: self.fail( "test_occupied_fail raised BuyException({}) unexpectedly!". format(e.reason))
def load_gtml(filename, skip=None): if skip is None: skip = [] l_countries = [] l_areas = [] l_calls = [] l_options = {} f_countries = [] area_pattern = re.compile(r"(?P<id>\w+)\((?P<iso>\w*),?(?P<buildtile>\w*),?(?P<unit>\w*)\)") with open(filename) as fh: status = None for line in fh: if not line or line[0] == '#' or line[0] == '\n': continue if line[0] == '>': status = line[1:-1] if status == 'STOP_TEST': l_calls.append((status,None)) break elif status[:9] == 'COUNTRIES': f_countries = status[10:].split() status = 'COUNTRIES' continue if status in skip: continue if status == 'COUNTRIES': attrs = line.split() cc = Country() for val,attr in zip(attrs, f_countries): if attr in ('iso', 'name', 'color'): if attr == 'name': val = val.replace('_', ' ') setattr(cc, attr, val) elif attr in ('emperor', 'ai'): setattr(cc, attr, val == 'true') else: setattr(cc, attr, int(val)) l_countries.append(cc) elif status == 'AREAS': match = area_pattern.findall(line) for (aid, iso, btile, unit) in match: area = Area(id=aid) if iso: area.iso = iso if btile in ('barr','house','cita'): area.build = btile area.tile = 'city' else: area.tile = btile if unit: area.unit = unit l_areas.append(area) elif status == 'OPTIONS': k,v = line.split() l_options[k] = v elif status == 'TEST_AREAS': ff = [] sareas = line.split() for sarea in sareas: sarea = sarea.split(',') area = Area(iso=sarea[0]) if len(sarea) > 1 and sarea[1]: tile = sarea[1] if tile in ('barr','house','cita'): area.tile = 'city' area.build = tile else: area.tile = tile if len(sarea) > 2: area.unit = sarea[2] ff.append(area) l_areas.append(ff) elif status == 'TEST_CALLS': method,params,iso,exps = line.split() params = params.split(',') try: exp_js = list(map(json.loads, exps.split('|'))) except: raise Exception("Failed to parse call: {}".format(exps)) if method == 'BUY': call = ('Game:buy', {'area_id': params[0], 'item_id': params[1]}, iso) elif method == 'MOVE': call = ('Game:move', {'area_id': params[0], 'to_id': params[1]}, iso) elif method == 'END': call = ('Game:end_turn', {}, iso) else: continue l_calls.append((call, exp_js)) if len(l_calls) == 0 and len(l_options) != 0: l_calls = l_options return l_countries, l_areas, l_calls