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) |
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 |
|
|