def test_005_roster_add_pokemon(self): s = pokemon.Species(1, 'bulbasaur') p1 = pokemon.Pokemon(1, s) p2 = pokemon.Pokemon(2, s) t = pokemon.Team(1, 'basic team', first=p1) r = pokemon.Roster(pokemon=[p1], teams=[t], active_team_id=1) r.add_pokemon(p2) assert r.get_pokemon(2) == p2
def test_006_roster_remove_pokemon(self): s = pokemon.Species(1, 'bulbasaur') p1 = pokemon.Pokemon(1, s) p2 = pokemon.Pokemon(2, s) t = pokemon.Team(1, 'basic team', first=p1, second=p2) r = pokemon.Roster(pokemon=[p1, p2], teams=[t], active_team_id=1) r.remove_pokemon(p2) self.assertRaises(KeyError, r.get_pokemon, 2) assert t.get_position(2) is None
def test_009_team_set_position_accepts_ints(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t = pokemon.Team(1, 'basic team') t.set_position(1, p) assert t.get_position(1) == p self.assertRaises(ValueError, t.set_position, -1, p)
def test_011_team_from_dict(self): mock_pokeapi_calls() s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t = pokemon.Team(1, 'basic team', first=p) data = t.to_dict() assert pokemon.Team.from_dict(data, [p]) == t
def test_002_roster_add_team(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t1 = pokemon.Team(1, 'basic team', first=p) t2 = pokemon.Team(2, 'another team', first=p) r = pokemon.Roster(pokemon=[p], teams=[t1], active_team_id=1) r.add_team(t2) assert r.get_team(2) == t2
def test_004_roster_remove_team_resets_active_team_id(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t1 = pokemon.Team(1, 'basic team', first=p) t2 = pokemon.Team(2, 'another team', first=p) r = pokemon.Roster(pokemon=[p], teams=[t1, t2], active_team_id=2) r.remove_team(t2) assert r.active_team() is None self.assertRaises(KeyError, r.get_team, 2)
def test_012_team_to_string(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t = pokemon.Team(1, 'basic team', first=p) assert 'basic team' in str(t) assert '1: bulbasaur' in str(t) assert '2: Empty' in str(t) assert '3: Empty' in str(t) assert '4: Empty' in str(t) assert '5: Empty' in str(t) assert '6: Empty' in str(t)
def test_013_team_from_dict_with_pokemon_list(self): mock_pokeapi_calls() s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t = pokemon.Team(1, 'basic team', first=p) data = t.to_dict() reloaded = pokemon.Team.from_dict(data, [p]) reloaded.get_position(1).id = 2 assert reloaded.get_position(1) is p assert reloaded.get_position(1).id == 2 assert p.id == 2
def test_001_roster_active_team(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t1 = pokemon.Team(1, 'basic team', first=p) t2 = pokemon.Team(2, 'another team', first=p) r = pokemon.Roster(pokemon=[p], teams=[t1, t2], active_team_id=1) assert r.active_team() == t1 r.set_active_team(2) assert r.active_team() == t2 r.set_active_team(None) assert r.active_team() is None
def test_007_roster_to_dict(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t1 = pokemon.Team(1, 'basic team', first=p) r = pokemon.Roster(pokemon=[p], teams=[t1], active_team_id=1) expected = { 'pokemon': [p.to_dict()], 'teams': [t1.to_dict()], 'active_team': 1 } assert r.to_dict() == expected
def test_003_team_full(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t = pokemon.Team(1, 'basic team', first=p, second=p, third=p, fourth=p, fifth=p, sixth=p) assert t.is_full()
def test_009_roster_from_dict(self): mock_pokeapi_calls() s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t1 = pokemon.Team(1, 'basic team', first=p) t2 = pokemon.Team(2, 'another team', first=p) r = pokemon.Roster(pokemon=[p], teams=[t1, t2], active_team_id=2) reloaded = pokemon.Roster.from_dict(r.to_dict()) p_reloaded = reloaded.get_pokemon(1) assert r == reloaded assert reloaded.teams[1].get_position(1) is p_reloaded assert reloaded.teams[2].get_position(1) is p_reloaded
def test_004_pokemon_to_dict(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) d = p.to_dict() expected = { 'id': 1, 'species': 1, 'nick_name': None, 'pokerus': False, 'item': None, 'evs': pokemon.StatSet().to_dict(), 'stats': pokemon.StatSet().to_dict(), 'move_set': None } assert d == expected
def test_010_team_to_dict(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t = pokemon.Team(1, 'basic team', first=p) expected = { 'id': 1, 'name': 'basic team', 'team': { 'first': p.id, 'second': None, 'third': None, 'fourth': None, 'fifth': None, 'sixth': None } } assert t.to_dict() == expected
def test_006_team_unordered(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t = pokemon.Team(1, 'basic team', first=p) assert t.team(ordered=False) == [p]
def test_002_pokemon_status_pokerus(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) assert 'Pokerus' not in p.status() p = pokemon.Pokemon(1, s, pokerus=True) assert 'Pokerus' in p.status()
def test_003_pokemon_status_item(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s, item='Macho Brace') assert 'Macho Brace' in p.status()
def test_006_pokemon_to_string(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) assert '1' in str(p) assert 'bulbasaur' in str(p)
def test_007_pokemon_to_string_nickname(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s, nick_name='Boots') assert 'Boots (bulbasaur)' in str(p)
def teast_001_team_with_pokemon(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t = pokemon.Team(1, 'basic team', first=p) assert t.get_position(pokemon.TeamPosition.first) == p
def test_004_set_position(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t = pokemon.Team(1, 'basic team') t.set_position(pokemon.TeamPosition.first, p) assert t.get_position(pokemon.TeamPosition.first) == p
def test_005_team_ordered(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t = pokemon.Team(1, 'basic team', first=p) assert t.team() == [p, None, None, None, None, None]
def test_001_pokemon_name_overrides_species_name(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s, nick_name='Boots') assert p.name == 'Boots'
def test_008_team_set_position_type_checks(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) t = pokemon.Team(1, 'basic team') self.assertRaises(ValueError, t.set_position, 'foo', p)
def test_000_pokemon_name_none_uses_species_name(self): s = pokemon.Species(1, 'bulbasaur') p = pokemon.Pokemon(1, s) assert p.name == 'bulbasaur'