コード例 #1
0
class QgisLayerWrapperTest(unittest.TestCase):
    """Test that context help works."""

    def setUp(self):
        data_path = os.path.join(
            UNITDATA,
            'impact',
            'aggregation_test_roads.shp')
        self.layer = QgsVectorLayer(
            data_path,
            'test vector',
            'ogr')
        self.wrapper = QgisWrapper(self.layer)

    def test_get_keywords(self):
        """Test get_keywords work
        """
        expected = {
            'map_title': 'Roads inundated',
            'target_field': 'flooded',
            'title': 'aggregation_test_roads'
        }
        keywords = self.wrapper.get_keywords()
        self.assertEquals(keywords, expected)

        expected = 'flooded'
        keyword = self.wrapper.get_keywords('target_field')
        self.assertEquals(keyword, expected)

        self.assertRaises(
            KeywordNotFoundError,
            self.wrapper.get_keywords,
            'ThisKeyDoesNotExist'
        )

    def test_get_name(self):
        """Test get_name work
        """
        name = self.wrapper.get_name()
        expected = 'aggregation_test_roads'
        self.assertEquals(name, expected)

        name = 'NewName'
        self.wrapper.set_name(name)
        self.assertEquals(self.wrapper.get_name(), name)

    def test_get_layer(self):
        """Test get_layer work
        """
        layer = self.wrapper.get_layer()
        self.assertEquals(self.layer, layer)
コード例 #2
0
class QgisLayerWrapperTest(unittest.TestCase):
    """Test that context help works."""
    def setUp(self):
        data_path = os.path.join(UNITDATA, 'impact',
                                 'aggregation_test_roads.shp')
        self.layer = QgsVectorLayer(data_path, 'test vector', 'ogr')
        self.wrapper = QgisWrapper(self.layer)

    def test_get_keywords(self):
        """Test get_keywords work
        """
        expected = {
            'map_title': 'Roads inundated',
            'target_field': 'flooded',
            'title': 'aggregation_test_roads'
        }
        keywords = self.wrapper.get_keywords()
        self.assertEquals(keywords, expected)

        expected = 'flooded'
        keyword = self.wrapper.get_keywords('target_field')
        self.assertEquals(keyword, expected)

        self.assertRaises(KeywordNotFoundError, self.wrapper.get_keywords,
                          'ThisKeyDoesNotExist')

    def test_get_name(self):
        """Test get_name work
        """
        name = self.wrapper.get_name()
        expected = 'aggregation_test_roads'
        self.assertEquals(name, expected)

        name = 'NewName'
        self.wrapper.set_name(name)
        self.assertEquals(self.wrapper.get_name(), name)

    def test_get_layer(self):
        """Test get_layer work
        """
        layer = self.wrapper.get_layer()
        self.assertEquals(self.layer, layer)