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__) |
48 |
|
|
49 |
def tearDown(self): |
def tearDown(self): |
50 |
"""Call the layer's Destroy method and set session and layer to None""" |
"""Call the layer's Destroy method and set session and layer to None""" |
51 |
|
self.session.Destroy() |
52 |
self.session = None |
self.session = None |
53 |
if self.layer is not None: |
if self.layer is not None: |
54 |
self.layer.Destroy() |
self.layer.Destroy() |
104 |
self.assertFloatSeqEqual(layer.BoundingBox(), |
self.assertFloatSeqEqual(layer.BoundingBox(), |
105 |
[-24.450359344482422, 63.426830291748047, |
[-24.450359344482422, 63.426830291748047, |
106 |
-13.55668830871582, 66.520111083984375]) |
-13.55668830871582, 66.520111083984375]) |
107 |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)), |
shapes = layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)) |
108 |
|
self.assertEquals([s.ShapeID() for s in shapes], |
109 |
[613, 726, 838]) |
[613, 726, 838]) |
110 |
|
|
111 |
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]), |
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]), |
143 |
self.assertFloatSeqEqual(layer.BoundingBox(), |
self.assertFloatSeqEqual(layer.BoundingBox(), |
144 |
[-24.546524047851562, 63.286754608154297, |
[-24.546524047851562, 63.286754608154297, |
145 |
-13.495815277099609, 66.563774108886719]) |
-13.495815277099609, 66.563774108886719]) |
146 |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.9, 64.1)), |
shapes = layer.ShapesInRegion((-24.0, 64.0, -23.9, 64.1)) |
147 |
|
self.assertEquals([s.ShapeID() for s in shapes], |
148 |
[91, 92, 144, 146, 148, 150, 152, 153]) |
[91, 92, 144, 146, 148, 150, 152, 153]) |
149 |
|
|
150 |
def test_point_layer(self): |
def test_point_layer(self): |
161 |
self.assertFloatSeqEqual(layer.BoundingBox(), |
self.assertFloatSeqEqual(layer.BoundingBox(), |
162 |
[-23.806047439575195, 63.405960083007812, |
[-23.806047439575195, 63.405960083007812, |
163 |
-15.12291431427002, 66.36572265625]) |
-15.12291431427002, 66.36572265625]) |
164 |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.80, 64.1)), |
shapes = layer.ShapesInRegion((-24.0, 64.0, -23.80, 64.1)) |
165 |
|
self.assertEquals([s.ShapeID() for s in shapes], |
166 |
[0, 1, 2, 3, 4, 5, 27, 28, 29, 30, 31]) |
[0, 1, 2, 3, 4, 5, 27, 28, 29, 30, 31]) |
167 |
|
|
168 |
|
def test_point_layer_with_projection(self): |
169 |
|
"""Test Layer with point shapes and a projection""" |
170 |
|
# We use mock data here so that we have precise control over the |
171 |
|
# values |
172 |
|
table = MemoryTable([("FOO", FIELDTYPE_STRING)], [("bla",)]) |
173 |
|
store = mockgeo.SimpleShapeStore(SHAPETYPE_POINT, [[[(10,10)]]], table) |
174 |
|
layer = self.layer = Layer("Test Layer", store) |
175 |
|
|
176 |
|
# Rotation by 45 degrees counter clockwise. This detects a bug |
177 |
|
# in the ShapesInRegion method which transforms the bounding box |
178 |
|
# by only transforming two opposite corners because they have |
179 |
|
# the same x or y coordinates after application of the |
180 |
|
# projection or its inverse. |
181 |
|
proj = mockgeo.AffineProjection((1, 1, -1, 1, 0, 0)) |
182 |
|
layer.SetProjection(proj) |
183 |
|
|
184 |
|
self.assertEquals(layer.BoundingBox(), (10, 10, 10, 10)) |
185 |
|
self.assertEquals(layer.LatLongBoundingBox(), (10.0, 0.0, 10.0, 0.0)) |
186 |
|
shapes = layer.ShapesInRegion((0, 0, 20, 20)) |
187 |
|
self.assertEquals([s.ShapeID() for s in shapes], [0]) |
188 |
|
|
189 |
def test_empty_layer(self): |
def test_empty_layer(self): |
190 |
"""Test Layer with empty shape file""" |
"""Test Layer with empty shape file""" |
191 |
# create an empty shape file |
# create an empty shape file |
214 |
|
|
215 |
def test_raster_layer(self): |
def test_raster_layer(self): |
216 |
if not Thuban.Model.resource.has_gdal_support(): |
if not Thuban.Model.resource.has_gdal_support(): |
217 |
return |
raise support.SkipTest("No gdal support") |
218 |
|
|
219 |
filename = self.build_path("island.tif") |
filename = self.build_path("island.tif") |
220 |
layer = RasterLayer("Test RasterLayer", filename) |
layer = RasterLayer("Test RasterLayer", filename) |
221 |
self.failIf(layer.HasClassification()) |
self.failIf(layer.HasClassification()) |
222 |
self.failIf(layer.HasShapes()) |
self.failIf(layer.HasShapes()) |
223 |
self.assertEquals(layer.GetImageFilename(), filename) |
self.assertEquals(layer.GetImageFilename(), os.path.abspath(filename)) |
224 |
self.assertFloatSeqEqual(layer.BoundingBox(), |
self.assertFloatSeqEqual(layer.BoundingBox(), |
225 |
[-24.5500000, 63.2833330, |
[-24.5500000, 63.2833330, |
226 |
-13.4916670, 66.5666670]) |
-13.4916670, 66.5666670]) |
247 |
self.assertFloatSeqEqual(layer.BoundingBox(), |
self.assertFloatSeqEqual(layer.BoundingBox(), |
248 |
[-24.450359344482422, 63.426830291748047, |
[-24.450359344482422, 63.426830291748047, |
249 |
-13.55668830871582, 66.520111083984375]) |
-13.55668830871582, 66.520111083984375]) |
250 |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, |
shapes = layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)) |
251 |
-23.75, 64.25)), |
self.assertEquals([s.ShapeID() for s in shapes], |
252 |
[613, 726, 838]) |
[613, 726, 838]) |
253 |
|
|
254 |
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]), |
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]), |