/[thuban]/branches/WIP-pyshapelib-bramz/test/test_layer.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/test/test_layer.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 599 by bh, Thu Apr 3 11:37:13 2003 UTC revision 1278 by jonathan, Fri Jun 20 17:46:58 2003 UTC
# Line 22  support.initthuban() Line 22  support.initthuban()
22  import shapelib  import shapelib
23  import dbflib  import dbflib
24    
25  from Thuban.Model.layer import Layer, SHAPETYPE_POLYGON, SHAPETYPE_ARC, \  from Thuban.Model.session import Session
26       SHAPETYPE_POINT  from Thuban.Model.layer import BaseLayer, Layer, RasterLayer, \
27         SHAPETYPE_POLYGON, SHAPETYPE_ARC, SHAPETYPE_POINT
28  from Thuban.Model.messages import LAYER_LEGEND_CHANGED, \  from Thuban.Model.messages import LAYER_LEGEND_CHANGED, \
29       LAYER_VISIBILITY_CHANGED       LAYER_VISIBILITY_CHANGED, LAYER_SHAPESTORE_REPLACED, LAYER_CHANGED
30  from Thuban.Model.color import Color  from Thuban.Model.color import Color
31    from Thuban.Model.table import FIELDTYPE_DOUBLE, FIELDTYPE_STRING, MemoryTable
32    from Thuban.Model.proj import Projection
33    from Thuban.Model.data import DerivedShapeStore
34    from Thuban.Model.classification import Classification, ClassGroupSingleton
35    
36    import Thuban.Model.resource
37    
38  class TestLayer(unittest.TestCase, support.FileTestMixin,  class TestLayer(unittest.TestCase, support.FileTestMixin,
39                  support.FloatComparisonMixin):                  support.FloatComparisonMixin):
# Line 38  class TestLayer(unittest.TestCase, suppo Line 45  class TestLayer(unittest.TestCase, suppo
45          for i in range(len(test)):          for i in range(len(test)):
46              self.assertFloatSeqEqual(test[i], value[i])              self.assertFloatSeqEqual(test[i], value[i])
47    
48        def setUp(self):
49            """Create a session self.session and initialize self.layer to None"""
50            self.session = Session("Test session for %s" % self.__class__)
51            self.layer = None
52    
53        def tearDown(self):
54            """Call the layer's Destroy method and set session and layer to None"""
55            self.session = None
56            if self.layer is not None:
57                self.layer.Destroy()
58                self.layer = None
59    
60        def build_path(self, filename):
61            return os.path.join("..", "Data", "iceland", filename)
62    
63        def open_shapefile(self, filename):
64            """Open and return a shapestore for filename in the iceland data set"""
65            return self.session.OpenShapefile(self.build_path(filename))
66    
67        def test_base_layer(self):
68            layer = self.layer = BaseLayer("Test BaseLayer")
69            self.assertEquals(layer.Title(), "Test BaseLayer")
70            self.failUnless(layer.Visible())
71    
72            # toggle visibility
73            layer.SetVisible(False)
74            self.failIf(layer.Visible())
75    
76            layer.SetVisible(True)
77            self.failUnless(layer.Visible())
78    
79            self.failIf(layer.HasClassification())
80            self.failIf(layer.HasShapes())
81    
82            self.assertEquals(layer.GetProjection(), None)
83    
84            # set/get projection
85            proj = Projection(["proj=utm", "zone=26"])
86    
87            layer.SetProjection(proj)
88            self.failUnless(layer.GetProjection() is proj)
89    
90            # __init__ with other arguments
91            layer = BaseLayer("Test BaseLayer", False, proj)
92            self.failIf(layer.Visible())
93            self.failUnless(layer.GetProjection() is proj)
94    
95      def test_arc_layer(self):      def test_arc_layer(self):
96          """Test Layer with arc shapes"""          """Test Layer with arc shapes"""
97          layer = Layer("Test Layer",          layer = self.layer = Layer("Test Layer",
98                        os.path.join("..", "Data", "iceland", "roads-line.shp"))                                     self.open_shapefile("roads-line.shp"))
99          self.assertEquals(layer.Title(), "Test Layer")          self.failUnless(layer.HasClassification())
100            self.failUnless(layer.HasShapes())
101          self.assertEquals(layer.ShapeType(), SHAPETYPE_ARC)          self.assertEquals(layer.ShapeType(), SHAPETYPE_ARC)
102          self.assertEquals(layer.NumShapes(), 839)          self.assertEquals(layer.NumShapes(), 839)
103          shape = layer.Shape(32)          shape = layer.Shape(32)
# Line 54  class TestLayer(unittest.TestCase, suppo Line 109  class TestLayer(unittest.TestCase, suppo
109                                    -13.55668830871582, 66.520111083984375])                                    -13.55668830871582, 66.520111083984375])
110          self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)),          self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.75, 64.25)),
111                            [613, 726, 838])                            [613, 726, 838])
112          layer.Destroy()  
113            self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]),
114                              [-15.082174301147461, 66.27339172363281,
115                               -15.026350021362305, 66.27738189697265])
116    
117            shape = layer.Shape(33)
118            self.assertFloatTuplesEqual(shape.Points(),
119                                        [(-22.248506546020508, 66.30645751953125),
120                                         (-22.232730865478516, 66.294075012207031),
121                                         (-22.23158073425293,  66.287689208984375),
122                                         (-22.246318817138672, 66.270065307617188)])
123    
124            self.assertFloatSeqEqual(layer.ShapesBoundingBox([32, 33]),
125                                     [-22.248506546020508, 66.270065307617188,
126                                      -15.026350021362305, 66.30645751953125])
127    
128            self.assertEquals(layer.ShapesBoundingBox([]), None)
129            self.assertEquals(layer.ShapesBoundingBox(None), None)
130    
131      def test_polygon_layer(self):      def test_polygon_layer(self):
132          """Test Layer with polygon shapes"""          """Test Layer with polygon shapes"""
133          layer = Layer("Test Layer",          layer = self.layer = Layer("Test Layer",
134                        os.path.join("..", "Data", "iceland", "political.shp"))                                     self.open_shapefile("political.shp"))
135          self.assertEquals(layer.Title(), "Test Layer")          self.failUnless(layer.HasClassification())
136            self.failUnless(layer.HasShapes())
137          self.assertEquals(layer.ShapeType(), SHAPETYPE_POLYGON)          self.assertEquals(layer.ShapeType(), SHAPETYPE_POLYGON)
138          self.assertEquals(layer.NumShapes(), 156)          self.assertEquals(layer.NumShapes(), 156)
139          shape = layer.Shape(4)          shape = layer.Shape(4)
# Line 74  class TestLayer(unittest.TestCase, suppo Line 147  class TestLayer(unittest.TestCase, suppo
147                                    -13.495815277099609, 66.563774108886719])                                    -13.495815277099609, 66.563774108886719])
148          self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.9, 64.1)),          self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.9, 64.1)),
149                            [91, 92, 144, 146, 148, 150, 152, 153])                            [91, 92, 144, 146, 148, 150, 152, 153])
         layer.Destroy()  
