/[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 1174 by jonathan, Thu Jun 12 13:37:29 2003 UTC revision 1452 by bh, Fri Jul 18 12:57:59 2003 UTC
# Line 27  from Thuban.Model.layer import BaseLayer Line 27  from Thuban.Model.layer import BaseLayer
27       SHAPETYPE_POLYGON, SHAPETYPE_ARC, SHAPETYPE_POINT       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_SHAPESTORE_REPLACED, LAYER_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  import Thuban.Model.resource
37    
# Line 77  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 94  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 128  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 146  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 189  class TestLayer(unittest.TestCase, suppo Line 197  class TestLayer(unittest.TestCase, suppo
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 206  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                # 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:          finally:
229              store = derived = None              store = derived = None
230                layer2.Destroy()
231                store2 = None
232    
233    
234  class SetShapeStoreTests(unittest.TestCase, support.SubscriberMixin):  class SetShapeStoreTests(unittest.TestCase, support.SubscriberMixin):
# Line 220  class SetShapeStoreTests(unittest.TestCa Line 241  class SetShapeStoreTests(unittest.TestCa
241                                            "cultural_landmark-point.dbf")                                            "cultural_landmark-point.dbf")
242          self.store = self.session.OpenShapefile(self.shapefilename)          self.store = self.session.OpenShapefile(self.shapefilename)
243          self.layer = Layer("test layer", self.store)          self.layer = Layer("test layer", self.store)
244          self.classification = Classification(field = "CLPTLABEL")          self.classification = Classification()
245          self.classification.AppendGroup(ClassGroupSingleton("FARM"))          self.classification.AppendGroup(ClassGroupSingleton("FARM"))
246            self.layer.SetClassificationColumn("CLPTLABEL")
247          self.layer.SetClassification(self.classification)          self.layer.SetClassification(self.classification)
248          self.layer.UnsetModified()          self.layer.UnsetModified()
249          self.layer.Subscribe(LAYER_SHAPESTORE_REPLACED,          self.layer.Subscribe(LAYER_SHAPESTORE_REPLACED,
# Line 243  class SetShapeStoreTests(unittest.TestCa Line 265  class SetShapeStoreTests(unittest.TestCa
265          """          """
266          cls = self.layer.GetClassification()          cls = self.layer.GetClassification()
267          self.assert_(cls is self.classification)          self.assert_(cls is self.classification)
268          self.assertEquals(cls.GetField(), "CLPTLABEL")          field = self.layer.GetClassificationColumn()
269          self.assertEquals(cls.GetFieldType(), FIELDTYPE_STRING)          self.assertEquals(field, "CLPTLABEL")
270            self.assertEquals(self.layer.GetFieldType(field), FIELDTYPE_STRING)
271          self.assertEquals(self.layer.GetClassification().GetNumGroups(), 1)          self.assertEquals(self.layer.GetClassification().GetNumGroups(), 1)
272          self.failIf(self.layer.WasModified())          self.failIf(self.layer.WasModified())
273    
# Line 287  class SetShapeStoreTests(unittest.TestCa Line 310  class SetShapeStoreTests(unittest.TestCa
310                               (self.layer, LAYER_SHAPESTORE_REPLACED)])                               (self.layer, LAYER_SHAPESTORE_REPLACED)])
311    
312    
313  class TestLayerLegend(unittest.TestCase, support.SubscriberMixin):  class TestLayerModification(unittest.TestCase, support.SubscriberMixin):
314    
315      """Test cases for Layer method that modify the layer.      """Test cases for Layer method that modify the layer.
316      """      """
# Line 307  class TestLayerLegend(unittest.TestCase, Line 330  class TestLayerLegend(unittest.TestCase,
330          self.layer.Subscribe(LAYER_VISIBILITY_CHANGED,          self.layer.Subscribe(LAYER_VISIBILITY_CHANGED,
331                               self.subscribe_with_params,                               self.subscribe_with_params,
332                               LAYER_VISIBILITY_CHANGED)                               LAYER_VISIBILITY_CHANGED)
333            self.layer.Subscribe(LAYER_CHANGED, self.subscribe_with_params,
334                                 LAYER_CHANGED)
335    
336      def tearDown(self):      def tearDown(self):
337          """Clear the list of received messages and explictly destroy self.layer          """Clear the list of received messages and explictly destroy self.layer
# Line 317  class TestLayerLegend(unittest.TestCase, Line 342  class TestLayerLegend(unittest.TestCase,
342          self.session = None          self.session = None
343          self.clear_messages()          self.clear_messages()
344    
345      def test_initial_settings(self):      def test_sanity(self):
346          """Test Layer's initial legend attributes"""          """TestLayerModification Sanity Checks"""
347          # test default settings          # test default settings
348          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)  
349          self.assertEquals(self.layer.Visible(), 1)          self.assertEquals(self.layer.Visible(), 1)
350          # no messages should have been produced          # no messages should have been produced
351          self.check_messages([])          self.check_messages([])
# Line 338  class TestLayerLegend(unittest.TestCase, Line 360  class TestLayerLegend(unittest.TestCase,
360          # the layer.          # the layer.
361          self.failIf(self.layer.WasModified())          self.failIf(self.layer.WasModified())
362    
363            def test_set_classification(self):
364            """Test Layer.SetClassification"""
365            classification = Classification()
366            classification.AppendGroup(ClassGroupRange((0.0, 0.1)))
367    
368            self.layer.SetClassification(classification)
369            self.layer.SetClassificationColumn("AREA")
370    
371            self.check_messages([(self.layer, LAYER_CHANGED),
372                                 (self.layer, LAYER_CHANGED)])
373            self.failUnless(self.layer.WasModified())
374    
375            self.clear_messages()
376            self.layer.UnsetModified()
377    
378            # change only the classification column. This should issue a
379            # LAYER_CHANGED message as well.
380            self.layer.SetClassificationColumn("PERIMETER")
381    
382            self.check_messages([(self.layer, LAYER_CHANGED)])
383            self.failUnless(self.layer.WasModified())
384    
385    
386  #  #
387  # the tree info now contains Color objects which are difficult to test  # the tree info now contains Color objects which are difficult to test
388  #  #
Legend:
Removed from v.1174  
changed lines
  Added in v.1452
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26