@@ -1656,6 +1656,21 @@ def setup_pokemon_stat_data(cls, pokemon, base_stat=10, effort=10):
16561656
16571657 return pokemon_stat
16581658
1659+ @classmethod
1660+ def setup_pokemon_past_stat_data (cls , pokemon , generation , base_stat = 10 , effort = 10 ):
1661+ stat = cls .setup_stat_data (name = "stt for pkmn" )
1662+
1663+ pokemon_stat_past = PokemonStatPast (
1664+ pokemon = pokemon ,
1665+ generation = generation ,
1666+ stat = stat ,
1667+ base_stat = base_stat ,
1668+ effort = effort ,
1669+ )
1670+ pokemon_stat_past .save ()
1671+
1672+ return pokemon_stat_past
1673+
16591674 @classmethod
16601675 def setup_pokemon_type_data (cls , pokemon , type = None , slot = 1 ):
16611676 type = type or cls .setup_type_data (name = "tp for pkmn" )
@@ -4904,6 +4919,9 @@ def test_pokemon_api(self):
49044919 pokemon = pokemon , generation = generation
49054920 )
49064921 pokemon_stat = self .setup_pokemon_stat_data (pokemon = pokemon )
4922+ pokemon_past_stat = self .setup_pokemon_past_stat_data (
4923+ pokemon = pokemon , generation = generation
4924+ )
49074925 pokemon_type = self .setup_pokemon_type_data (pokemon = pokemon )
49084926 pokemon_past_type = self .setup_pokemon_past_type_data (
49094927 pokemon = pokemon , generation = generation
@@ -5029,6 +5047,29 @@ def test_pokemon_api(self):
50295047 response .data ["stats" ][0 ]["stat" ]["url" ],
50305048 "{}{}/stat/{}/" .format (TEST_HOST , API_V2 , pokemon_stat .stat .pk ),
50315049 )
5050+ # past stat params
5051+ past_stats_obj = response .data ["past_stats" ][0 ]
5052+ self .assertEqual (
5053+ past_stats_obj ["generation" ]["name" ],
5054+ pokemon_past_stat .generation .name ,
5055+ )
5056+ self .assertEqual (
5057+ past_stats_obj ["generation" ]["url" ],
5058+ "{}{}/generation/{}/" .format (
5059+ TEST_HOST , API_V2 , pokemon_past_stat .generation .pk
5060+ ),
5061+ )
5062+
5063+ past_stats_stats_obj = past_stats_obj ["stats" ][0 ]
5064+ self .assertEqual (past_stats_stats_obj ["base_stat" ], pokemon_past_stat .base_stat )
5065+ self .assertEqual (past_stats_stats_obj ["effort" ], pokemon_past_stat .effort )
5066+ self .assertEqual (
5067+ past_stats_stats_obj ["stat" ]["name" ], pokemon_past_stat .stat .name
5068+ )
5069+ self .assertEqual (
5070+ past_stats_stats_obj ["stat" ]["url" ],
5071+ "{}{}/stat/{}/" .format (TEST_HOST , API_V2 , pokemon_past_stat .stat .pk ),
5072+ )
50325073 # type params
50335074 self .assertEqual (response .data ["types" ][0 ]["slot" ], pokemon_type .slot )
50345075 self .assertEqual (
@@ -5719,6 +5760,9 @@ def test_case_insensitive_api(self):
57195760 pokemon = pokemon , generation = generation
57205761 )
57215762 pokemon_stat = self .setup_pokemon_stat_data (pokemon = pokemon )
5763+ pokemon_past_stat = self .setup_pokemon_past_stat_data (
5764+ pokemon = pokemon , generation = generation
5765+ )
57225766 pokemon_type = self .setup_pokemon_type_data (pokemon = pokemon )
57235767 pokemon_past_type = self .setup_pokemon_past_type_data (
57245768 pokemon = pokemon , generation = generation
0 commit comments