16 |
import os |
import os |
17 |
import unittest |
import unittest |
18 |
|
|
19 |
|
import mockgeo |
20 |
import support |
import support |
21 |
support.initthuban() |
support.initthuban() |
22 |
|
|
41 |
|
|
42 |
"""Test cases for different layer (shape) types""" |
"""Test cases for different layer (shape) types""" |
43 |
|
|
|
def assertPointListEquals(self, test, value): |
|
|
"""Assert equality of two lists of lists of tuples of float""" |
|
|
for i in range(len(test)): |
|
|
self.assertEquals(len(test[i]), len(value[i])) |
|
|
for j in range(len(test[i])): |
|
|
self.assertFloatSeqEqual(test[i][j], value[i][j]) |
|
|
|
|
44 |
def setUp(self): |
def setUp(self): |
45 |
"""Create a session self.session and initialize self.layer to None""" |
"""Create a session self.session and initialize self.layer to None""" |
46 |
self.session = Session("Test session for %s" % self.__class__) |
self.session = Session("Test session for %s" % self.__class__) |
103 |
self.assertFloatSeqEqual(layer.BoundingBox(), |
self.assertFloatSeqEqual(layer.BoundingBox(), |
104 |
[-24.450359344482422, 63.426830291748047, |
[-24.450359344482422, 63.426830291748047, |
105 |
-13.55668830871582, 66.520111083984375]) |
-13.55668830871582, 66.520111083984375]) |
106 |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)), |
shapes = layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)) |
107 |
|
self.assertEquals([s.ShapeID() for s in shapes], |
108 |
[613, 726, 838]) |
[613, 726, 838]) |
109 |
|
|
110 |
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]), |
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]), |
142 |
self.assertFloatSeqEqual(layer.BoundingBox(), |
self.assertFloatSeqEqual(layer.BoundingBox(), |
143 |
[-24.546524047851562, 63.286754608154297, |
[-24.546524047851562, 63.286754608154297, |
144 |
-13.495815277099609, 66.563774108886719]) |
-13.495815277099609, 66.563774108886719]) |
145 |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.9, 64.1)), |
shapes = layer.ShapesInRegion((-24.0, 64.0, -23.9, 64.1)) |
146 |
|
self.assertEquals([s.ShapeID() for s in shapes], |
147 |
[91, 92, 144, 146, 148, 150, 152, 153]) |
[91, 92, 144, 146, 148, 150, 152, 153]) |
148 |
|
|
149 |
def test_point_layer(self): |
def test_point_layer(self): |
160 |
self.assertFloatSeqEqual(layer.BoundingBox(), |
self.assertFloatSeqEqual(layer.BoundingBox(), |
161 |
[-23.806047439575195, 63.405960083007812, |
[-23.806047439575195, 63.405960083007812, |
162 |
-15.12291431427002, 66.36572265625]) |
-15.12291431427002, 66.36572265625]) |
163 |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.80, 64.1)), |
shapes = layer.ShapesInRegion((-24.0, 64.0, -23.80, 64.1)) |
164 |
|
self.assertEquals([s.ShapeID() for s in shapes], |
165 |
[0, 1, 2, 3, 4, 5, 27, 28, 29, 30, 31]) |
[0, 1, 2, 3, 4, 5, 27, 28, 29, 30, 31]) |
166 |
|
|
167 |
|
def test_point_layer_with_projection(self): |
168 |
|
"""Test Layer with point shapes and a projection""" |
169 |
|
# We use mock data here so that we have precise control over the |
170 |
|
# values |
171 |
|
table = MemoryTable([("FOO", FIELDTYPE_STRING)], [("bla",)]) |
172 |
|
store = mockgeo.SimpleShapeStore(SHAPETYPE_POINT, [[[(10,10)]]], table) |
173 |
|
layer = self.layer = Layer("Test Layer", store) |
174 |
|
|
175 |
|
# Rotation by 45 degrees counter clockwise. This detects a bug |
176 |
|
# in the ShapesInRegion method which transforms the bounding box |
177 |
|
# by only transforming two opposite corners because they have |
178 |
|
# the same x or y coordinates after application of the |
179 |
|
# projection or its inverse. |
180 |
|
proj = mockgeo.AffineProjection((1, 1, -1, 1, 0, 0)) |
181 |
|
layer.SetProjection(proj) |
182 |
|
|
183 |
|
self.assertEquals(layer.BoundingBox(), (10, 10, 10, 10)) |
184 |
|
self.assertEquals(layer.LatLongBoundingBox(), (10.0, 0.0, 10.0, 0.0)) |
185 |
|
shapes = layer.ShapesInRegion((0, 0, 20, 20)) |
186 |
|
self.assertEquals([s.ShapeID() for s in shapes], [0]) |
187 |
|
|
188 |
def test_empty_layer(self): |
def test_empty_layer(self): |
189 |
"""Test Layer with empty shape file""" |
"""Test Layer with empty shape file""" |
190 |
# create an empty shape file |
# create an empty shape file |
213 |
|
|
214 |
def test_raster_layer(self): |
def test_raster_layer(self): |
215 |
if not Thuban.Model.resource.has_gdal_support(): |
if not Thuban.Model.resource.has_gdal_support(): |
216 |
return |
raise support.SkipTest("No gdal support") |
217 |
|
|
218 |
filename = self.build_path("island.tif") |
filename = self.build_path("island.tif") |
219 |
layer = RasterLayer("Test RasterLayer", filename) |
layer = RasterLayer("Test RasterLayer", filename) |
220 |
self.failIf(layer.HasClassification()) |
self.failIf(layer.HasClassification()) |
221 |
self.failIf(layer.HasShapes()) |
self.failIf(layer.HasShapes()) |
222 |
self.assertEquals(layer.GetImageFilename(), filename) |
self.assertEquals(layer.GetImageFilename(), os.path.abspath(filename)) |
223 |
self.assertFloatSeqEqual(layer.BoundingBox(), |
self.assertFloatSeqEqual(layer.BoundingBox(), |
224 |
[-24.5500000, 63.2833330, |
[-24.5500000, 63.2833330, |
225 |
-13.4916670, 66.5666670]) |
-13.4916670, 66.5666670]) |
246 |
self.assertFloatSeqEqual(layer.BoundingBox(), |
self.assertFloatSeqEqual(layer.BoundingBox(), |
247 |
[-24.450359344482422, 63.426830291748047, |
[-24.450359344482422, 63.426830291748047, |
248 |
-13.55668830871582, 66.520111083984375]) |
-13.55668830871582, 66.520111083984375]) |
249 |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, |
shapes = layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)) |
250 |
-23.75, 64.25)), |
self.assertEquals([s.ShapeID() for s in shapes], |
251 |
[613, 726, 838]) |
[613, 726, 838]) |
252 |
|
|
253 |
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]), |
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]), |