/[thuban]/trunk/thuban/test/test_classgen.py
ViewVC logotype

Diff of /trunk/thuban/test/test_classgen.py

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

revision 1166 by jonathan, Thu Jun 12 12:42:24 2003 UTC revision 1450 by bh, Fri Jul 18 10:18:56 2003 UTC
# Line 21  from Thuban.Model.classgen import \ Line 21  from Thuban.Model.classgen import \
21      generate_uniform_distribution, \      generate_uniform_distribution, \
22      generate_quantiles, \      generate_quantiles, \
23      calculate_quantiles, \      calculate_quantiles, \
24      GreyRamp      GreyRamp, CustomRamp, FixedRamp
25  from Thuban.Model.range import Range  from Thuban.Model.range import Range
26    from Thuban.Model.color import Color
27    
28  from Thuban.Model.classification import ClassGroupRange  from Thuban.Model.classification import ClassGroupRange, ClassGroupProperties
29    
30  class ClassGenTest(unittest.TestCase):  class ClassGenTest(unittest.TestCase):
31    
32      def doClassRangeTest(self, clazz, ranges):      def doClassRangeTest(self, clazz, ranges):
33          self.assertEquals(clazz.GetNumGroups(), len(ranges))          self.assertEquals(clazz.GetNumGroups(), len(ranges))
34          for i in range(len(ranges)):          for i in range(clazz.GetNumGroups()):
35              group = clazz.GetGroup(i)              group = clazz.GetGroup(i)
36              r1 = str(Range(ranges[i]))              r1 = str(Range(ranges[i]))
37              r2 = group.GetRange()              r2 = group.GetRange()
38              self.assertEquals(r1, r2)              self.assertEquals(r1, r2)
39    
40            self.doBoundsTest(clazz)
41    
42      def doClassSingleTest(self, clazz, _list):      def doClassSingleTest(self, clazz, _list):
43          self.assertEquals(clazz.GetNumGroups(), len(_list))          self.assertEquals(clazz.GetNumGroups(), len(_list))
44          for i in range(len(_list)):          for i in range(clazz.GetNumGroups()):
45              group = clazz.GetGroup(i)              group = clazz.GetGroup(i)
46              self.assertEquals(group.GetValue(), _list[i])              self.assertEquals(group.GetValue(), _list[i])
47    
48            self.doBoundsTest(clazz)
49    
50        def doBoundsTest(self, clazz, ramp = GreyRamp):
51    
52            #
53            # check that the properties are right (i.e. the first group
54            # is all white, the last is all black). This assumes that
55            # the GreyRamp, unless another is provided.
56            #
57            if clazz.GetNumGroups() >= 1:
58                groupF = clazz.GetGroup(0)
59                first = ramp.GetProperties(0)
60                self.assertEquals(groupF.GetProperties(), first)
61    
62            if clazz.GetNumGroups() >= 2:
63                groupL = clazz.GetGroup(clazz.GetNumGroups() - 1)
64                last = ramp.GetProperties(1)
65                self.assertEquals(groupL.GetProperties(), last)
66                            
67      def test_generate_singletons(self):      def test_generate_singletons(self):
68          """Test generate_singletons"""          """Test generate_singletons"""
69    
70          eq = self.assertEquals          eq = self.assertEquals
71          gs = generate_singletons          gs = generate_singletons
72          ramp = GreyRamp()          ramp = GreyRamp
73    
74          _list = [1, 2, 3, 4]          _list = [1, 2, 3, 4]
75          cl = gs(_list, len(_list), ramp)          cl = gs(_list, ramp)
76          self.doClassSingleTest(cl, _list)          self.doClassSingleTest(cl, _list)
77    
78          _list = range(0, 100)          _list = range(0, 100)
79          cl = gs(_list, len(_list), ramp)          cl = gs(_list, ramp)
80          self.doClassSingleTest(cl, _list)          self.doClassSingleTest(cl, _list)
81    
82          _list = range(-100, 100)          _list = range(-100, 100)
83          cl = gs(_list, len(_list), ramp)          cl = gs(_list, ramp)
84          self.doClassSingleTest(cl, _list)          self.doClassSingleTest(cl, _list)
85    
86          _list = ['a', 'b', 'c', 'd']          _list = ['a', 'b', 'c', 'd']
87          cl = gs(_list, len(_list), ramp)          cl = gs(_list, ramp)
88          self.doClassSingleTest(cl, _list)          self.doClassSingleTest(cl, _list)
89    
90          _list = []          _list = []
91          cl = gs(_list, len(_list), ramp)          cl = gs(_list, ramp)
92          self.doClassSingleTest(cl, _list)          self.doClassSingleTest(cl, _list)
93    
94          _list = [1]          _list = [1]
95          cl = gs(_list, len(_list), ramp)          cl = gs(_list, ramp)
96          self.doClassSingleTest(cl, _list)          self.doClassSingleTest(cl, _list)
97                    
98    
# Line 79  class ClassGenTest(unittest.TestCase): Line 101  class ClassGenTest(unittest.TestCase):
101    
102          eq = self.assertEquals          eq = self.assertEquals
103          gud = generate_uniform_distribution          gud = generate_uniform_distribution
104          ramp = GreyRamp()          ramp = GreyRamp
105    
106          cl = gud(0, 99, 10, ramp, True)          cl = gud(0, 99, 10, ramp, True)
107          self.doClassRangeTest(cl, ("[0;10[", "[10;20[", "[20;30[", "[30;40[",          self.doClassRangeTest(cl, ("[0;10[", "[10;20[", "[20;30[", "[30;40[",
# Line 110  class ClassGenTest(unittest.TestCase): Line 132  class ClassGenTest(unittest.TestCase):
132    
133          gq = generate_quantiles          gq = generate_quantiles
134    
135          ramp = GreyRamp()          ramp = GreyRamp
136    
137          adj, cl = gq([1, 2, 3, 4], [.25, .5, .75, 1.0], ramp, Range("[1;4]"))          adj, cl = gq([1, 2, 3, 4], [.25, .5, .75, 1.0], ramp, Range("[1;4]"))
138          self.failIf(adj)          self.failIf(adj)
# Line 277  class ClassGenTest(unittest.TestCase): Line 299  class ClassGenTest(unittest.TestCase):
299                      [.12, .24, .36, .50, .62, .76, .88, 1.0], Range("[1;5]"))                      [.12, .24, .36, .50, .62, .76, .88, 1.0], Range("[1;5]"))
300          eq(result, (1, 0, 24, [(4, .2), (9, .4), (14, .6), (19, .8), (24, 1.0)]))          eq(result, (1, 0, 24, [(4, .2), (9, .4), (14, .6), (19, .8), (24, 1.0)]))
301    
302    
303    class TestCustomRamp(unittest.TestCase):
304    
305        def test_color_interpolation(self):
306            """Test CustomRamp color interpolation"""
307            start = ClassGroupProperties()
308            start.SetFill(Color(1, 1, 1))
309            start.SetLineColor(Color(0, 0, 0))
310    
311            end = ClassGroupProperties()
312            end.SetFill(Color(1, 0, 0))
313            end.SetLineColor(Color(0, 1, 0))
314    
315            ramp = CustomRamp(start, end)
316            half = ramp.GetProperties(0.5)
317            self.assertEquals(half.GetFill(), Color(1, 0.5, 0.5))
318            self.assertEquals(half.GetLineColor(), Color(0, 0.5, 0))
319    
320    
321    class TestFixedRamp(unittest.TestCase):
322    
323        def test(self):
324            """Test FixedRamp"""
325            eq = self.assertEquals
326    
327            for lineColor, lineWidth, fillColor in \
328                [(None, None, None), (Color(1, 1, 1), None, None),
329                 (None, 4, None), (None, None, Color(0, 1, 0)),
330                 (Color(1, 1, 1), 4, None), (Color(1, 1, 1), None, Color(0, 1, 0)),
331                 (None, 4, Color(0, 1, 0)), (Color(1, 1, 1), 4, Color(0, 1, 0))]:
332    
333                framp = FixedRamp(GreyRamp, (lineColor, lineWidth, fillColor))
334    
335                for i in [0.0, 0.2, 0.4, 0.6, 0.8, 1.0]:
336                    props = framp.GetProperties(i)
337                    grey = Color(1 - i, 1 - i, 1 - i)
338                    if lineColor is not None:
339                        eq(props.GetLineColor(), lineColor)
340                    else:
341                        eq(props.GetLineColor(), grey)
342                    if lineWidth is not None:
343                        eq(props.GetLineWidth(), lineWidth)
344                    if fillColor is not None:
345                        eq(props.GetFill(), fillColor)
346                    else:
347                        eq(props.GetFill(), grey)
348    
349  if __name__ == "__main__":  if __name__ == "__main__":
350      unittest.main()      unittest.main()
351    
Legend:
Removed from v.1166  
changed lines
  Added in v.1450
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26