def has_itree_code(self):
# Import done here to prevent circular imports
from treemap.ecobenefits import (itree_code_for_species_in_region,
get_default_region)
if self.species is None:
return False
qs = ITreeRegion.objects.filter(geometry__contains=self.plot.geom)
region = qs[0] if qs else get_default_region(self.species.instance)
itree_code = itree_code_for_species_in_region(self.species, region)
return itree_code is not None
def assert_itree_code(self, species, expected_itree_code):
itree_code = itree_code_for_species_in_region(species, self.region)
self.assertEqual(itree_code, expected_itree_code)
