Source code for safe.gis.test.test_gdal_ogr_tools

# coding=utf-8
"""
InaSAFE Disaster risk assessment tool developed by AusAid and World Bank
- **Converter Test Cases.**

Contact : [email protected]

.. note:: This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.

"""

__author__ = '[email protected]'
__date__ = '20/01/2014'
__copyright__ = ('Copyright 2012, Australia Indonesia Facility for '
                 'Disaster Reduction')

import unittest
from osgeo import ogr

from safe.gis.gdal_ogr_tools import polygonize_thresholds
from safe.test.utilities import test_data_path


[docs]class TestGDALOGRTools(unittest.TestCase):
[docs] def test_polygonize_thresholds(self): """Test polygonize raster using gdal """ raster_path = test_data_path('hazard', 'jakarta_flood_design.tif') inside_file_name, inside_layer_name, outside_file_name, \ outside_layer_name = polygonize_thresholds( raster_path, 0.5) # Syntactic sugar to ignore unused vars. _ = inside_layer_name _ = outside_layer_name driver = ogr.GetDriverByName('ESRI Shapefile') data_source = driver.Open(inside_file_name, 0) layer = data_source.GetLayer() feature_count = layer.GetFeatureCount() # print 'inside %s' % (inside_file_name) self.assertEquals(feature_count, 3) data_source2 = driver.Open(outside_file_name, 0) layer2 = data_source2.GetLayer() feature_count2 = layer2.GetFeatureCount() # print 'outside %s' % (outside_file_name) self.assertEquals(feature_count2, 1)
if __name__ == '__main__': suite = unittest.makeSuite(TestGDALOGRTools, 'test') runner = unittest.TextTestRunner() runner.run(suite)