150    
151      def test_point_layer(self):      def test_point_layer(self):
152          """Test Layer with point shapes"""          """Test Layer with point shapes"""
153          layer = Layer("Test Layer",          layer = self.layer = Layer("Test Layer",
154                        os.path.join("..", "Data", "iceland",                             self.open_shapefile("cultural_landmark-point.shp"))
155                                     "cultural_landmark-point.shp"))          self.failUnless(layer.HasClassification())
156          self.assertEquals(layer.Title(), "Test Layer")          self.failUnless(layer.HasShapes())
157          self.assertEquals(layer.ShapeType(), SHAPETYPE_POINT)          self.assertEquals(layer.ShapeType(), SHAPETYPE_POINT)
158          self.assertEquals(layer.NumShapes(), 34)          self.assertEquals(layer.NumShapes(), 34)
159          shape = layer.Shape(0)          shape = layer.Shape(0)
# Line 92  class TestLayer(unittest.TestCase, suppo Line 164  class TestLayer(unittest.TestCase, suppo
164                                    -15.12291431427002, 66.36572265625])                                    -15.12291431427002, 66.36572265625])
165          self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.80, 64.1)),          self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.80, 64.1)),
166                            [0, 1, 2, 3, 4, 5, 27, 28, 29, 30, 31])                            [0, 1, 2, 3, 4, 5, 27, 28, 29, 30, 31])
         layer.Destroy()  
