/[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 1088 by bh, Wed May 28 12:42:23 2003 UTC revision 1538 by bh, Fri Aug 1 14:27:46 2003 UTC
# Line 23  import shapelib Line 23  import shapelib
23  import dbflib  import dbflib
24    
25  from Thuban.Model.session import Session  from Thuban.Model.session import Session
26  from Thuban.Model.layer import BaseLayer, Layer, RasterLayer, \  from Thuban.Model.layer import BaseLayer, Layer, RasterLayer
27       SHAPETYPE_POLYGON, SHAPETYPE_ARC, SHAPETYPE_POINT  from Thuban.Model.data import 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
 from Thuban.Model.color import Color  
30  from Thuban.Model.table import FIELDTYPE_DOUBLE, FIELDTYPE_STRING, MemoryTable  from Thuban.Model.table import FIELDTYPE_DOUBLE, FIELDTYPE_STRING, MemoryTable
31  from Thuban.Model.proj import Projection  from Thuban.Model.proj import Projection
32  from Thuban.Model.data import DerivedShapeStore  from Thuban.Model.data import DerivedShapeStore
33  from Thuban.Model.classification import Classification, ClassGroupSingleton  from Thuban.Model.classification import Classification, ClassGroupSingleton, \
34         ClassGroupRange
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 75  class TestLayer(unittest.TestCase, suppo Line 77  class TestLayer(unittest.TestCase, suppo
77          self.failUnless(layer.Visible())          self.failUnless(layer.Visible())
78    
79          self.failIf(layer.HasClassification())          self.failIf(layer.HasClassification())
80            self.failIf(layer.HasShapes())
81    
82          self.assertEquals(layer.GetProjection(), None)          self.assertEquals(layer.GetProjection(), None)
83    
84          # set/get projection          # set/get projection
# Line 92  class TestLayer(unittest.TestCase, suppo Line 96  class TestLayer(unittest.TestCase, suppo
96          """Test Layer with arc shapes"""          """Test Layer with arc shapes"""
97          layer = self.layer = Layer("Test Layer",          layer = self.layer = Layer("Test Layer",
98                                     self.open_shapefile("roads-line.shp"))                                     self.open_shapefile("roads-line.shp"))
99            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 126  class TestLayer(unittest.TestCase, suppo Line 132  class TestLayer(unittest.TestCase, suppo
132          """Test Layer with polygon shapes"""          """Test Layer with polygon shapes"""
133          layer = self.layer = Layer("Test Layer",          layer = self.layer = Layer("Test Layer",
134                                     self.open_shapefile("political.shp"))                                     self.open_shapefile("political.shp"))
135            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 144  class TestLayer(unittest.TestCase, suppo Line 152  class TestLayer(unittest.TestCase, suppo
152          """Test Layer with point shapes"""          """Test Layer with point shapes"""
153          layer = self.layer = Layer("Test Layer",          layer = self.layer = Layer("Test Layer",
154                             self.open_shapefile("cultural_landmark-point.shp"))                             self.open_shapefile("cultural_landmark-point.shp"))
155            self.failUnless(layer.HasClassification())
156            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 182  class TestLayer(unittest.TestCase, suppo Line 192  class TestLayer(unittest.TestCase, suppo
192          self.assertEquals(layer.GetFieldType("non existing"), None)          self.assertEquals(layer.GetFieldType("non existing"), None)
193    
194      def test_raster_layer(self):      def test_raster_layer(self):
195            if not Thuban.Model.resource.has_gdal_support():
196                return
197    
198          filename = self.build_path("island.tif")          filename = self.build_path("island.tif")
199          layer = RasterLayer("Test RasterLayer", filename)          layer = RasterLayer("Test RasterLayer", filename)
200            self.failIf(layer.HasClassification())
201            self.failIf(layer.HasShapes())
202          self.assertEquals(layer.GetImageFilename(), filename)          self.assertEquals(layer.GetImageFilename(), filename)
203          self.assertFloatSeqEqual(layer.BoundingBox(),          self.assertFloatSeqEqual(layer.BoundingBox(),
204                                   [-24.5500000, 63.2833330,                                   [-24.5500000, 63.2833330,
# Line 201  class TestLayer(unittest.TestCase, suppo Line 216  class TestLayer(unittest.TestCase, suppo
216              derived = DerivedShapeStore(store, store.Table())              derived = DerivedShapeStore(store, store.Table())
217              layer.SetShapeStore(derived)              layer.SetShapeStore(derived)
218              self.assert_(layer.ShapeStore() is derived)              self.assert_(layer.ShapeStore() is derived)
219    
220                self.assertEquals(layer.ShapeType(), SHAPETYPE_ARC)
221                self.assertEquals(layer.NumShapes(), 839)
222                shape = layer.Shape(32)
223                self.assertFloatTuplesEqual(shape.Points(),
224                                        [(-15.082174301147461, 66.27738189697265),
225                                         (-15.026350021362305, 66.27339172363281)])
226                self.assertFloatSeqEqual(layer.BoundingBox(),
227                                         [-24.450359344482422, 63.426830291748047,
228                                          -13.55668830871582, 66.520111083984375])
229                self.assertEquals(layer.ShapesInRegion((-24.0, 64.0,
230                                                        -23.75, 64.25)),
231                                  [613, 726, 838])
232    
233                self.assertFloatSeqEqual(layer.ShapesBoundingBox([32]),
234                                         [-15.082174301147461, 66.27339172363281,
235                                          -15.026350021362305, 66.27738189697265])
236    
237          finally:          finally:
238              store = derived = None              store = derived = None
239    
240    
241  class SetShapeStoreTests(unittest.TestCase):  class SetShapeStoreTests(unittest.TestCase, support.SubscriberMixin):
242    
243      def setUp(self):      def setUp(self):
244          """Create a layer with a classification as self.layer"""          """Create a layer with a classification as self.layer"""
245            self.clear_messages()
246          self.session = Session("Test session for %s" % self.__class__)          self.session = Session("Test session for %s" % self.__class__)
247          self.shapefilename = os.path.join("..", "Data", "iceland",          self.shapefilename = os.path.join("..", "Data", "iceland",
248                                            "cultural_landmark-point.dbf")                                            "cultural_landmark-point.dbf")
249          self.store = self.session.OpenShapefile(self.shapefilename)          self.store = self.session.OpenShapefile(self.shapefilename)
250          self.layer = Layer("test layer", self.store)          self.layer = Layer("test layer", self.store)
251          self.classification = Classification(field = "CLPTLABEL")          self.classification = Classification()
252          self.classification.AppendGroup(ClassGroupSingleton("FARM"))          self.classification.AppendGroup(ClassGroupSingleton("FARM"))
253            self.layer.SetClassificationColumn("CLPTLABEL")
254          self.layer.SetClassification(self.classification)          self.layer.SetClassification(self.classification)
255            self.layer.UnsetModified()
256            self.layer.Subscribe(LAYER_SHAPESTORE_REPLACED,
257                                 self.subscribe_with_params,
258                                 LAYER_SHAPESTORE_REPLACED)
259            self.layer.Subscribe(LAYER_CHANGED,
260                                 self.subscribe_with_params, LAYER_CHANGED)
261    
262      def tearDown(self):      def tearDown(self):
263            self.clear_messages()
264          self.layer.Destroy()          self.layer.Destroy()
265          self.session.Destroy()          self.session.Destroy()
266          self.session = self.layer = self.store = self.classification = None          self.session = self.layer = self.store = self.classification = None
267    
268      def test_sanity(self):      def test_sanity(self):
269          """SetShapeStoreTests sanity check"""          """SetShapeStoreTests sanity check
270    
271            Test the initial state of the test case instances after setUp.
272            """
273          cls = self.layer.GetClassification()          cls = self.layer.GetClassification()
274          self.assert_(cls is self.classification)          self.assert_(cls is self.classification)
275          self.assertEquals(cls.GetField(), "CLPTLABEL")          field = self.layer.GetClassificationColumn()
276          self.assertEquals(cls.GetFieldType(), FIELDTYPE_STRING)          self.assertEquals(field, "CLPTLABEL")
277            self.assertEquals(self.layer.GetFieldType(field), FIELDTYPE_STRING)
278          self.assertEquals(self.layer.GetClassification().GetNumGroups(), 1)          self.assertEquals(self.layer.GetClassification().GetNumGroups(), 1)
279            self.failIf(self.layer.WasModified())
280    
281        def test_set_shape_store_modified_flag(self):
282            """Test whether Layer.SetShapeStore() sets the modified flag"""
283            memtable = MemoryTable([("FOO", FIELDTYPE_STRING)],
284                          [("bla",)] * self.layer.ShapeStore().Table().NumRows())
285            self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
286    
287            self.assert_(self.layer.WasModified())
288    
289      def test_set_shape_store_different_field_name(self):      def test_set_shape_store_different_field_name(self):
290          """Test Layer.SetShapeStore() with different column name"""          """Test Layer.SetShapeStore() with different column name"""
# Line 238  class SetShapeStoreTests(unittest.TestCa Line 293  class SetShapeStoreTests(unittest.TestCa
293          self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))          self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
294          # The classification should contain only the default group now.          # The classification should contain only the default group now.
295          self.assertEquals(self.layer.GetClassification().GetNumGroups(), 0)          self.assertEquals(self.layer.GetClassification().GetNumGroups(), 0)
296            self.check_messages([(self.layer, LAYER_CHANGED),
297                                 (self.layer, LAYER_SHAPESTORE_REPLACED)])
298    
299      def test_set_shape_store_same_field(self):      def test_set_shape_store_same_field(self):
300          """Test Layer.SetShapeStore() with same column name and type"""          """Test Layer.SetShapeStore() with same column name and type"""
# Line 246  class SetShapeStoreTests(unittest.TestCa Line 303  class SetShapeStoreTests(unittest.TestCa
303          self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))          self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
304          # The classification should be the same as before          # The classification should be the same as before
305          self.assert_(self.layer.GetClassification() is self.classification)          self.assert_(self.layer.GetClassification() is self.classification)
306            self.check_messages([(self.layer, LAYER_SHAPESTORE_REPLACED)])
307    
308      def test_set_shape_store_same_field_different_type(self):      def test_set_shape_store_same_field_different_type(self):
309          """Test Layer.SetShapeStore() with same column name but different type          """Test Layer.SetShapeStore() with same column name but different type
# Line 255  class SetShapeStoreTests(unittest.TestCa Line 313  class SetShapeStoreTests(unittest.TestCa
313          self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))          self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
314          # The classification should contain only the default group now.          # The classification should contain only the default group now.
315          self.assertEquals(self.layer.GetClassification().GetNumGroups(), 0)          self.assertEquals(self.layer.GetClassification().GetNumGroups(), 0)
316            self.check_messages([(self.layer, LAYER_CHANGED),
317                                 (self.layer, LAYER_SHAPESTORE_REPLACED)])
318    
319    
320  class TestLayerLegend(unittest.TestCase, support.SubscriberMixin):  class TestLayerModification(unittest.TestCase, support.SubscriberMixin):
321    
322      """Test cases for Layer method that modify the layer.      """Test cases for Layer method that modify the layer.
323      """      """
# Line 277  class TestLayerLegend(unittest.TestCase, Line 337  class TestLayerLegend(unittest.TestCase,
337          self.layer.Subscribe(LAYER_VISIBILITY_CHANGED,          self.layer.Subscribe(LAYER_VISIBILITY_CHANGED,
338                               self.subscribe_with_params,                               self.subscribe_with_params,
339                               LAYER_VISIBILITY_CHANGED)                               LAYER_VISIBILITY_CHANGED)
340            self.layer.Subscribe(LAYER_CHANGED, self.subscribe_with_params,
341                                 LAYER_CHANGED)
342    
343      def tearDown(self):      def tearDown(self):
344          """Clear the list of received messages and explictly destroy self.layer          """Clear the list of received messages and explictly destroy self.layer
# Line 287  class TestLayerLegend(unittest.TestCase, Line 349  class TestLayerLegend(unittest.TestCase,
349          self.session = None          self.session = None
350          self.clear_messages()          self.clear_messages()
351    
352      def test_initial_settings(self):      def test_sanity(self):
353          """Test Layer's initial legend attributes"""          """TestLayerModification Sanity Checks"""
354          # test default settings          # test default settings
355          self.failIf(self.layer.WasModified())          self.failIf(self.layer.WasModified())
         #self.assertEquals(self.layer.fill, None)  
         #self.assertEquals(self.layer.stroke.hex(), "#000000")  
         #self.assertEquals(self.layer.stroke_width, 1)  
356          self.assertEquals(self.layer.Visible(), 1)          self.assertEquals(self.layer.Visible(), 1)
357          # no messages should have been produced          # no messages should have been produced
358          self.check_messages([])          self.check_messages([])
# Line 308  class TestLayerLegend(unittest.TestCase, Line 367  class TestLayerLegend(unittest.TestCase,
367          # the layer.          # the layer.
368          self.failIf(self.layer.WasModified())          self.failIf(self.layer.WasModified())
369    
370            def test_set_classification(self):
371            """Test Layer.SetClassification"""
372            classification = Classification()
373            classification.AppendGroup(ClassGroupRange((0.0, 0.1)))
374    
375            self.layer.SetClassification(classification)
376            self.layer.SetClassificationColumn("AREA")
377    
378            self.check_messages([(self.layer, LAYER_CHANGED),
379                                 (self.layer, LAYER_CHANGED)])
380            self.failUnless(self.layer.WasModified())
381    
382            self.clear_messages()
383            self.layer.UnsetModified()
384    
385            # change only the classification column. This should issue a
386            # LAYER_CHANGED message as well.
387            self.layer.SetClassificationColumn("PERIMETER")
388    
389            self.check_messages([(self.layer, LAYER_CHANGED)])
390            self.failUnless(self.layer.WasModified())
391    
392    
393  #  #
394  # the tree info now contains Color objects which are difficult to test  # the tree info now contains Color objects which are difficult to test
395  #  #
Legend:
Removed from v.1088  
changed lines
  Added in v.1538
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26