/[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 369 by jonathan, Mon Jan 27 11:53:51 2003 UTC revision 395 by jonathan, Mon Feb 10 15:28:02 2003 UTC
# Line 19  import support Line 19  import support
19  support.initthuban()  support.initthuban()
20    
21  from Thuban.Model.classification import Classification  from Thuban.Model.classification import Classification
22    from Thuban.Model.layer import Layer
23    
24    
25  class TestClassification(unittest.TestCase):  class TestClassification(unittest.TestCase):
# Line 26  class TestClassification(unittest.TestCa Line 27  class TestClassification(unittest.TestCa
27      def test_classification(self):      def test_classification(self):
28          """Test Classification"""          """Test Classification"""
29    
30            layer = Layer("asdf", "../Data/iceland/political.shp")
31    
32          #          #
33          # init with no params          # init with no params
34          #          #
35          c = Classification()          c = Classification(layer)
36          self.assertEquals(c.field, None)          self.assertEqual(c.field, None)
37          self.assertEquals(c.NullData, None)          self.assertNotEqual(c.DefaultData, None)
38    
39          #          #
40          # setField          # SetField
41          #          #
42          c = Classification()          c = Classification(layer)
43          c.setField("Test")          c.SetField("Test")
44          self.assertEquals(c.field, "Test")          self.assertEqual(c.field, "Test")
45          c.setField(None)          c.SetField(None)
46          self.assertEquals(c.field, None)          self.assertEqual(c.field, None)
47    
48          #          #
49          # init with field          # init with field
50          #          #
51          c = Classification("Test")          c = Classification(layer, "Test")
52          self.assertEquals(c.field, "Test")          self.assertEqual(c.field, "Test")
         self.assertEquals(c.NullData, None)  
53    
54          #          #
55          # addRange          # AddRange
56          #          #
57          c0 = Classification("c0")          c0 = Classification(layer, "c0")
58          self.assertRaises(ValueError, c0.addRange, 0, 0, None)          default0 = c0.GetDefaultData()
59          self.assertRaises(ValueError, c0.addRange, 10, 0, None)          self.assertRaises(ValueError, c0.AddRange, 0, 0, None)
60          c0.addRange(-10, 0, "1")          self.assertRaises(ValueError, c0.AddRange, 10, 0, None)
61          c0.addRange(0, 10, "2")          c0.AddRange(-10, 0, "1")
62            c0.AddRange(0, 10, "2")
63    
64          #          #
65          # addPoint          # AddPoint
66          #          #
67          c1 = Classification("c1")          c1 = Classification(layer, "c1")
68          c1.addPoint(0, "1")          default1 = c1.GetDefaultData()
69          c1.addPoint("0", "2")          c1.AddPoint(0, "1")
70          self.assertRaises(TypeError, c1.addPoint, {'monty':'python'}, "3")          c1.AddPoint("0", "2")
71            self.assertRaises(TypeError, c1.AddPoint, {'monty':'python'}, "3")
72    
73          #          #
74          # getProperties          # GetProperties
75          #          #
76    
77          self.assertEqual(c0.getProperties(-10), "1")          self.assertEqual(c0.GetProperties(-10), "1")
78          self.assertEqual(c0.getProperties(-11), None)          self.assertEqual(c0.GetProperties(-11), default0)
79          self.assertEqual(c0.getProperties(0), "2") # min <= x < max          self.assertEqual(c0.GetProperties(0), "2") # min <= x < max
80          self.assertEqual(c0.getProperties(10), None)          self.assertEqual(c0.GetProperties(10), default0)
81          self.assertEqual(c0.getProperties(11), None)          self.assertEqual(c0.GetProperties(11), default0)
82    
83          self.assertEqual(c1.getProperties(0), "1")          self.assertEqual(c1.GetProperties(0), "1")
84          self.assertEqual(c1.getProperties("0"), "2")          self.assertEqual(c1.GetProperties("0"), "2")
85          self.assertEqual(c1.getProperties(-1), None)          self.assertEqual(c1.GetProperties(-1), default1)
86          self.assertEqual(c1.getProperties(1), None)          self.assertEqual(c1.GetProperties(1), default1)
87          self.assertRaises(TypeError, c1.getProperties, {'monty':'python'})          self.assertRaises(TypeError, c1.GetProperties, {'monty':'python'})
88    
89          #          #
90          # toggle field          # toggle field
91          #          #
92    
93          c0.setField(None)          c0.SetField(None)
94          self.assertEqual(c0.getProperties(10), None)          self.assertEqual(c0.GetProperties(10), default0)
95          self.assertEqual(c0.getProperties(11), None)          self.assertEqual(c0.GetProperties(11), default0)
96    
97          c1.setField(None)          c1.SetField(None)
98          self.assertEqual(c1.getProperties(0), None)          self.assertEqual(c1.GetProperties(0), default1)
99          self.assertEqual(c1.getProperties("0"), None)          self.assertEqual(c1.GetProperties("0"), default1)
100    
101          c0.setField("c0")          c0.SetField("c0")
102          self.assertEqual(c0.getProperties(10), None)          self.assertEqual(c0.GetProperties(10), default0)
103          self.assertEqual(c0.getProperties(11), None)          self.assertEqual(c0.GetProperties(11), default0)
104    
105          c1.setField("c1")          c1.SetField("c1")
106          self.assertEqual(c1.getProperties(0), "1")          self.assertEqual(c1.GetProperties(0), "1")
107          self.assertEqual(c1.getProperties("0"), "2")          self.assertEqual(c1.GetProperties("0"), "2")
108    
109  if __name__ == "__main__":  if __name__ == "__main__":
110      unittest.main()      unittest.main()
Legend:
Removed from v.369  
changed lines
  Added in v.395
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26