28 |
from Thuban.Model.messages import LAYER_LEGEND_CHANGED, \ |
from Thuban.Model.messages import LAYER_LEGEND_CHANGED, \ |
29 |
LAYER_VISIBILITY_CHANGED |
LAYER_VISIBILITY_CHANGED |
30 |
from Thuban.Model.color import Color |
from Thuban.Model.color import Color |
31 |
|
from Thuban.Model.table import FIELDTYPE_DOUBLE |
32 |
|
|
33 |
class TestLayer(unittest.TestCase, support.FileTestMixin, |
class TestLayer(unittest.TestCase, support.FileTestMixin, |
34 |
support.FloatComparisonMixin): |
support.FloatComparisonMixin): |
47 |
def tearDown(self): |
def tearDown(self): |
48 |
self.session = None |
self.session = None |
49 |
|
|
50 |
|
def open_shapefile(self, filename): |
51 |
|
"""Open and return a shapestore for filename in the iceland data set""" |
52 |
|
return self.session.OpenShapefile(os.path.join("..", "Data", "iceland", |
53 |
|
filename)) |
54 |
|
|
55 |
def test_arc_layer(self): |
def test_arc_layer(self): |
56 |
"""Test Layer with arc shapes""" |
"""Test Layer with arc shapes""" |
57 |
filename = os.path.join("..", "Data", "iceland", "roads-line.shp") |
layer = Layer("Test Layer", self.open_shapefile("roads-line.shp")) |
|
layer = Layer("Test Layer", self.session.OpenShapefile(filename)) |
|
58 |
self.assertEquals(layer.Title(), "Test Layer") |
self.assertEquals(layer.Title(), "Test Layer") |
59 |
self.assertEquals(layer.ShapeType(), SHAPETYPE_ARC) |
self.assertEquals(layer.ShapeType(), SHAPETYPE_ARC) |
60 |
self.assertEquals(layer.NumShapes(), 839) |
self.assertEquals(layer.NumShapes(), 839) |
67 |
-13.55668830871582, 66.520111083984375]) |
-13.55668830871582, 66.520111083984375]) |
68 |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)), |
self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)), |
69 |
[613, 726, 838]) |
[613, 726, 838]) |
70 |
|
|
71 |
|
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]), |
72 |
|
[-15.082174301147461, 66.27339172363281, |
73 |
|
-15.026350021362305, 66.27738189697265]) |
74 |
|
|
75 |
|
shape = layer.Shape(33) |
76 |
|
self.assertFloatTuplesEqual(shape.Points(), |
77 |
|
[(-22.248506546020508, 66.30645751953125), |
78 |
|
(-22.232730865478516, 66.294075012207031), |
79 |
|
(-22.23158073425293, 66.287689208984375), |
80 |
|
(-22.246318817138672, 66.270065307617188)]) |
81 |
|
|
82 |
|
self.assertFloatSeqEqual(layer.ShapesBoundingBox([32, 33]), |
83 |
|
[-22.248506546020508, 66.270065307617188, |
84 |
|
-15.026350021362305, 66.30645751953125]) |
85 |
|
|
86 |
|
self.assertEquals(layer.ShapesBoundingBox([]), None) |
87 |
|
self.assertEquals(layer.ShapesBoundingBox(None), None) |
88 |
|
|
89 |
layer.Destroy() |
layer.Destroy() |
90 |
|
|
91 |
def test_polygon_layer(self): |
def test_polygon_layer(self): |
92 |
"""Test Layer with polygon shapes""" |
"""Test Layer with polygon shapes""" |
93 |
filename = os.path.join("..", "Data", "iceland", "political.shp") |
layer = Layer("Test Layer", self.open_shapefile("political.shp")) |
|
layer = Layer("Test Layer", self.session.OpenShapefile(filename)) |
|
94 |
self.assertEquals(layer.Title(), "Test Layer") |
self.assertEquals(layer.Title(), "Test Layer") |
95 |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POLYGON) |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POLYGON) |
96 |
self.assertEquals(layer.NumShapes(), 156) |
self.assertEquals(layer.NumShapes(), 156) |
109 |
|
|
110 |
def test_point_layer(self): |
def test_point_layer(self): |
111 |
"""Test Layer with point shapes""" |
"""Test Layer with point shapes""" |
112 |
filename = os.path.join("..", "Data", "iceland", |
layer = Layer("Test Layer", |
113 |
"cultural_landmark-point.shp") |
self.open_shapefile("cultural_landmark-point.shp")) |
|
layer = Layer("Test Layer", self.session.OpenShapefile(filename)) |
|
114 |
self.assertEquals(layer.Title(), "Test Layer") |
self.assertEquals(layer.Title(), "Test Layer") |
115 |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POINT) |
self.assertEquals(layer.ShapeType(), SHAPETYPE_POINT) |
116 |
self.assertEquals(layer.NumShapes(), 34) |
self.assertEquals(layer.NumShapes(), 34) |
144 |
self.assertEquals(layer.NumShapes(), 0) |
self.assertEquals(layer.NumShapes(), 0) |
145 |
layer.Destroy() |
layer.Destroy() |
146 |
|
|
147 |
|
def test_get_field_type(self): |
148 |
|
"""Test Layer.GetFieldType()""" |
149 |
|
layer = Layer("Test Layer", self.open_shapefile("roads-line.shp")) |
150 |
|
self.assertEquals(layer.GetFieldType("LENGTH"), FIELDTYPE_DOUBLE) |
151 |
|
self.assertEquals(layer.GetFieldType("non existing"), None) |
152 |
|
layer.Destroy() |
153 |
|
|
154 |
|
|
155 |
class TestLayerLegend(unittest.TestCase, support.SubscriberMixin): |
class TestLayerLegend(unittest.TestCase, support.SubscriberMixin): |
156 |
|
|