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 |
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""" |
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) |
291 |
|
|
292 |
c.SetField(None) |
c.SetField(None) |
293 |
self.assertEquals(c.GetFieldType(), None) |
self.assertEquals(c.GetFieldType(), None) |
294 |
self.assertEquals(c.GetGroup(5), c.GetDefaultGroup()) |
self.assertEquals(c.FindGroup(5), c.GetDefaultGroup()) |
295 |
|
|
296 |
c.SetField("AREA") |
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() |
layer.Destroy() |
311 |
|
|