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_absent(self):
self.assertEqual('absent',
self.distribution_map._get_label(False, False))
def test_get_label_native(self):
self.assertEqual('native',
self.distribution_map._get_label(True, True))
def test_get_label_non_native(self):
self.assertEqual('non-native',
self.distribution_map._get_label(True, False))
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(['county documented', 'absent'])
labels = get_legend_labels(self.distribution_map.legend)
self.assertEqual('county documented', labels[0])
self.assertEqual('state documented', labels[1])
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 = get_legend_labels(self.distribution_map.legend)
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(['county documented', 'absent'])
labels = get_legend_labels(self.distribution_map.legend)
self.assertEqual(
['county documented', 'state documented', '', '', 'Native', ''],
labels)
self.assertEqual('%s: New England Distribution Map' % SCIENTIFIC_NAME,
self.distribution_map.get_title())
def test_legend_correct_with_conflicting_state_and_county_records(self):
# Ensure that if all of a plant's county-level records override
# its state-level record, that the map legend lists only those
# items that are visible on the final map.
SCIENTIFIC_NAME = 'Sambucus nigra'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
self.distribution_map.shade()
labels = get_legend_labels(self.distribution_map.legend)
legend_shows_non_native = ('non-native' in labels)
self.assertFalse(legend_shows_non_native)
def test_species_and_infraspecific_taxa_shaded_together(self):
# Ensure that the distribution records for a species and any
# associated infraspecific taxa are shaded together on the map,
# with native overriding non-native.
SCIENTIFIC_NAME = 'Sambucus nigra'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
self.distribution_map.shade()
labels = get_legend_labels(self.distribution_map.legend)
legend_shows_native = ('native, st.' or 'native, co.' in labels)
self.assertTrue(legend_shows_native)
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_absent(self):
self.assertEqual('absent',
self.distribution_map._get_label(False, False))
def test_get_label_native(self):
self.assertEqual('native',
self.distribution_map._get_label(True, True))
def test_get_label_non_native(self):
self.assertEqual('non-native',
self.distribution_map._get_label(True, False))
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(['county documented', 'absent'])
labels = get_legend_labels(self.distribution_map.legend)
self.assertEqual('county documented', labels[0])
self.assertEqual('state documented', labels[1])
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 = get_legend_labels(self.distribution_map.legend)
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(['county documented', 'absent'])
labels = get_legend_labels(self.distribution_map.legend)
self.assertEqual(['county documented', 'state documented', '', '',
'Native', ''], labels)
self.assertEqual('%s: New England Distribution Map' % SCIENTIFIC_NAME,
self.distribution_map.get_title())
def test_legend_correct_with_conflicting_state_and_county_records(self):
# Ensure that if all of a plant's county-level records override
# its state-level record, that the map legend lists only those
# items that are visible on the final map.
SCIENTIFIC_NAME = 'Sambucus nigra'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
self.distribution_map.shade()
labels = get_legend_labels(self.distribution_map.legend)
legend_shows_non_native = ('non-native' in labels)
self.assertFalse(legend_shows_non_native)
def test_species_and_infraspecific_taxa_shaded_together(self):
# Ensure that the distribution records for a species and any
# associated infraspecific taxa are shaded together on the map,
# with native overriding non-native.
SCIENTIFIC_NAME = 'Sambucus nigra'
self.distribution_map.set_plant(SCIENTIFIC_NAME)
self.distribution_map.shade()
labels = get_legend_labels(self.distribution_map.legend)
legend_shows_native = ('native, st.' or 'native, co.' in labels)
self.assertTrue(legend_shows_native)
