/[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 947 by jonathan, Tue May 20 15:27:19 2003 UTC revision 1174 by jonathan, Thu Jun 12 13:37:29 2003 UTC
# Line 26  from Thuban.Model.session import Session Line 26  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       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  from Thuban.Model.table import FIELDTYPE_DOUBLE, FIELDTYPE_STRING, MemoryTable
32  from Thuban.Model.proj import Projection  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 42  class TestLayer(unittest.TestCase, suppo Line 46  class TestLayer(unittest.TestCase, suppo
46              self.assertFloatSeqEqual(test[i], value[i])              self.assertFloatSeqEqual(test[i], value[i])
47    
48      def setUp(self):      def setUp(self):
49          """Create a session"""          """Create a session self.session and initialize self.layer to None"""
50          self.session = Session("Test session for %s" % self.__class__)          self.session = Session("Test session for %s" % self.__class__)
51            self.layer = None
52    
53      def tearDown(self):      def tearDown(self):
54            """Call the layer's Destroy method and set session and layer to None"""
55          self.session = None          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):      def build_path(self, filename):
61          return os.path.join("..", "Data", "iceland", filename)          return os.path.join("..", "Data", "iceland", filename)
62            
63      def open_shapefile(self, filename):      def open_shapefile(self, filename):
64          """Open and return a shapestore for filename in the iceland data set"""          """Open and return a shapestore for filename in the iceland data set"""
65          return self.session.OpenShapefile(self.build_path(filename))          return self.session.OpenShapefile(self.build_path(filename))
66    
67      def test_base_layer(self):      def test_base_layer(self):
68          layer = BaseLayer("Test BaseLayer")          layer = self.layer = BaseLayer("Test BaseLayer")
69          self.assertEquals(layer.Title(), "Test BaseLayer")          self.assertEquals(layer.Title(), "Test BaseLayer")
70          self.failUnless(layer.Visible())          self.failUnless(layer.Visible())
71    
# Line 83  class TestLayer(unittest.TestCase, suppo Line 92  class TestLayer(unittest.TestCase, suppo
92    
93      def test_arc_layer(self):      def test_arc_layer(self):
94          """Test Layer with arc shapes"""          """Test Layer with arc shapes"""
95          layer = Layer("Test Layer", self.open_shapefile("roads-line.shp"))          layer = self.layer = Layer("Test Layer",
96                                       self.open_shapefile("roads-line.shp"))
97          self.assertEquals(layer.ShapeType(), SHAPETYPE_ARC)          self.assertEquals(layer.ShapeType(), SHAPETYPE_ARC)
98          self.assertEquals(layer.NumShapes(), 839)          self.assertEquals(layer.NumShapes(), 839)
99          shape = layer.Shape(32)          shape = layer.Shape(32)
# Line 114  class TestLayer(unittest.TestCase, suppo Line 124  class TestLayer(unittest.TestCase, suppo
124          self.assertEquals(layer.ShapesBoundingBox([]), None)          self.assertEquals(layer.ShapesBoundingBox([]), None)
125          self.assertEquals(layer.ShapesBoundingBox(None), None)          self.assertEquals(layer.ShapesBoundingBox(None), None)
126    
         layer.Destroy()  
   
127      def test_polygon_layer(self):      def test_polygon_layer(self):
128          """Test Layer with polygon shapes"""          """Test Layer with polygon shapes"""
129          layer = Layer("Test Layer", self.open_shapefile("political.shp"))          layer = self.layer = Layer("Test Layer",
130                                       self.open_shapefile("political.shp"))
131          self.assertEquals(layer.ShapeType(), SHAPETYPE_POLYGON)          self.assertEquals(layer.ShapeType(), SHAPETYPE_POLYGON)
132          self.assertEquals(layer.NumShapes(), 156)          self.assertEquals(layer.NumShapes(), 156)
133          shape = layer.Shape(4)          shape = layer.Shape(4)
# Line 132  class TestLayer(unittest.TestCase, suppo Line 141  class TestLayer(unittest.TestCase, suppo
141                                    -13.495815277099609, 66.563774108886719])                                    -13.495815277099609, 66.563774108886719])
142          self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.9, 64.1)),          self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.9, 64.1)),
143                            [91, 92, 144, 146, 148, 150, 152, 153])                            [91, 92, 144, 146, 148, 150, 152, 153])
         layer.Destroy()  
