22 |
import shapelib |
import shapelib |
23 |
import dbflib |
import dbflib |
24 |
|
|
25 |
|
from Thuban.Model.session import Session |
26 |
from Thuban.Model.layer import Layer, SHAPETYPE_POLYGON, SHAPETYPE_ARC, \ |
from Thuban.Model.layer import Layer, SHAPETYPE_POLYGON, SHAPETYPE_ARC, \ |
27 |
SHAPETYPE_POINT |
SHAPETYPE_POINT |
28 |
from Thuban.Model.messages import LAYER_LEGEND_CHANGED, \ |
from Thuban.Model.messages import LAYER_LEGEND_CHANGED, \ |
39 |
for i in range(len(test)): |
for i in range(len(test)): |
40 |
self.assertFloatSeqEqual(test[i], value[i]) |
self.assertFloatSeqEqual(test[i], value[i]) |
41 |
|
|
42 |
|
def setUp(self): |
43 |
|
"""Create a session""" |
44 |
|
self.session = Session("Test session for %s" % self.__class__) |
45 |
|
|
46 |
|
def tearDown(self): |
47 |
|
self.session = None |
48 |
|
|
49 |
def test_arc_layer(self): |
def test_arc_layer(self): |
50 |
"""Test Layer with arc shapes""" |
"""Test Layer with arc shapes""" |
51 |
layer = Layer("Test Layer", |
filename = os.path.join("..", "Data", "iceland", "roads-line.shp") |
52 |
os.path.join("..", "Data", "iceland", "roads-line.shp")) |
layer = Layer("Test Layer", self.session.OpenShapefile(filename)) |
53 |
self.assertEquals(layer.Title(), "Test Layer") |
self.assertEquals(layer.Title(), "Test Layer") |
54 |
self.assertEquals(layer.ShapeType(), SHAPETYPE_ARC) |
self.assertEquals(layer.ShapeType(), SHAPETYPE_ARC) |
55 |
self.assertEquals(layer.NumShapes(), 839) |
self.assertEquals(layer.NumShapes(), 839) |
62 |
-13.55668830871582, 66.520111083984375]) |
-13.55668830871582, 66.520111083984375]) |
63 |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)), |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)), |
64 |
[613, 726, 838]) |
[613, 726, 838]) |
65 |
|
|
66 |
|
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]), |
67 |
|
[-15.082174301147461, 66.27339172363281, |
68 |
|
-15.026350021362305, 66.27738189697265]) |
69 |
|
|
70 |
|
shape = layer.Shape(33) |
71 |
|
self.assertFloatTuplesEqual(shape.Points(), |
72 |
|
[(-22.248506546020508, 66.30645751953125), |
73 |
|
(-22.232730865478516, 66.294075012207031), |
74 |
|
(-22.23158073425293, 66.287689208984375), |
75 |
|
(-22.246318817138672, 66.270065307617188)]) |
76 |
|
|
77 |
|
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32, 33]), |
78 |
|
[-22.248506546020508, 66.270065307617188, |
79 |
|
-15.026350021362305, 66.30645751953125]) |
80 |
|
|
81 |
|
self.assertEquals(layer.ShapesBoundingBox([]), None) |
82 |
|
self.assertEquals(layer.ShapesBoundingBox(None), None) |
83 |
|
|
84 |
layer.Destroy() |
layer.Destroy() |
85 |
|
|
86 |
def test_polygon_layer(self): |
def test_polygon_layer(self): |
87 |
"""Test Layer with polygon shapes""" |
"""Test Layer with polygon shapes""" |
88 |
layer = Layer("Test Layer", |
filename = os.path.join("..", "Data", "iceland", "political.shp") |
89 |
os.path.join("..", "Data", "iceland", "political.shp")) |
layer = Layer("Test Layer", self.session.OpenShapefile(filename)) |
90 |
self.assertEquals(layer.Title(), "Test Layer") |
self.assertEquals(layer.Title(), "Test Layer") |
91 |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POLYGON) |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POLYGON) |
92 |
self.assertEquals(layer.NumShapes(), 156) |
self.assertEquals(layer.NumShapes(), 156) |
105 |
|
|
106 |
def test_point_layer(self): |
def test_point_layer(self): |
107 |
"""Test Layer with point shapes""" |
"""Test Layer with point shapes""" |
108 |
layer = Layer("Test Layer", |
filename = os.path.join("..", "Data", "iceland", |
109 |
os.path.join("..", "Data", "iceland", |
"cultural_landmark-point.shp") |
110 |
"cultural_landmark-point.shp")) |
layer = Layer("Test Layer", self.session.OpenShapefile(filename)) |
111 |
self.assertEquals(layer.Title(), "Test Layer") |
self.assertEquals(layer.Title(), "Test Layer") |
112 |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POINT) |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POINT) |
113 |
self.assertEquals(layer.NumShapes(), 34) |
self.assertEquals(layer.NumShapes(), 34) |
134 |
dbf.add_field("NAME", dbflib.FTString, 20, 0) |
dbf.add_field("NAME", dbflib.FTString, 20, 0) |
135 |
|
|
136 |
# Now try to open it. |
# Now try to open it. |
137 |
layer = Layer("Empty Layer", shapefilename) |
layer = Layer("Empty Layer", |
138 |
|
self.session.OpenShapefile(shapefilename)) |
139 |
self.assertEquals(layer.BoundingBox(), None) |
self.assertEquals(layer.BoundingBox(), None) |
140 |
self.assertEquals(layer.LatLongBoundingBox(), None) |
self.assertEquals(layer.LatLongBoundingBox(), None) |
141 |
self.assertEquals(layer.NumShapes(), 0) |
self.assertEquals(layer.NumShapes(), 0) |
148 |
""" |
""" |
149 |
|
|
150 |
def setUp(self): |
def setUp(self): |
151 |
"""Clear the list of received messages and create a layer |
"""Clear the list of received messages and create a layer and a session |
152 |
|
|
153 |
The layer is bound to self.layer. |
The layer is bound to self.layer and the session to self.session. |
154 |
""" |
""" |
155 |
self.clear_messages() |
self.clear_messages() |
156 |
|
self.session = Session("Test session for %s" % self.__class__) |
157 |
|
filename = os.path.join("..", "Data", "iceland", "political.shp") |
158 |
self.layer = Layer("Test Layer", |
self.layer = Layer("Test Layer", |
159 |
os.path.join("..", "Data", "iceland", |
self.session.OpenShapefile(filename)) |
|
"political.shp")) |
|
160 |
self.layer.Subscribe(LAYER_LEGEND_CHANGED, self.subscribe_with_params, |
self.layer.Subscribe(LAYER_LEGEND_CHANGED, self.subscribe_with_params, |
161 |
LAYER_LEGEND_CHANGED) |
LAYER_LEGEND_CHANGED) |
162 |
self.layer.Subscribe(LAYER_VISIBILITY_CHANGED, |
self.layer.Subscribe(LAYER_VISIBILITY_CHANGED, |
167 |
"""Clear the list of received messages and explictly destroy self.layer |
"""Clear the list of received messages and explictly destroy self.layer |
168 |
""" |
""" |
169 |
self.layer.Destroy() |
self.layer.Destroy() |
170 |
|
self.layer = None |
171 |
|
self.session.Destroy() |
172 |
|
self.session = None |
173 |
self.clear_messages() |
self.clear_messages() |
174 |
|
|
175 |
def test_initial_settings(self): |
def test_initial_settings(self): |