PkmnLib_rs/tests/common/data_getter.rs

23 lines
527 B
Rust
Executable File

use serde::Deserialize;
use pkmn_lib::dynamic_data::Battle;
#[derive(Deserialize)]
#[serde(rename_all = "snake_case")]
pub enum TestDataGetter {
PokemonHealth { index: [u8; 2] },
}
impl TestDataGetter {
pub fn get(&self, battle: &Battle) -> String {
match self {
TestDataGetter::PokemonHealth { index } => battle
.get_pokemon(index[0], index[1])
.as_ref()
.unwrap()
.current_health()
.to_string(),
}
}
}