class LegendTestCase(TestCase): def setUp(self): self.dist_map = NewEnglandPlantDistributionMap() self.legend = Legend(self.dist_map.svg_map, maximum_categories=2, maximum_items=5) def test_set_item_label(self): LABEL = 'native' label_node = self.legend.svg_map.xpath('svg:text', namespaces=NAMESPACES)[0] self.legend._set_item_label(label_node, LABEL) label_text_node = label_node.find('{http://www.w3.org/2000/svg}tspan') self.assertEqual(LABEL, label_text_node.text) def _get_paths(self): return [ Path(box_node) for box_node in self.legend.svg_map.xpath('svg:rect', namespaces=NAMESPACES) ] def test_set_item(self): SLOT = 1 FILL_COLOR = '#ff0' STROKE_COLOR = '#ccc' LABEL = 'county documented' self.legend._set_item(SLOT, FILL_COLOR, STROKE_COLOR, LABEL) labels = get_legend_labels(self.legend) self.assertEqual('county documented', labels[0]) paths = self._get_paths() self.assertTrue(paths[0].get_style().find('fill:%s' % FILL_COLOR) > -1) self.assertTrue( paths[0].get_style().find('stroke:%s' % STROKE_COLOR) > -1) def test_show_items(self): legend_labels_found = ['county documented na'] self.legend.show_items(legend_labels_found) labels = get_legend_labels(self.legend) self.assertEqual('county documented', labels[0]) paths = self._get_paths() self.assertTrue(paths[0].get_style().find('fill:#35880c') > -1) self.assertTrue(paths[1].get_style().find('fill:#fff') > -1) [ self.assertTrue(path.get_style().find('fill:#fff') > -1) for path in paths[2:] ] [ self.assertTrue(path.get_style().find('stroke:#fff') > -1) for path in paths[0:2] ] [ self.assertTrue(path.get_style().find('stroke:#fff') > -1) for path in paths[2:] ]
class LegendTestCase(TestCase): def setUp(self): self.dist_map = NewEnglandPlantDistributionMap() self.legend = Legend(self.dist_map.svg_map, maximum_items=5) def test_set_item_label(self): LABEL = 'native' label_node = self.legend.svg_map.xpath('svg:text', namespaces=NAMESPACES)[0] self.legend._set_item_label(label_node, LABEL) label_text_node = label_node.find('{http://www.w3.org/2000/svg}tspan') self.assertEqual(LABEL, label_text_node.text) def _get_labels(self): return [label_node.text for label_node in self.legend.svg_map.xpath( 'svg:text/svg:tspan', namespaces=NAMESPACES)] def _get_paths(self): return [Path(box_node) for box_node in self.legend.svg_map.xpath( 'svg:rect', namespaces=NAMESPACES)] def test_set_item(self): SLOT = 1 FILL_COLOR = '#ff0' STROKE_COLOR = '#ccc' LABEL = 'native' self.legend._set_item(SLOT, FILL_COLOR, STROKE_COLOR, LABEL) labels = self._get_labels() self.assertEqual('native', labels[0]) paths = self._get_paths() self.assertTrue(paths[0].get_style().find( 'fill:%s' % FILL_COLOR) > -1) self.assertTrue(paths[0].get_style().find( 'stroke:%s' % STROKE_COLOR) > -1) def test_show_items(self): legend_labels_found = ['native', 'absent'] self.legend.show_items(legend_labels_found) labels = self._get_labels() self.assertEqual('native', labels[0]) self.assertEqual('absent', labels[1]) [self.assertEqual('', label) for label in labels[2:]] paths = self._get_paths() self.assertTrue(paths[0].get_style().find('fill:#78bf47') > -1) self.assertTrue(paths[1].get_style().find('fill:#fff') > -1) [self.assertTrue(path.get_style().find('fill:#fff') > -1) for path in paths[2:]] [self.assertTrue(path.get_style().find('stroke:#000') > -1) for path in paths[0:2]] [self.assertTrue(path.get_style().find('stroke:#fff') > -1) for path in paths[2:]]
def setUp(self): self.dist_map = NewEnglandPlantDistributionMap() self.legend = Legend(self.dist_map.svg_map, maximum_items=5)
def setUp(self): self.dist_map = NewEnglandPlantDistributionMap() self.legend = Legend(self.dist_map.svg_map, maximum_categories=2, maximum_items=5)