/[thuban]/branches/WIP-pyshapelib-bramz/test/test_classgen.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/test/test_classgen.py

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

revision 1362 by jonathan, Wed Jul 2 12:59:34 2003 UTC revision 1528 by jonathan, Wed Jul 30 15:43:41 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      grey_ramp, CustomRamp, FixedRamp
25  from Thuban.Model.range import Range  from Thuban.Model.range import Range
26  from Thuban.Model.color import Color  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    
# Line 47  class ClassGenTest(unittest.TestCase): Line 47  class ClassGenTest(unittest.TestCase):
47    
48          self.doBoundsTest(clazz)          self.doBoundsTest(clazz)
49    
50      def doBoundsTest(self, clazz, ramp = GreyRamp):      def doBoundsTest(self, clazz, ramp = grey_ramp):
51    
52          #          #
53          # check that the properties are right (i.e. the first group          # check that the properties are right (i.e. the first group
54          # is all white, the last is all black). This assumes that          # is all white, the last is all black). This assumes that
55          # the GreyRamp, unless another is provided.          # the grey_ramp, unless another is provided.
56          #          #
57          if clazz.GetNumGroups() >= 1:          if clazz.GetNumGroups() >= 1:
58              groupF = clazz.GetGroup(0)              groupF = clazz.GetGroup(0)
# Line 69  class ClassGenTest(unittest.TestCase): Line 69  class ClassGenTest(unittest.TestCase):
69    
70          eq = self.assertEquals          eq = self.assertEquals
71          gs = generate_singletons          gs = generate_singletons
72          ramp = GreyRamp          ramp = grey_ramp
73    
74          _list = [1, 2, 3, 4]          _list = [1, 2, 3, 4]
75          cl = gs(_list, ramp)          cl = gs(_list, ramp)
# Line 101  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 = grey_ramp
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 132  class ClassGenTest(unittest.TestCase): Line 132  class ClassGenTest(unittest.TestCase):
132    
133          gq = generate_quantiles          gq = generate_quantiles
134    
135          ramp = GreyRamp          ramp = grey_ramp
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 299  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(grey_ramp, (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.1362  
changed lines
  Added in v.1528
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26