21 |
generate_uniform_distribution, \ |
generate_uniform_distribution, \ |
22 |
generate_quantiles, \ |
generate_quantiles, \ |
23 |
calculate_quantiles, \ |
calculate_quantiles, \ |
24 |
GreyRamp |
GreyRamp, CustomRamp |
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 |
|
|
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 |
if __name__ == "__main__": |
if __name__ == "__main__": |
322 |
unittest.main() |
unittest.main() |
323 |
|
|