def set_animal_parameters(species, params): """Set parameters for animal species. :param species: String, name of animal species :param params: Dict with valid parameter specification for species """ if species == "Herbivore": Herbivore.set_params(params) elif species == "Carnivore": Carnivore.set_params(params) else: raise ValueError("species needs to be either Herbivore or Carnivore!")
def reset_carnivore_params(): """ Set parameters of carnivores back to defaults """ yield Carnivore.set_params(Carnivore.p)