167    
168      def test_empty_layer(self):      def test_empty_layer(self):
169          """Test Layer with empty shape file"""          """Test Layer with empty shape file"""
# Line 107  class TestLayer(unittest.TestCase, suppo Line 178  class TestLayer(unittest.TestCase, suppo
178          dbf.add_field("NAME", dbflib.FTString, 20, 0)          dbf.add_field("NAME", dbflib.FTString, 20, 0)
179    
180          # Now try to open it.          # Now try to open it.
181          layer = Layer("Empty Layer", shapefilename)          layer = self.layer = Layer("Empty Layer",
182                                       self.session.OpenShapefile(shapefilename))
183          self.assertEquals(layer.BoundingBox(), None)          self.assertEquals(layer.BoundingBox(), None)
184          self.assertEquals(layer.LatLongBoundingBox(), None)          self.assertEquals(layer.LatLongBoundingBox(), None)
185          self.assertEquals(layer.NumShapes(), 0)          self.assertEquals(layer.NumShapes(), 0)
186          layer.Destroy()  
187        def test_get_field_type(self):
188            """Test Layer.GetFieldType()"""
189            layer = self.layer = Layer("Test Layer",
190                                       self.open_shapefile("roads-line.shp"))
191            self.assertEquals(layer.GetFieldType("LENGTH"), FIELDTYPE_DOUBLE)
192            self.assertEquals(layer.GetFieldType("non existing"), None)
193    
194        def test_raster_layer(self):
195            if not Thuban.Model.resource.has_gdal_support():
196                return
197    
198            filename = self.build_path("island.tif")
199            layer = RasterLayer("Test RasterLayer", filename)
200            self.failIf(layer.HasClassification())
201            self.failIf(layer.HasShapes())
202            self.assertEquals(layer.GetImageFilename(), filename)
203            self.assertFloatSeqEqual(layer.BoundingBox(),
204                                     [-24.5500000, 63.2833330,
205                                      -13.4916670, 66.5666670])
206            self.assertFloatSeqEqual(layer.LatLongBoundingBox(),
207                                     [-24.5500000, 63.2833330,
208                                      -13.4916670, 66.5666670])
209    
210        def test_derived_store(self):
211            """Test layer with derived store"""
212            layer = self.layer = Layer("Test Layer",
213                                       self.open_shapefile("roads-line.shp"))
214            try:
215                store = layer.ShapeStore()
216                derived = DerivedShapeStore(store, store.Table())
217                layer.SetShapeStore(derived)
218                self.assert_(layer.ShapeStore() is derived)
219    
220                # check that an exception is raised when the number
221                # of shapes differ from the number of rows in a table.
222                layer2 = Layer("Test Layer",
223                               self.open_shapefile("political.shp"))
224                store2 = layer2.ShapeStore()
225                self.assertRaises(ValueError,
226                                  DerivedShapeStore, store2, store.Table())
227    
228            finally:
229                store = derived = None
230                layer2.Destroy()
231                store2 = None
232    
233    
234    class SetShapeStoreTests(unittest.TestCase, support.SubscriberMixin):
235    
236        def setUp(self):
237            """Create a layer with a classification as self.layer"""
238            self.clear_messages()
239            self.session = Session("Test session for %s" % self.__class__)
240            self.shapefilename = os.path.join("..", "Data", "iceland",
241                                              "cultural_landmark-point.dbf")
242            self.store = self.session.OpenShapefile(self.shapefilename)
243            self.layer = Layer("test layer", self.store)
244            self.classification = Classification(field = "CLPTLABEL")
245            self.classification.AppendGroup(ClassGroupSingleton("FARM"))
246            self.layer.SetClassification(self.classification)
247            self.layer.UnsetModified()
248            self.layer.Subscribe(LAYER_SHAPESTORE_REPLACED,
249                                 self.subscribe_with_params,
250                                 LAYER_SHAPESTORE_REPLACED)
251            self.layer.Subscribe(LAYER_CHANGED,
252                                 self.subscribe_with_params, LAYER_CHANGED)
253    
254        def tearDown(self):
255            self.clear_messages()
256            self.layer.Destroy()
257            self.session.Destroy()
258            self.session = self.layer = self.store = self.classification = None
259    
260        def test_sanity(self):
261            """SetShapeStoreTests sanity check
262    
263            Test the initial state of the test case instances after setUp.
264            """
265            cls = self.layer.GetClassification()
266            self.assert_(cls is self.classification)
267            self.assertEquals(cls.GetField(), "CLPTLABEL")
268            self.assertEquals(cls.GetFieldType(), FIELDTYPE_STRING)
269            self.assertEquals(self.layer.GetClassification().GetNumGroups(), 1)
270            self.failIf(self.layer.WasModified())
271    
272        def test_set_shape_store_modified_flag(self):
273            """Test whether Layer.SetShapeStore() sets the modified flag"""
274            memtable = MemoryTable([("FOO", FIELDTYPE_STRING)],
275                          [("bla",)] * self.layer.ShapeStore().Table().NumRows())
276            self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
277    
278            self.assert_(self.layer.WasModified())
279    
280        def test_set_shape_store_different_field_name(self):
281            """Test Layer.SetShapeStore() with different column name"""
282            memtable = MemoryTable([("FOO", FIELDTYPE_STRING)],
283                          [("bla",)] * self.layer.ShapeStore().Table().NumRows())
284            self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
285            # The classification should contain only the default group now.
286            self.assertEquals(self.layer.GetClassification().GetNumGroups(), 0)
287            self.check_messages([(self.layer, LAYER_CHANGED),
288                                 (self.layer, LAYER_SHAPESTORE_REPLACED)])
289    
290        def test_set_shape_store_same_field(self):
291            """Test Layer.SetShapeStore() with same column name and type"""
292            memtable = MemoryTable([("CLPTLABEL", FIELDTYPE_STRING)],
293                          [("bla",)] * self.layer.ShapeStore().Table().NumRows())
294            self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
295            # The classification should be the same as before
296            self.assert_(self.layer.GetClassification() is self.classification)
297            self.check_messages([(self.layer, LAYER_SHAPESTORE_REPLACED)])
298    
299        def test_set_shape_store_same_field_different_type(self):
300            """Test Layer.SetShapeStore() with same column name but different type
301            """
302            memtable = MemoryTable([("CLPTLABEL", FIELDTYPE_DOUBLE)],
303                          [(0.0,)] * self.layer.ShapeStore().Table().NumRows())
304            self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
305            # The classification should contain only the default group now.
306            self.assertEquals(self.layer.GetClassification().GetNumGroups(), 0)
307            self.check_messages([(self.layer, LAYER_CHANGED),
308                                 (self.layer, LAYER_SHAPESTORE_REPLACED)])
309    
310    
311  class TestLayerLegend(unittest.TestCase, support.SubscriberMixin):  class TestLayerLegend(unittest.TestCase, support.SubscriberMixin):
# Line 120  class TestLayerLegend(unittest.TestCase, Line 314  class TestLayerLegend(unittest.TestCase,
314      """      """
315    
316      def setUp(self):      def setUp(self):
317          """Clear the list of received messages and create a layer          """Clear the list of received messages and create a layer and a session
318    
319          The layer is bound to self.layer.          The layer is bound to self.layer and the session to self.session.
320          """          """
321          self.clear_messages()          self.clear_messages()
322            self.session = Session("Test session for %s" % self.__class__)
323            filename = os.path.join("..", "Data", "iceland", "political.shp")
324          self.layer = Layer("Test Layer",          self.layer = Layer("Test Layer",
325                             os.path.join("..", "Data", "iceland",                             self.session.OpenShapefile(filename))
                                         "political.shp"))  
326          self.layer.Subscribe(LAYER_LEGEND_CHANGED, self.subscribe_with_params,          self.layer.Subscribe(LAYER_LEGEND_CHANGED, self.subscribe_with_params,
327                               LAYER_LEGEND_CHANGED)                               LAYER_LEGEND_CHANGED)
328          self.layer.Subscribe(LAYER_VISIBILITY_CHANGED,          self.layer.Subscribe(LAYER_VISIBILITY_CHANGED,
# Line 138  class TestLayerLegend(unittest.TestCase, Line 333  class TestLayerLegend(unittest.TestCase,
333          """Clear the list of received messages and explictly destroy self.layer          """Clear the list of received messages and explictly destroy self.layer
334          """          """
335          self.layer.Destroy()          self.layer.Destroy()
336            self.layer = None
337            self.session.Destroy()
338            self.session = None
339          self.clear_messages()          self.clear_messages()
340    
341      def test_initial_settings(self):      def test_initial_settings(self):
Legend:
Removed from v.599  
changed lines
  Added in v.1278
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26