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

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

Parent Directory | Revision Log | View Patch Patch

-revision 996 by bh,
Thu May 22 18:21:10 2003 UTC
+revision 1174 by jonathan,
Thu Jun 12 13:37:29 2003 UTC
 Line 26 
 from Thuban.Model.session import Session
  from Thuban.Model.layer import BaseLayer, Layer, RasterLayer, \
       SHAPETYPE_POLYGON, SHAPETYPE_ARC, SHAPETYPE_POINT
  from Thuban.Model.messages import LAYER_LEGEND_CHANGED, \
-      LAYER_VISIBILITY_CHANGED
+      LAYER_VISIBILITY_CHANGED, LAYER_SHAPESTORE_REPLACED, LAYER_CHANGED
  from Thuban.Model.color import Color
- from Thuban.Model.table import FIELDTYPE_DOUBLE
+ from Thuban.Model.table import FIELDTYPE_DOUBLE, FIELDTYPE_STRING, MemoryTable
  from Thuban.Model.proj import Projection
  from Thuban.Model.data import DerivedShapeStore
+ from Thuban.Model.classification import Classification, ClassGroupSingleton
+ import Thuban.Model.resource
  class TestLayer(unittest.TestCase, support.FileTestMixin,
                  support.FloatComparisonMixin):
-Line 181 
 class TestLayer(unittest.TestCase, suppo
+Line 184 
 class TestLayer(unittest.TestCase, suppo
          self.assertEquals(layer.GetFieldType("non existing"), None)
      def test_raster_layer(self):
+         if not Thuban.Model.resource.has_gdal_support():
+             return
          filename = self.build_path("island.tif")
          layer = RasterLayer("Test RasterLayer", filename)
          self.assertEquals(layer.GetImageFilename(), filename)
-Line 204 
 class TestLayer(unittest.TestCase, suppo
+Line 210 
 class TestLayer(unittest.TestCase, suppo
              store = derived = None
+ class SetShapeStoreTests(unittest.TestCase, support.SubscriberMixin):
+     def setUp(self):
+         """Create a layer with a classification as self.layer"""
+         self.clear_messages()
+         self.session = Session("Test session for %s" % self.__class__)
+         self.shapefilename = os.path.join("..", "Data", "iceland",
+                                           "cultural_landmark-point.dbf")
+         self.store = self.session.OpenShapefile(self.shapefilename)
+         self.layer = Layer("test layer", self.store)
+         self.classification = Classification(field = "CLPTLABEL")
+         self.classification.AppendGroup(ClassGroupSingleton("FARM"))
+         self.layer.SetClassification(self.classification)
+         self.layer.UnsetModified()
+         self.layer.Subscribe(LAYER_SHAPESTORE_REPLACED,
+                              self.subscribe_with_params,
+                              LAYER_SHAPESTORE_REPLACED)
+         self.layer.Subscribe(LAYER_CHANGED,
+                              self.subscribe_with_params, LAYER_CHANGED)
+     def tearDown(self):
+         self.clear_messages()
+         self.layer.Destroy()
+         self.session.Destroy()
+         self.session = self.layer = self.store = self.classification = None
+     def test_sanity(self):
+         """SetShapeStoreTests sanity check
+         Test the initial state of the test case instances after setUp.
+         """
+         cls = self.layer.GetClassification()
+         self.assert_(cls is self.classification)
+         self.assertEquals(cls.GetField(), "CLPTLABEL")
+         self.assertEquals(cls.GetFieldType(), FIELDTYPE_STRING)
+         self.assertEquals(self.layer.GetClassification().GetNumGroups(), 1)
+         self.failIf(self.layer.WasModified())
+     def test_set_shape_store_modified_flag(self):
+         """Test whether Layer.SetShapeStore() sets the modified flag"""
+         memtable = MemoryTable([("FOO", FIELDTYPE_STRING)],
+                       [("bla",)] * self.layer.ShapeStore().Table().NumRows())
+         self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
+         self.assert_(self.layer.WasModified())
+     def test_set_shape_store_different_field_name(self):
+         """Test Layer.SetShapeStore() with different column name"""
+         memtable = MemoryTable([("FOO", FIELDTYPE_STRING)],
+                       [("bla",)] * self.layer.ShapeStore().Table().NumRows())
+         self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
+         # The classification should contain only the default group now.
+         self.assertEquals(self.layer.GetClassification().GetNumGroups(), 0)
+         self.check_messages([(self.layer, LAYER_CHANGED),
+                              (self.layer, LAYER_SHAPESTORE_REPLACED)])
+     def test_set_shape_store_same_field(self):
+         """Test Layer.SetShapeStore() with same column name and type"""
+         memtable = MemoryTable([("CLPTLABEL", FIELDTYPE_STRING)],
+                       [("bla",)] * self.layer.ShapeStore().Table().NumRows())
+         self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
+         # The classification should be the same as before
+         self.assert_(self.layer.GetClassification() is self.classification)
+         self.check_messages([(self.layer, LAYER_SHAPESTORE_REPLACED)])
+     def test_set_shape_store_same_field_different_type(self):
+         """Test Layer.SetShapeStore() with same column name but different type
+         """
+         memtable = MemoryTable([("CLPTLABEL", FIELDTYPE_DOUBLE)],
+                       [(0.0,)] * self.layer.ShapeStore().Table().NumRows())
+         self.layer.SetShapeStore(DerivedShapeStore(self.store, memtable))
+         # The classification should contain only the default group now.
+         self.assertEquals(self.layer.GetClassification().GetNumGroups(), 0)
+         self.check_messages([(self.layer, LAYER_CHANGED),
+                              (self.layer, LAYER_SHAPESTORE_REPLACED)])
  class TestLayerLegend(unittest.TestCase, support.SubscriberMixin):
      """Test cases for Layer method that modify the layer.

 Legend:



Removed from v.996
 


changed lines


 
Added in v.1174
 Legend:



Removed from v.996
 


changed lines


 
Added in v.1174
-Removed from v.996
+Added in v.1174

[email protected]	ViewVC Help
Powered by ViewVC 1.1.26