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)
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)