21 |
support.initthuban() |
support.initthuban() |
22 |
|
|
23 |
import os |
import os |
24 |
from Thuban.Model.color import Color |
from Thuban.Model.color import Color, Transparent, Black |
25 |
from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_STRING, FIELDTYPE_DOUBLE |
from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_STRING, FIELDTYPE_DOUBLE |
26 |
from Thuban.Model.classification import \ |
from Thuban.Model.classification import \ |
27 |
Classification, ClassGroup, \ |
Classification, ClassGroup, \ |
40 |
"""Test ClassGroupProperties""" |
"""Test ClassGroupProperties""" |
41 |
|
|
42 |
props = ClassGroupProperties() |
props = ClassGroupProperties() |
43 |
self.assertEqual(props.GetLineColor(), Color.Black) |
self.assertEqual(props.GetLineColor(), Black) |
44 |
self.assertEqual(props.GetLineWidth(), 1) |
self.assertEqual(props.GetLineWidth(), 1) |
45 |
self.assertEqual(props.GetFill(), Color.Transparent) |
self.assertEqual(props.GetFill(), Transparent) |
46 |
|
|
47 |
red = Color(1, 0, 0) |
red = Color(1, 0, 0) |
48 |
props.SetLineColor(red) |
props.SetLineColor(red) |
276 |
|
|
277 |
c.SetDefaultLineColor(red) |
c.SetDefaultLineColor(red) |
278 |
self.assertEqual(c.GetDefaultLineColor(), red) |
self.assertEqual(c.GetDefaultLineColor(), red) |
279 |
self.assertEqual(c.GetDefaultFill(), Color.Transparent) |
self.assertEqual(c.GetDefaultFill(), Transparent) |
280 |
|
|
281 |
c.SetDefaultFill(green) |
c.SetDefaultFill(green) |
282 |
self.assertEqual(c.GetDefaultFill(), green) |
self.assertEqual(c.GetDefaultFill(), green) |
283 |
self.assertEqual(c.GetDefaultLineColor(), red) |
self.assertEqual(c.GetDefaultLineColor(), red) |
284 |
|
|
285 |
c.SetField("hallo") |
c.SetFieldInfo("hallo", FIELDTYPE_STRING) |
286 |
self.assertEqual(c.GetField(), "hallo") |
self.assertEqual(c.GetField(), "hallo") |
|
|
|
|
c.SetFieldType(FIELDTYPE_STRING) |
|
287 |
self.assertEqual(c.GetFieldType(), FIELDTYPE_STRING) |
self.assertEqual(c.GetFieldType(), FIELDTYPE_STRING) |
288 |
|
|
289 |
# should raise an exception because 'hallo' doesn't |
# should raise an exception because 'hallo' doesn't |
290 |
# exist in the table |
# exist in the table |
291 |
self.assertRaises(ValueError, c.SetLayer, layer) |
self.assertRaises(ValueError, c._set_layer, layer) |
292 |
|
|
293 |
c.SetField("AREA") |
c.SetFieldInfo("AREA", None) |
294 |
c.SetLayer(layer) |
layer.SetClassification(c) |
295 |
self.assertEqual(c.GetLayer(), layer) |
self.assertEqual(c.GetLayer(), layer) |
296 |
self.assertEqual(c.GetField(), "AREA") |
self.assertEqual(c.GetField(), "AREA") |
297 |
self.assertEqual(c.GetFieldType(), FIELDTYPE_DOUBLE) |
self.assertEqual(c.GetFieldType(), FIELDTYPE_DOUBLE) |
298 |
|
|
299 |
c.SetField(None) |
c.SetFieldInfo(None, None) |
300 |
self.assertEquals(c.GetFieldType(), None) |
self.assertEquals(c.GetFieldType(), None) |
301 |
self.assertEquals(c.FindGroup(5), c.GetDefaultGroup()) |
self.assertEquals(c.FindGroup(5), c.GetDefaultGroup()) |
302 |
|
|
303 |
c.SetField("AREA") |
c.SetFieldInfo("AREA", None) |
304 |
s = ClassGroupSingleton(5) |
s = ClassGroupSingleton(5) |
305 |
c.AppendGroup(s) |
c.AppendGroup(s) |
306 |
self.assertEquals(c.FindGroup(5), s) |
self.assertEquals(c.FindGroup(5), s) |