class PlantDistributionMapTestCase(TestCase):
def setUp(self):
self.distribution_map = NewEnglandPlantDistributionMap()
create_distribution_records()
def test_map_init(self):
self.assertTrue(self.distribution_map)
def test_get_label_for_status_native(self):
statuses = [
'Species present in state and native',
'Species present and not rare',
'Species native, but adventive in state',
]
for status in statuses:
self.assertEqual('native',
self.distribution_map._get_label_for_status(status))
def test_get_label_for_status_rare(self):
statuses = [
'Species present and rare',
]
for status in statuses:
self.assertEqual('native',
self.distribution_map._get_label_for_status(status))
def test_get_label_for_status_introduced(self):
statuses = [
'Species present in state and exotic',
'Species exotic and present',
'Species waif',
]
for status in statuses:
self.assertEqual('non-native',
self.distribution_map._get_label_for_status(status))
def test_get_label_for_status_invasive(self):
statuses = [
'Species noxious',
]
for status in statuses:
self.assertEqual('non-native',
self.distribution_map._get_label_for_status(status))
def test_get_label_for_status_historic(self):
statuses = [
'Species extirpated (historic)',
'Species extinct',
]
for status in statuses:
self.assertEqual('absent',
self.distribution_map._get_label_for_status(status))
def test_get_label_for_status_absent(self):
statuses = [
'Species not present in state',
'Species eradicated',
'Questionable Presence (cross-hatched)',
]
for status in statuses:
self.assertEqual('absent',
self.distribution_map._get_label_for_status(status))
def test_add_name_to_title(self):
SCIENTIFIC_NAME = 'Tsuga canadensis'
self.distribution_map._add_name_to_title(SCIENTIFIC_NAME)
self.assertEqual('%s: New England Distribution Map' % SCIENTIFIC_NAME,
self.distribution_map.get_title())
def test_get_distribution_records(self):
SCIENTIFIC_NAME = 'Dendrolycopodium dendroideum'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
records = (self.distribution_map._get_distribution_records(
SCIENTIFIC_NAME))
self.assertTrue(len(records) > 0)
def test_set_plant(self):
SCIENTIFIC_NAME = 'Dendrolycopodium dendroideum'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
self.assertEqual(SCIENTIFIC_NAME,
self.distribution_map.scientific_name)
self.assertEqual('%s: New England Distribution Map' % SCIENTIFIC_NAME,
self.distribution_map.get_title())
self.assertTrue(len(self.distribution_map.distribution_records) > 0)
def _check_equal(self, list1, list2):
return list1[1:] == list2[:-1]
def _verify_shaded_counties(self, legend_labels_found):
path_nodes = self.distribution_map.svg_map.findall(
'{http://www.w3.org/2000/svg}path')
paths = [Path(path_node) for path_node in path_nodes]
statuses_verified = []
for path in paths:
style = path.get_style()
status = None
if style.find('fill:#78bf47') > -1:
status = 'native'
elif style.find('fill:#fff') > -1:
status = 'absent'
if status and status not in statuses_verified:
statuses_verified.append(status)
if statuses_verified == legend_labels_found:
break
self.assertEqual(statuses_verified.sort(), legend_labels_found.sort())
def test_shade_counties(self):
SCIENTIFIC_NAME = 'Dendrolycopodium dendroideum'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
legend_labels_found = self.distribution_map._shade_areas()
self._verify_shaded_counties(legend_labels_found)
def test_shade(self):
SCIENTIFIC_NAME = 'Vaccinium vitis-idaea ssp. minus'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
self.distribution_map.shade()
self._verify_shaded_counties(['native', 'absent'])
labels = [label_node.text for label_node in
self.distribution_map.legend.svg_map.xpath('svg:text/svg:tspan',
namespaces=NAMESPACES)]
self.assertEqual('native', labels[0])
self.assertEqual('absent', labels[1])
[self.assertEqual('', label) for label in labels[2:]]
def test_plant_with_distribution_data_has_plant_name_in_title(self):
SCIENTIFIC_NAME = 'Dendrolycopodium dendroideum'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
self.distribution_map.shade()
self.assertEqual('%s: New England Distribution Map' % SCIENTIFIC_NAME,
self.distribution_map.get_title())
def test_plant_with_no_distribution_data_returns_blank_map(self):
SCIENTIFIC_NAME = 'Foo bar'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
self.distribution_map.shade()
# Verify that the map is not shaded.
path_nodes = self.distribution_map.svg_map.findall(
'{http://www.w3.org/2000/svg}path')
paths = [Path(path_node) for path_node in path_nodes]
for path in paths:
style = path.get_style()
self.assertTrue(style.find('fill:#fff') > -1 or
style.find('fill:none') > -1)
# Verify that the legend contains only a 'no data' label.
labels = [label_node.text for label_node in
self.distribution_map.legend.svg_map.xpath('svg:text/svg:tspan',
namespaces=NAMESPACES)]
self.assertEqual(['no data', '', '', '', ''], labels)
def test_plant_with_no_distribution_data_has_no_plant_name_in_title(self):
SCIENTIFIC_NAME = 'Foo bar'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
self.distribution_map.shade()
self.assertEqual('New England Distribution Map',
self.distribution_map.get_title())
def test_plant_with_data_only_under_synonym_returns_shaded_map(self):
# This plant's distribution data is listed only under the
# synonym Vaccinium vitis-idaea ssp. minus.
SCIENTIFIC_NAME = 'Vaccinium vitis-idaea'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
self.distribution_map.shade()
self._verify_shaded_counties(['present', 'absent'])
labels = [label_node.text for label_node in
self.distribution_map.legend.svg_map.xpath('svg:text/svg:tspan',
namespaces=NAMESPACES)]
self.assertEqual(['native', 'absent', '', '', ''], labels)
self.assertEqual('%s: New England Distribution Map' % SCIENTIFIC_NAME,
self.distribution_map.get_title())
