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'
