--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/app/django/contrib/gis/tests/test_gdal_ds.py Tue Oct 14 16:00:59 2008 +0000
@@ -0,0 +1,181 @@
+import os, os.path, unittest
+from django.contrib.gis.gdal import DataSource, Envelope, OGRException, OGRIndexError
+from django.contrib.gis.gdal.field import OFTReal, OFTInteger, OFTString
+
+# Path for SHP files
+data_path = os.path.join(os.path.dirname(__file__), 'data')
+def get_ds_file(name, ext):
+ return os.sep.join([data_path, name, name + '.%s' % ext])
+
+# Test SHP data source object
+class TestDS:
+ def __init__(self, name, **kwargs):
+ ext = kwargs.pop('ext', 'shp')
+ self.ds = get_ds_file(name, ext)
+ for key, value in kwargs.items():
+ setattr(self, key, value)
+
+# List of acceptable data sources.
+ds_list = (TestDS('test_point', nfeat=5, nfld=3, geom='POINT', gtype=1, driver='ESRI Shapefile',
+ fields={'dbl' : OFTReal, 'int' : OFTInteger, 'str' : OFTString,},
+ extent=(-1.35011,0.166623,-0.524093,0.824508), # Got extent from QGIS
+ srs_wkt='GEOGCS["GCS_WGS_1984",DATUM["WGS_1984",SPHEROID["WGS_1984",6378137,298.257223563]],PRIMEM["Greenwich",0],UNIT["Degree",0.017453292519943295]]',
+ field_values={'dbl' : [float(i) for i in range(1, 6)], 'int' : range(1, 6), 'str' : [str(i) for i in range(1, 6)]},
+ fids=range(5)),
+ TestDS('test_vrt', ext='vrt', nfeat=3, nfld=3, geom='POINT', gtype=1, driver='VRT',
+ fields={'POINT_X' : OFTString, 'POINT_Y' : OFTString, 'NUM' : OFTString}, # VRT uses CSV, which all types are OFTString.
+ extent=(1.0, 2.0, 100.0, 523.5), # Min/Max from CSV
+ field_values={'POINT_X' : ['1.0', '5.0', '100.0'], 'POINT_Y' : ['2.0', '23.0', '523.5'], 'NUM' : ['5', '17', '23']},
+ fids=range(1,4)),
+ TestDS('test_poly', nfeat=3, nfld=3, geom='POLYGON', gtype=3,
+ driver='ESRI Shapefile',
+ fields={'float' : OFTReal, 'int' : OFTInteger, 'str' : OFTString,},
+ extent=(-1.01513,-0.558245,0.161876,0.839637), # Got extent from QGIS
+ srs_wkt='GEOGCS["GCS_WGS_1984",DATUM["WGS_1984",SPHEROID["WGS_1984",6378137,298.257223563]],PRIMEM["Greenwich",0],UNIT["Degree",0.017453292519943295]]'),
+ )
+
+bad_ds = (TestDS('foo'),
+ )
+
+class DataSourceTest(unittest.TestCase):
+
+ def test01_valid_shp(self):
+ "Testing valid SHP Data Source files."
+
+ for source in ds_list:
+ # Loading up the data source
+ ds = DataSource(source.ds)
+
+ # Making sure the layer count is what's expected (only 1 layer in a SHP file)
+ self.assertEqual(1, len(ds))
+
+ # Making sure GetName works
+ self.assertEqual(source.ds, ds.name)
+
+ # Making sure the driver name matches up
+ self.assertEqual(source.driver, str(ds.driver))
+
+ # Making sure indexing works
+ try:
+ ds[len(ds)]
+ except OGRIndexError:
+ pass
+ else:
+ self.fail('Expected an IndexError!')
+
+ def test02_invalid_shp(self):
+ "Testing invalid SHP files for the Data Source."
+ for source in bad_ds:
+ self.assertRaises(OGRException, DataSource, source.ds)
+
+ def test03a_layers(self):
+ "Testing Data Source Layers."
+ print "\nBEGIN - expecting out of range feature id error; safe to ignore.\n"
+ for source in ds_list:
+ ds = DataSource(source.ds)
+
+ # Incrementing through each layer, this tests DataSource.__iter__
+ for layer in ds:
+ # Making sure we get the number of features we expect
+ self.assertEqual(len(layer), source.nfeat)
+
+ # Making sure we get the number of fields we expect
+ self.assertEqual(source.nfld, layer.num_fields)
+ self.assertEqual(source.nfld, len(layer.fields))
+
+ # Testing the layer's extent (an Envelope), and it's properties
+ self.assertEqual(True, isinstance(layer.extent, Envelope))
+ self.assertAlmostEqual(source.extent[0], layer.extent.min_x, 5)
+ self.assertAlmostEqual(source.extent[1], layer.extent.min_y, 5)
+ self.assertAlmostEqual(source.extent[2], layer.extent.max_x, 5)
+ self.assertAlmostEqual(source.extent[3], layer.extent.max_y, 5)
+
+ # Now checking the field names.
+ flds = layer.fields
+ for f in flds: self.assertEqual(True, f in source.fields)
+
+ # Negative FIDs are not allowed.
+ self.assertRaises(OGRIndexError, layer.__getitem__, -1)
+ self.assertRaises(OGRIndexError, layer.__getitem__, 50000)
+
+ if hasattr(source, 'field_values'):
+ fld_names = source.field_values.keys()
+
+ # Testing `Layer.get_fields` (which uses Layer.__iter__)
+ for fld_name in fld_names:
+ self.assertEqual(source.field_values[fld_name], layer.get_fields(fld_name))
+
+ # Testing `Layer.__getitem__`.
+ for i, fid in enumerate(source.fids):
+ feat = layer[fid]
+ self.assertEqual(fid, feat.fid)
+ # Maybe this should be in the test below, but we might as well test
+ # the feature values here while in this loop.
+ for fld_name in fld_names:
+ self.assertEqual(source.field_values[fld_name][i], feat.get(fld_name))
+ print "\nEND - expecting out of range feature id error; safe to ignore."
+
+ def test03b_layer_slice(self):
+ "Test indexing and slicing on Layers."
+ # Using the first data-source because the same slice
+ # can be used for both the layer and the control values.
+ source = ds_list[0]
+ ds = DataSource(source.ds)
+
+ sl = slice(1, 3)
+ feats = ds[0][sl]
+
+ for fld_name in ds[0].fields:
+ test_vals = [feat.get(fld_name) for feat in feats]
+ control_vals = source.field_values[fld_name][sl]
+ self.assertEqual(control_vals, test_vals)
+
+ def test04_features(self):
+ "Testing Data Source Features."
+ for source in ds_list:
+ ds = DataSource(source.ds)
+
+ # Incrementing through each layer
+ for layer in ds:
+ # Incrementing through each feature in the layer
+ for feat in layer:
+ # Making sure the number of fields, and the geometry type
+ # are what's expected.
+ self.assertEqual(source.nfld, len(list(feat)))
+ self.assertEqual(source.gtype, feat.geom_type)
+
+ # Making sure the fields match to an appropriate OFT type.
+ for k, v in source.fields.items():
+ # Making sure we get the proper OGR Field instance, using
+ # a string value index for the feature.
+ self.assertEqual(True, isinstance(feat[k], v))
+
+ # Testing Feature.__iter__
+ for fld in feat: self.assertEqual(True, fld.name in source.fields.keys())
+
+ def test05_geometries(self):
+ "Testing Geometries from Data Source Features."
+ for source in ds_list:
+ ds = DataSource(source.ds)
+
+ # Incrementing through each layer and feature.
+ for layer in ds:
+ for feat in layer:
+ g = feat.geom
+
+ # Making sure we get the right Geometry name & type
+ self.assertEqual(source.geom, g.geom_name)
+ self.assertEqual(source.gtype, g.geom_type)
+
+ # Making sure the SpatialReference is as expected.
+ if hasattr(source, 'srs_wkt'):
+ self.assertEqual(source.srs_wkt, g.srs.wkt)
+
+
+def suite():
+ s = unittest.TestSuite()
+ s.addTest(unittest.makeSuite(DataSourceTest))
+ return s
+
+def run(verbosity=2):
+ unittest.TextTestRunner(verbosity=verbosity).run(suite())