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) |
66 |
|
|
67 |
def test_polygon_layer(self): |
def test_polygon_layer(self): |
68 |
"""Test Layer with polygon shapes""" |
"""Test Layer with polygon shapes""" |
69 |
layer = Layer("Test Layer", |
filename = os.path.join("..", "Data", "iceland", "political.shp") |
70 |
os.path.join("..", "Data", "iceland", "political.shp")) |
layer = Layer("Test Layer", self.session.OpenShapefile(filename)) |
71 |
self.assertEquals(layer.Title(), "Test Layer") |
self.assertEquals(layer.Title(), "Test Layer") |
72 |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POLYGON) |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POLYGON) |
73 |
self.assertEquals(layer.NumShapes(), 156) |
self.assertEquals(layer.NumShapes(), 156) |
86 |
|
|
87 |
def test_point_layer(self): |
def test_point_layer(self): |
88 |
"""Test Layer with point shapes""" |
"""Test Layer with point shapes""" |
89 |
layer = Layer("Test Layer", |
filename = os.path.join("..", "Data", "iceland", |
90 |
os.path.join("..", "Data", "iceland", |
"cultural_landmark-point.shp") |
91 |
"cultural_landmark-point.shp")) |
layer = Layer("Test Layer", self.session.OpenShapefile(filename)) |
92 |
self.assertEquals(layer.Title(), "Test Layer") |
self.assertEquals(layer.Title(), "Test Layer") |
93 |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POINT) |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POINT) |
94 |
self.assertEquals(layer.NumShapes(), 34) |
self.assertEquals(layer.NumShapes(), 34) |
115 |
dbf.add_field("NAME", dbflib.FTString, 20, 0) |
dbf.add_field("NAME", dbflib.FTString, 20, 0) |
116 |
|
|
117 |
# Now try to open it. |
# Now try to open it. |
118 |
layer = Layer("Empty Layer", shapefilename) |
layer = Layer("Empty Layer", |
119 |
|
self.session.OpenShapefile(shapefilename)) |
120 |
self.assertEquals(layer.BoundingBox(), None) |
self.assertEquals(layer.BoundingBox(), None) |
121 |
self.assertEquals(layer.LatLongBoundingBox(), None) |
self.assertEquals(layer.LatLongBoundingBox(), None) |
122 |
self.assertEquals(layer.NumShapes(), 0) |
self.assertEquals(layer.NumShapes(), 0) |
129 |
""" |
""" |
130 |
|
|
131 |
def setUp(self): |
def setUp(self): |
132 |
"""Clear the list of received messages and create a layer |
"""Clear the list of received messages and create a layer and a session |
133 |
|
|
134 |
The layer is bound to self.layer. |
The layer is bound to self.layer and the session to self.session. |
135 |
""" |
""" |
136 |
self.clear_messages() |
self.clear_messages() |
137 |
|
self.session = Session("Test session for %s" % self.__class__) |
138 |
|
filename = os.path.join("..", "Data", "iceland", "political.shp") |
139 |
self.layer = Layer("Test Layer", |
self.layer = Layer("Test Layer", |
140 |
os.path.join("..", "Data", "iceland", |
self.session.OpenShapefile(filename)) |
|
"political.shp")) |
|
141 |
self.layer.Subscribe(LAYER_LEGEND_CHANGED, self.subscribe_with_params, |
self.layer.Subscribe(LAYER_LEGEND_CHANGED, self.subscribe_with_params, |
142 |
LAYER_LEGEND_CHANGED) |
LAYER_LEGEND_CHANGED) |
143 |
self.layer.Subscribe(LAYER_VISIBILITY_CHANGED, |
self.layer.Subscribe(LAYER_VISIBILITY_CHANGED, |
148 |
"""Clear the list of received messages and explictly destroy self.layer |
"""Clear the list of received messages and explictly destroy self.layer |
149 |
""" |
""" |
150 |
self.layer.Destroy() |
self.layer.Destroy() |
151 |
|
self.layer = None |
152 |
|
self.session.Destroy() |
153 |
|
self.session = None |
154 |
self.clear_messages() |
self.clear_messages() |
155 |
|
|
156 |
def test_initial_settings(self): |
def test_initial_settings(self): |