def test_gets_all_moves_when_an_unknown_set_is_there(self):
self.get_sets_mock.return_value = {
constants.SETS: {
"grassknot|hiddenpowerice60|voltswitch|volttackle": 16,
"grassknot|hiddenpowerice60|irontail|poisonjab": 33
},
constants.MOVES: {
"grassknot": 1,
"hiddenpowerice60": 1,
"voltswitch": 1,
"volttackle": 1,
"irontail": 1,
"poisonjab": 1,
}
}
expected_moves = [
Move("tackle"),
Move("grassknot"),
Move("hiddenpowerice60"),
Move("voltswitch"),
Move("volttackle"),
Move("irontail"),
Move("poisonjab")
]
p = Pokemon('pikachu', 100)
p.moves = [Move("tackle")]
p.update_moves_for_random_battles()
self.assertEqual(expected_moves, p.moves)
def test_gets_only_set_available(self):
self.get_sets_mock.return_value = {
constants.SETS: {
"grassknot|hiddenpowerice60|voltswitch|volttackle": 16,
"grassknot|hiddenpowerice60|irontail|volttackle": 33
}
}
expected_moves = [
Move("irontail"),
Move("grassknot"),
Move("hiddenpowerice60"),
Move("volttackle"),
]
p = Pokemon('pikachu', 100)
p.moves = [Move('irontail')]
p.update_moves_for_random_battles()
self.assertEqual(expected_moves, p.moves)
def test_gets_all_possible_moves_when_none_have_yet_been_revealed(self):
self.get_sets_mock.return_value = {
constants.SETS: {
"grassknot|hiddenpowerice60|voltswitch|volttackle": 16,
"grassknot|hiddenpowerice60|irontail|volttackle": 33
}
}
expected_moves = [
Move("grassknot"),
Move("hiddenpowerice60"),
Move("voltswitch"),
Move("volttackle"),
Move("irontail")
]
p = Pokemon('pikachu', 100)
p.moves = list()
p.update_moves_for_random_battles()
self.assertEqual(expected_moves, p.moves)
def test_gets_combination_of_two_sets_available_but_not_the_third(self):
self.get_sets_mock.return_value = {
constants.SETS: {
"grassknot|hiddenpowerice60|voltswitch|volttackle": 16,
"grassknot|hiddenpowerice60|irontail|volttackle": 33,
"tackle|hiddenpowerice60|irontail|volttackle": 33
}
}
expected_moves = [
Move("grassknot"),
Move("hiddenpowerice60"),
Move("voltswitch"),
Move("volttackle"),
Move("irontail")
]
p = Pokemon('pikachu', 100)
p.moves = [Move('grassknot')]
p.update_moves_for_random_battles()
self.assertEqual(expected_moves, p.moves)
def test_no_new_moves_are_added_when_4_have_been_revealed(self):
self.get_sets_mock.return_value = {
constants.SETS: {
"grassknot|hiddenpowerice60|voltswitch|volttackle": 16,
"grassknot|hiddenpowerice60|irontail|volttackle": 33,
"tackle|hiddenpowerice60|irontail|volttackle": 33
}
}
expected_moves = [
Move("grassknot"),
Move("hiddenpowerice60"),
Move("voltswitch"),
Move("volttackle"),
]
p = Pokemon('pikachu', 100)
p.moves = [
Move("grassknot"),
Move("hiddenpowerice60"),
Move("voltswitch"),
Move("volttackle"),
]
p.update_moves_for_random_battles()
self.assertEqual(expected_moves, p.moves)
def test_three_revealed_moves_gets_only_the_possible_sets(self):
self.get_sets_mock.return_value = {
constants.SETS: {
"grassknot|hiddenpowerice60|voltswitch|volttackle": 16,
"grassknot|hiddenpowerice60|irontail|volttackle": 33,
"tackle|hiddenpowerice60|irontail|volttackle": 33
}
}
expected_moves = [
Move('grassknot'),
Move('hiddenpowerice60'),
Move('volttackle'),
Move('voltswitch'),
Move('irontail'),
]
p = Pokemon('pikachu', 100)
p.moves = [
Move('grassknot'),
Move('hiddenpowerice60'),
Move('volttackle'),
]
p.update_moves_for_random_battles()
self.assertEqual(expected_moves, p.moves)