144    
145      def test_point_layer(self):      def test_point_layer(self):
146          """Test Layer with point shapes"""          """Test Layer with point shapes"""
147          layer = Layer("Test Layer",          layer = self.layer = Layer("Test Layer",
148                        self.open_shapefile("cultural_landmark-point.shp"))                             self.open_shapefile("cultural_landmark-point.shp"))
149          self.assertEquals(layer.ShapeType(), SHAPETYPE_POINT)          self.assertEquals(layer.ShapeType(), SHAPETYPE_POINT)
150          self.assertEquals(layer.NumShapes(), 34)          self.assertEquals(layer.NumShapes(), 34)
151          shape = layer.Shape(0)          shape = layer.Shape(0)
# Line 148  class TestLayer(unittest.TestCase, suppo Line 156  class TestLayer(unittest.TestCase, suppo
156                                    -15.12291431427002, 66.36572265625])                                    -15.12291431427002, 66.36572265625])
157          self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.80, 64.1)),          self.assertEquals(layer.ShapesInRegion((-24.0, 64.0, -23.80, 64.1)),
158                            [0, 1, 2, 3, 4, 5, 27, 28, 29, 30, 31])                            [0, 1, 2, 3, 4, 5, 27, 28, 29, 30, 31])
         layer.Destroy()  
159    
160      def test_empty_layer(self):      def test_empty_layer(self):
161          """Test Layer with empty shape file"""          """Test Layer with empty shape file"""
# Line 163  class TestLayer(unittest.TestCase, suppo Line 170  class TestLayer(unittest.TestCase, suppo
170          dbf.add_field("NAME", dbflib.FTString, 20, 0)          dbf.add_field("NAME", dbflib.FTString, 20, 0)
171    
172          # Now try to open it.          # Now try to open it.
173          layer = Layer("Empty Layer",          layer = self.layer = Layer("Empty Layer",
174                        self.session.OpenShapefile(shapefilename))                                     self.session.OpenShapefile(shapefilename))
175          self.assertEquals(layer.BoundingBox(), None)          self.assertEquals(layer.BoundingBox(), None)
176          self.assertEquals(layer.LatLongBoundingBox(), None)          self.assertEquals(layer.LatLongBoundingBox(), None)
177          self.assertEquals(layer.NumShapes(), 0)          self.assertEquals(layer.NumShapes(), 0)
         layer.Destroy()  
178    
179      def test_get_field_type(self):      def test_get_field_type(self):
180          """Test Layer.GetFieldType()"""          """Test Layer.GetFieldType()"""
181          layer = Layer("Test Layer", self.open_shapefile("roads-line.shp"))          layer = self.layer = Layer("Test Layer",
182                                       self.open_shapefile("roads-line.shp"))
183          self.assertEquals(layer.GetFieldType("LENGTH"), FIELDTYPE_DOUBLE)          self.assertEquals(layer.GetFieldType("LENGTH"), FIELDTYPE_DOUBLE)
184          self.assertEquals(layer.GetFieldType("non existing"), None)          self.assertEquals(layer.GetFieldType("non existing"), None)
         layer.Destroy()  
