Пример #1
0
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:]
        ]
Пример #2
0
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:]]
Пример #3
0
 def setUp(self):
     self.dist_map = NewEnglandPlantDistributionMap()
     self.legend = Legend(self.dist_map.svg_map, maximum_items=5)
Пример #4
0
 def setUp(self):
     self.dist_map = NewEnglandPlantDistributionMap()
     self.legend = Legend(self.dist_map.svg_map,
                          maximum_categories=2,
                          maximum_items=5)