24 |
MAP_STACKING_CHANGED, LAYER_VISIBILITY_CHANGED, LAYER_LEGEND_CHANGED, \ |
MAP_STACKING_CHANGED, LAYER_VISIBILITY_CHANGED, LAYER_LEGEND_CHANGED, \ |
25 |
LAYER_CHANGED |
LAYER_CHANGED |
26 |
|
|
27 |
|
from Thuban.Model.session import Session |
28 |
from Thuban.Model.map import Map |
from Thuban.Model.map import Map |
29 |
from Thuban.Model.layer import Layer |
from Thuban.Model.layer import Layer |
30 |
from Thuban.Model.proj import Projection |
from Thuban.Model.proj import Projection |
91 |
def test_add_layer(self): |
def test_add_layer(self): |
92 |
"""Test Map.AddLayer""" |
"""Test Map.AddLayer""" |
93 |
# make sure the created Map is unmodified |
# make sure the created Map is unmodified |
94 |
|
session = Session("Test session for %s" % self.__class__) |
95 |
self.failIf(self.map.WasModified()) |
self.failIf(self.map.WasModified()) |
96 |
self.failIf(self.map.HasLayers()) |
self.failIf(self.map.HasLayers()) |
97 |
|
|
98 |
# add a layer and check the result |
# add a layer and check the result |
99 |
roads = Layer("Roads", |
filename = os.path.join("..", "Data", "iceland", "roads-line.shp") |
100 |
os.path.join("..", "Data", "iceland", "roads-line.shp")) |
roads = Layer("Roads", session.OpenShapefile(filename)) |
101 |
self.map.AddLayer(roads) |
self.map.AddLayer(roads) |
102 |
self.assertEquals(self.map.Layers(), [roads]) |
self.assertEquals(self.map.Layers(), [roads]) |
103 |
self.check_messages([(self.map, MAP_LAYERS_CHANGED), |
self.check_messages([(self.map, MAP_LAYERS_CHANGED), |
121 |
received messages is empty. |
received messages is empty. |
122 |
""" |
""" |
123 |
TestMapBase.setUp(self) |
TestMapBase.setUp(self) |
124 |
|
self.session = Session("Test session for %s" % self.__class__) |
125 |
|
open_shp = self.session.OpenShapefile |
126 |
self.arc_layer = Layer("Roads", |
self.arc_layer = Layer("Roads", |
127 |
os.path.join("..", "Data", "iceland", |
open_shp(os.path.join("..", "Data", "iceland", |
128 |
"roads-line.shp")) |
"roads-line.shp"))) |
129 |
self.poly_layer = Layer("Political", |
self.poly_layer = Layer("Political", |
130 |
os.path.join("..", "Data", "iceland", |
open_shp(os.path.join("..", "Data", "iceland", |
131 |
"political.shp")) |
"political.shp"))) |
132 |
self.map.AddLayer(self.arc_layer) |
self.map.AddLayer(self.arc_layer) |
133 |
self.map.AddLayer(self.poly_layer) |
self.map.AddLayer(self.poly_layer) |
134 |
self.map.UnsetModified() |
self.map.UnsetModified() |
135 |
self.clear_messages() |
self.clear_messages() |
136 |
|
|
137 |
|
def tearDown(self): |
138 |
|
TestMapBase.tearDown(self) |
139 |
|
self.session = None |
140 |
|
self.map = None |
141 |
|
self.poly_layer = self.arc_layer = None |
142 |
|
|
143 |
def test_remove_layer(self): |
def test_remove_layer(self): |
144 |
"""Test Map.RemoveLayer""" |
"""Test Map.RemoveLayer""" |
145 |
self.map.RemoveLayer(self.arc_layer) |
self.map.RemoveLayer(self.arc_layer) |
169 |
(self.map, MAP_STACKING_CHANGED)]) |
(self.map, MAP_STACKING_CHANGED)]) |
170 |
self.assert_(self.map.WasModified()) |
self.assert_(self.map.WasModified()) |
171 |
|
|
172 |
|
def test_raise_layer_top(self): |
173 |
|
"""Test Map.MoveLayerToTop""" |
174 |
|
open_shp = self.session.OpenShapefile |
175 |
|
dummy = Layer("Roads", |
176 |
|
open_shp(os.path.join("..", "Data", "iceland", |
177 |
|
"roads-line.shp"))) |
178 |
|
self.map.AddLayer(dummy) |
179 |
|
self.clear_messages() |
180 |
|
|
181 |
|
self.map.MoveLayerToTop(self.poly_layer) |
182 |
|
self.assertEquals(self.map.Layers(), |
183 |
|
[self.arc_layer, dummy, self.poly_layer]) |
184 |
|
self.check_messages([(self.map, MAP_LAYERS_CHANGED), |
185 |
|
(self.map, MAP_STACKING_CHANGED)]) |
186 |
|
self.assert_(self.map.WasModified()) |
187 |
|
|
188 |
|
self.map.RemoveLayer(dummy) |
189 |
|
|
190 |
|
def test_lower_layer_bottom(self): |
191 |
|
"""Test Map.MoveLayerToBottom""" |
192 |
|
open_shp = self.session.OpenShapefile |
193 |
|
dummy = Layer("Roads", |
194 |
|
open_shp(os.path.join("..", "Data", "iceland", |
195 |
|
"roads-line.shp"))) |
196 |
|
self.map.AddLayer(dummy) |
197 |
|
self.clear_messages() |
198 |
|
|
199 |
|
self.map.MoveLayerToBottom(dummy) |
200 |
|
self.assertEquals(self.map.Layers(), |
201 |
|
[dummy, self.arc_layer, self.poly_layer]) |
202 |
|
self.check_messages([(self.map, MAP_LAYERS_CHANGED), |
203 |
|
(self.map, MAP_STACKING_CHANGED)]) |
204 |
|
self.assert_(self.map.WasModified()) |
205 |
|
|
206 |
|
self.map.RemoveLayer(dummy) |
207 |
|
|
208 |
def test_raise_highest_layer(self): |
def test_raise_highest_layer(self): |
209 |
"""Test Map.RaiseLayer with highest layer |
"""Test Map.RaiseLayer with highest layer |
210 |
|
|
254 |
"""Test Map.SetProjection""" |
"""Test Map.SetProjection""" |
255 |
proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"]) |
proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"]) |
256 |
self.map.SetProjection(proj) |
self.map.SetProjection(proj) |
257 |
self.check_messages([(self.map, MAP_PROJECTION_CHANGED)]) |
self.check_messages([(self.map, None, MAP_PROJECTION_CHANGED)]) |
258 |
self.assert_(self.map.WasModified()) |
self.assert_(self.map.WasModified()) |
259 |
|
|
260 |
def test_tree_info(self): |
def test_tree_info(self): |
299 |
(CHANGED,)]) |
(CHANGED,)]) |
300 |
|
|
301 |
if __name__ == "__main__": |
if __name__ == "__main__": |
302 |
unittest.main() |
support.run_tests() |
303 |
|
|