185    
186      def test_raster_layer(self):      def test_raster_layer(self):
187            if not Thuban.Model.resource.has_gdal_support():
188                return
189    
190          filename = self.build_path("island.tif")          filename = self.build_path("island.tif")
191          layer = RasterLayer("Test RasterLayer", filename)          layer = RasterLayer("Test RasterLayer", filename)
192          self.assertEquals(layer.GetImageFilename(), filename)          self.assertEquals(layer.GetImageFilename(), filename)
# Line 188  class TestLayer(unittest.TestCase, suppo Line 197  class TestLayer(unittest.TestCase, suppo
197                                   [-24.5500000, 63.2833330,                                   [-24.5500000, 63.2833330,
198                                    -13.4916670, 66.5666670])                                    -13.4916670, 66.5666670])
199    
200        def test_derived_store(self):
201            """Test layer with derived store"""
202            layer = self.layer = Layer("Test Layer",
203                                       self.open_shapefile("roads-line.shp"))
204            try:
205                store = layer.ShapeStore()
206                derived = DerivedShapeStore(store, store.Table())
207                layer.SetShapeStore(derived)
208                self.assert_(layer.ShapeStore() is derived)
209            finally:
210                store = derived = None
211    
212    
213    class SetShapeStoreTests(unittest.TestCase, support.SubscriberMixin):
214    
215        def setUp(self):
216            """Create a layer with a classification as self.layer"""
217            self.clear_messages()
218            self.session = Session("Test session for %s" % self.__class__)
219            self.shapefilename = os.path.join("..", "Data", "iceland",
220                                              "cultural_landmark-point.dbf")
221            self.store = self.session.OpenShapefile(self.shapefilename)
222            self.layer = Layer("test layer", self.store)
223            self.classification = Classification(field = "CLPTLABEL")
224            self.classification.AppendGroup(ClassGroupSingleton("FARM"))
225            self.layer.SetClassification(self.classification)
226            self.layer.UnsetModified()
227            self.layer.Subscribe(LAYER_SHAPESTORE_REPLACED,
228                                 self.subscribe_with_params,
229                                 LAYER_SHAPESTORE_REPLACED)
230            self.layer.Subscribe(LAYER_CHANGED,
231                                 self.subscribe_with_params, LAYER_CHANGED)
232    
233        def tearDown(self):
234            self.clear_messages()
235            self.layer.Destroy()
236            self.session.Destroy()
237            self.session = self.layer = self.store = self.classification = None
238    
239        def test_sanity(self):
240            """SetShapeStoreTests sanity check
241    
242            Test the initial state of the test case instances after setUp.
243            """
244            cls = self.layer.GetClassification()
245            self.assert_(cls is self.classification)
246            self.assertEquals(cls.GetField(), "CLPTLABEL")
247            self.assertEquals(cls.GetFieldType(), FIELDTYPE_STRING)
248            self.assertEquals(self.layer.GetClassification().GetNumGroups(), 1)
249            self.failIf(self.layer.WasModified())
250    
251        def test_set_shape_store_modified_flag(self):
252            """Test whether Layer.SetShapeStore() sets the modified flag"""
253            memtable = MemoryTable([("FOO", FIELDTYPE_STRING)],
254                          [("bla",)] * self.layer.ShapeStore().Table().NumRows())
255            self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
256    
257            self.assert_(self.layer.WasModified())
258    
259        def test_set_shape_store_different_field_name(self):
260            """Test Layer.SetShapeStore() with different column name"""
261            memtable = MemoryTable([("FOO", FIELDTYPE_STRING)],
262                          [("bla",)] * self.layer.ShapeStore().Table().NumRows())
263            self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
264            # The classification should contain only the default group now.
265            self.assertEquals(self.layer.GetClassification().GetNumGroups(), 0)
266            self.check_messages([(self.layer, LAYER_CHANGED),
267                                 (self.layer, LAYER_SHAPESTORE_REPLACED)])
268    
269        def test_set_shape_store_same_field(self):
270            """Test Layer.SetShapeStore() with same column name and type"""
271            memtable = MemoryTable([("CLPTLABEL", FIELDTYPE_STRING)],
272                          [("bla",)] * self.layer.ShapeStore().Table().NumRows())
273            self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
274            # The classification should be the same as before
275            self.assert_(self.layer.GetClassification() is self.classification)
276            self.check_messages([(self.layer, LAYER_SHAPESTORE_REPLACED)])
277    
278        def test_set_shape_store_same_field_different_type(self):
279            """Test Layer.SetShapeStore() with same column name but different type
280            """
281            memtable = MemoryTable([("CLPTLABEL", FIELDTYPE_DOUBLE)],
282                          [(0.0,)] * self.layer.ShapeStore().Table().NumRows())
283            self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
284            # The classification should contain only the default group now.
285            self.assertEquals(self.layer.GetClassification().GetNumGroups(), 0)
286            self.check_messages([(self.layer, LAYER_CHANGED),
287                                 (self.layer, LAYER_SHAPESTORE_REPLACED)])
288    
289    
290  class TestLayerLegend(unittest.TestCase, support.SubscriberMixin):  class TestLayerLegend(unittest.TestCase, support.SubscriberMixin):
291    
Legend:
Removed from v.947  
changed lines
  Added in v.1174
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26