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): |
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.GetField(), None) |
37 |
self.assertEquals(c.NullData, None) |
self.assertNotEqual(c.GetDefaultGroup(), 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 |
# GetProperties |
56 |
# |
# |
|
c0 = Classification("c0") |
|
|
self.assertRaises(ValueError, c0.addRange, 0, 0, None) |
|
|
self.assertRaises(ValueError, c0.addRange, 10, 0, None) |
|
|
c0.addRange(-10, 0, "1") |
|
|
c0.addRange(0, 10, "2") |
|
57 |
|
|
58 |
# |
# self.assertEqual(c0.GetProperties(-10), "1") |
59 |
# addPoint |
# self.assertEqual(c0.GetProperties(-11), default0) |
60 |
# |
# self.assertEqual(c0.GetProperties(0), "2") # min <= x < max |
61 |
c1 = Classification("c1") |
# self.assertEqual(c0.GetProperties(10), default0) |
62 |
c1.addPoint(0, "1") |
# self.assertEqual(c0.GetProperties(11), default0) |
|
c1.addPoint("0", "2") |
|
|
self.assertRaises(TypeError, c1.addPoint, {'monty':'python'}, "3") |
|
63 |
|
|
64 |
# |
# self.assertEqual(c1.GetProperties(0), "1") |
65 |
# getProperties |
# self.assertEqual(c1.GetProperties("0"), "2") |
66 |
# |
# self.assertEqual(c1.GetProperties(-1), default1) |
67 |
|
# self.assertEqual(c1.GetProperties(1), default1) |
68 |
|
# self.assertRaises(TypeError, c1.GetProperties, {'monty':'python'}) |
69 |
|
|
70 |
self.assertEqual(c0.getProperties(-10), "1") |
# # |
71 |
self.assertEqual(c0.getProperties(-11), None) |
# # toggle field |
72 |
self.assertEqual(c0.getProperties(0), "2") # min <= x < max |
# # |
|
self.assertEqual(c0.getProperties(10), None) |
|
|
self.assertEqual(c0.getProperties(11), None) |
|
|
|
|
|
self.assertEqual(c1.getProperties(0), "1") |
|
|
self.assertEqual(c1.getProperties("0"), "2") |
|
|
self.assertEqual(c1.getProperties(-1), None) |
|
|
self.assertEqual(c1.getProperties(1), None) |
|
|
self.assertRaises(TypeError, c1.getProperties, {'monty':'python'}) |
|
|
|
|
|
# |
|
|
# toggle field |
|
|
# |
|
73 |
|
|
74 |
c0.setField(None) |
# c0.SetField(None) |
75 |
self.assertEqual(c0.getProperties(10), None) |
# self.assertEqual(c0.GetProperties(10), default0) |
76 |
self.assertEqual(c0.getProperties(11), None) |
# self.assertEqual(c0.GetProperties(11), default0) |
77 |
|
|
78 |
c1.setField(None) |
# c1.SetField(None) |
79 |
self.assertEqual(c1.getProperties(0), None) |
# self.assertEqual(c1.GetProperties(0), default1) |
80 |
self.assertEqual(c1.getProperties("0"), None) |
# self.assertEqual(c1.GetProperties("0"), default1) |
81 |
|
|
82 |
c0.setField("c0") |
# c0.SetField("c0") |
83 |
self.assertEqual(c0.getProperties(10), None) |
# self.assertEqual(c0.GetProperties(10), default0) |
84 |
self.assertEqual(c0.getProperties(11), None) |
# self.assertEqual(c0.GetProperties(11), default0) |
85 |
|
|
86 |
c1.setField("c1") |
# c1.SetField("c1") |
87 |
self.assertEqual(c1.getProperties(0), "1") |
# self.assertEqual(c1.GetProperties(0), "1") |
88 |
self.assertEqual(c1.getProperties("0"), "2") |
# self.assertEqual(c1.GetProperties("0"), "2") |
89 |
|
|
90 |
if __name__ == "__main__": |
if __name__ == "__main__": |
91 |
unittest.main() |
unittest.main() |