/[thuban]/branches/WIP-pyshapelib-bramz/test/test_classification.py
ViewVC logotype

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

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 482 by jonathan, Thu Mar 6 16:47:02 2003 UTC revision 618 by jonathan, Mon Apr 7 08:58:26 2003 UTC
# Line 1  Line 1 
1  # Copyright (c) 2002 by Intevation GmbH  # Copyright (c) 2002, 2003 by Intevation GmbH
2  # Authors:  # Authors:
3  # Jonathan Coles <[email protected]>  # Jonathan Coles <[email protected]>
4  #  #
# Line 35  class TestClassification(unittest.TestCa Line 35  class TestClassification(unittest.TestCa
35          props = ClassGroupProperties()          props = ClassGroupProperties()
36          self.assertEqual(props.GetLineColor(), Color.Black)          self.assertEqual(props.GetLineColor(), Color.Black)
37          self.assertEqual(props.GetLineWidth(), 1)          self.assertEqual(props.GetLineWidth(), 1)
38          self.assertEqual(props.GetFill(), Color.None)          self.assertEqual(props.GetFill(), Color.Transparent)
39    
40          red = Color(1, 0, 0)          red = Color(1, 0, 0)
41          props.SetLineColor(red)          props.SetLineColor(red)
# Line 76  class TestClassification(unittest.TestCa Line 76  class TestClassification(unittest.TestCa
76          self.assertEqual(group.GetLabel(), "")          self.assertEqual(group.GetLabel(), "")
77    
78          # test Matches          # test Matches
79          self.assertEqual(group.Matches(None), False)          # Matches() is a virtual function...can't test it here
80          self.assertEqual(group.Matches(1), False)          #
81          self.assertEqual(group.Matches("hallo"), False)          #self.assertEqual(group.Matches(None), False)
82          self.assertEqual(group.Matches([]), False)          #self.assertEqual(group.Matches(1), False)
83            #self.assertEqual(group.Matches("hallo"), False)
84            #self.assertEqual(group.Matches([]), False)
85    
86          # test GetProperties          # test GetProperties...also a virtual function
87          self.assertEqual(group.GetProperties(), None)          #self.assertEqual(group.GetProperties(), None)
88    
89      def test_ClassGroupDefault(self):      def test_ClassGroupDefault(self):
90          """Test ClassGroupDefault"""          """Test ClassGroupDefault"""
# Line 232  class TestClassification(unittest.TestCa Line 234  class TestClassification(unittest.TestCa
234          clazz = Classification()          clazz = Classification()
235    
236          for g in groups:          for g in groups:
237              clazz.AddGroup(g)              clazz.AppendGroup(g)
238    
239          def convert(clazz):          def convert(clazz):
240              if isinstance(clazz, ClassGroupDefault):   return 0              if isinstance(clazz, ClassGroupDefault):   return 0
# Line 243  class TestClassification(unittest.TestCa Line 245  class TestClassification(unittest.TestCa
245          for g in clazz:          for g in clazz:
246              list.append(convert(g))              list.append(convert(g))
247    
248          self.assertEquals(list, [0, 1, 1, 2, 1])          self.assertEquals(list, [0, 1, 1, 2, 1, 0])
249    
250      def test_classification(self):      def test_classification(self):
251          """Test Classification"""          """Test Classification"""
# Line 253  class TestClassification(unittest.TestCa Line 255  class TestClassification(unittest.TestCa
255          green = Color(0, 1, 0)          green = Color(0, 1, 0)
256          blue  = Color(0, 0, 1)          blue  = Color(0, 0, 1)
257    
258          layer = Layer("asdf", "../Data/iceland/political.shp")          layer = Layer("asdf", "../Data/iceland/political.dbf")
259    
260          #          #
261          # init with no params          # init with no params
# Line 261  class TestClassification(unittest.TestCa Line 263  class TestClassification(unittest.TestCa
263          c = Classification()          c = Classification()
264          self.assertEqual(c.GetField(), None)          self.assertEqual(c.GetField(), None)
265          self.assertEqual(c.GetFieldType(), None)          self.assertEqual(c.GetFieldType(), None)
266          self.assertEqual(c.GetGroup(-1), c.GetDefaultGroup())          self.assertEqual(c.FindGroup(-1), c.GetDefaultGroup())
267    
268          c.SetDefaultLineColor(red)          c.SetDefaultLineColor(red)
269          self.assertEqual(c.GetDefaultLineColor(), red)          self.assertEqual(c.GetDefaultLineColor(), red)
270          self.assertEqual(c.GetDefaultFill(), Color.None)          self.assertEqual(c.GetDefaultFill(), Color.Transparent)
271    
272          c.SetDefaultFill(green)          c.SetDefaultFill(green)
273          self.assertEqual(c.GetDefaultFill(), green)          self.assertEqual(c.GetDefaultFill(), green)
# Line 277  class TestClassification(unittest.TestCa Line 279  class TestClassification(unittest.TestCa
279          c.SetFieldType(FIELDTYPE_STRING)          c.SetFieldType(FIELDTYPE_STRING)
280          self.assertEqual(c.GetFieldType(), FIELDTYPE_STRING)          self.assertEqual(c.GetFieldType(), FIELDTYPE_STRING)
281    
282            # should raise an exception because 'hallo' doesn't
283            # exist in the table
284            self.assertRaises(ValueError, c.SetLayer, layer)
285            
286            c.SetField("AREA")
287          c.SetLayer(layer)          c.SetLayer(layer)
288          self.assertEqual(c.GetLayer(), layer)          self.assertEqual(c.GetLayer(), layer)
289            self.assertEqual(c.GetField(), "AREA")
290            self.assertEqual(c.GetFieldType(), FIELDTYPE_DOUBLE)
291    
292          c.SetField(None)          c.SetField(None)
293          self.assertEquals(c.GetGroup(5), c.GetDefaultGroup())          self.assertEquals(c.GetFieldType(), None)
294            self.assertEquals(c.FindGroup(5), c.GetDefaultGroup())
295    
296          c.SetField("hallo")          c.SetField("AREA")
297          s = ClassGroupSingleton(5)          s = ClassGroupSingleton(5)
298          c.AddGroup(s)          c.AppendGroup(s)
299          self.assertEquals(c.GetGroup(5), s)          self.assertEquals(c.FindGroup(5), s)
300          self.assertEquals(c.GetGroup(0), c.GetDefaultGroup())          self.assertEquals(c.FindGroup(0), c.GetDefaultGroup())
301    
302          r = ClassGroupRange(-10, 10)          r = ClassGroupRange(-10, 10)
303          c.AddGroup(r)          c.AppendGroup(r)
304          self.assertEquals(c.GetGroup(-11), c.GetDefaultGroup())          self.assertEquals(c.FindGroup(-11), c.GetDefaultGroup())
305          self.assertEquals(c.GetGroup(-10), r)          self.assertEquals(c.FindGroup(-10), r)
306          self.assertEquals(c.GetGroup(9), r)          self.assertEquals(c.FindGroup(9), r)
307          self.assertEquals(c.GetGroup(5), s)          self.assertEquals(c.FindGroup(5), s)
308          self.assertEquals(c.GetGroup(10), c.GetDefaultGroup())          self.assertEquals(c.FindGroup(10), c.GetDefaultGroup())
309    
310            layer.Destroy()
311    
312  if __name__ == "__main__":  if __name__ == "__main__":
313      unittest.main()      support.run_tests()
Legend:
Removed from v.482  
changed lines
  Added in v.618
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26