--- trunk/thuban/test/test_classgen.py 2003/07/08 13:23:20 1379 +++ trunk/thuban/test/test_classgen.py 2003/07/16 13:25:04 1436 @@ -21,7 +21,7 @@ generate_uniform_distribution, \ generate_quantiles, \ calculate_quantiles, \ - GreyRamp, CustomRamp + GreyRamp, CustomRamp, FixedRamp from Thuban.Model.range import Range from Thuban.Model.color import Color @@ -317,6 +317,24 @@ self.assertEquals(half.GetFill(), Color(1, 0.5, 0.5)) self.assertEquals(half.GetLineColor(), Color(0, 0.5, 0)) +class TestFixedRamp(unittest.TestCase): + + def test(self): + eq = self.assertEquals + + for lineColor, lineWidth, fillColor in \ + [(None, None, None), (Color(1, 1, 1), None, None), + (None, 4, None), (None, None, Color(0, 1, 0)), + (Color(1, 1, 1), 4, None), (Color(1, 1, 1), None, Color(0, 1, 0)), + (None, 4, Color(0, 1, 0)), (Color(1, 1, 1), 4, Color(0, 1, 0))]: + + framp = FixedRamp(GreyRamp, (lineColor, lineWidth, fillColor)) + + for i in [0.0, 0.2, 0.4, 0.6, 0.8, 1.0]: + props = framp.GetProperties(i) + if lineColor: eq(props.GetLineColor(), lineColor) + if lineWidth: eq(props.GetLineWidth(), lineWidth) + if fillColor: eq(props.GetFill(), fillColor) if __name__ == "__main__": unittest.main()