--- trunk/thuban/test/test_classgen.py	2003/07/02 12:59:34	1362
+++ trunk/thuban/test/test_classgen.py	2003/07/08 13:23:20	1379
@@ -21,11 +21,11 @@
     generate_uniform_distribution, \
     generate_quantiles, \
     calculate_quantiles, \
-    GreyRamp
+    GreyRamp, CustomRamp
 from Thuban.Model.range import Range
 from Thuban.Model.color import Color
 
-from Thuban.Model.classification import ClassGroupRange
+from Thuban.Model.classification import ClassGroupRange, ClassGroupProperties
 
 class ClassGenTest(unittest.TestCase):
 
@@ -299,6 +299,25 @@
                     [.12, .24, .36, .50, .62, .76, .88, 1.0], Range("[1;5]"))
         eq(result, (1, 0, 24, [(4, .2), (9, .4), (14, .6), (19, .8), (24, 1.0)]))
 
+
+class TestCustomRamp(unittest.TestCase):
+
+    def test_color_interpolation(self):
+        """Test CustomRamp color interpolation"""
+        start = ClassGroupProperties()
+        start.SetFill(Color(1, 1, 1))
+        start.SetLineColor(Color(0, 0, 0))
+
+        end = ClassGroupProperties()
+        end.SetFill(Color(1, 0, 0))
+        end.SetLineColor(Color(0, 1, 0))
+
+        ramp = CustomRamp(start, end)
+        half = ramp.GetProperties(0.5)
+        self.assertEquals(half.GetFill(), Color(1, 0.5, 0.5))
+        self.assertEquals(half.GetLineColor(), Color(0, 0.5, 0))
+
+
 if __name__ == "__main__":
     unittest.main()