/[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 1088 by bh,
Wed May 28 12:42:23 2003 UTC
 Line 28 
 from Thuban.Model.layer import BaseLayer
  from Thuban.Model.messages import LAYER_LEGEND_CHANGED, \
       LAYER_VISIBILITY_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
  class TestLayer(unittest.TestCase, support.FileTestMixin,
                  support.FloatComparisonMixin):
-Line 204 
 class TestLayer(unittest.TestCase, suppo
+Line 205 
 class TestLayer(unittest.TestCase, suppo
              store = derived = None
+ class SetShapeStoreTests(unittest.TestCase):
+     def setUp(self):
+         """Create a layer with a classification as self.layer"""
+         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)
+     def tearDown(self):
+         self.layer.Destroy()
+         self.session.Destroy()
+         self.session = self.layer = self.store = self.classification = None
+     def test_sanity(self):
+         """SetShapeStoreTests sanity check"""
+         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)
+     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)
+     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)
+     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)
  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.1088
 Legend:



Removed from v.996
 


changed lines


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

[email protected]	ViewVC Help
Powered by ViewVC 1.1.26