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)