21 |
generate_uniform_distribution, \ |
generate_uniform_distribution, \ |
22 |
generate_quantiles, \ |
generate_quantiles, \ |
23 |
calculate_quantiles, \ |
calculate_quantiles, \ |
24 |
GreyRamp, CustomRamp |
GreyRamp, 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 |
|
|
317 |
self.assertEquals(half.GetFill(), Color(1, 0.5, 0.5)) |
self.assertEquals(half.GetFill(), Color(1, 0.5, 0.5)) |
318 |
self.assertEquals(half.GetLineColor(), Color(0, 0.5, 0)) |
self.assertEquals(half.GetLineColor(), Color(0, 0.5, 0)) |
319 |
|
|
320 |
|
class TestFixedRamp(unittest.TestCase): |
321 |
|
|
322 |
|
def test(self): |
323 |
|
eq = self.assertEquals |
324 |
|
|
325 |
|
for lineColor, lineWidth, fillColor in \ |
326 |
|
[(None, None, None), (Color(1, 1, 1), None, None), |
327 |
|
(None, 4, None), (None, None, Color(0, 1, 0)), |
328 |
|
(Color(1, 1, 1), 4, None), (Color(1, 1, 1), None, Color(0, 1, 0)), |
329 |
|
(None, 4, Color(0, 1, 0)), (Color(1, 1, 1), 4, Color(0, 1, 0))]: |
330 |
|
|
331 |
|
framp = FixedRamp(GreyRamp, (lineColor, lineWidth, fillColor)) |
332 |
|
|
333 |
|
for i in [0.0, 0.2, 0.4, 0.6, 0.8, 1.0]: |
334 |
|
props = framp.GetProperties(i) |
335 |
|
if lineColor: eq(props.GetLineColor(), lineColor) |
336 |
|
if lineWidth: eq(props.GetLineWidth(), lineWidth) |
337 |
|
if fillColor: eq(props.GetFill(), fillColor) |
338 |
|
|
339 |
if __name__ == "__main__": |
if __name__ == "__main__": |
340 |
unittest.main() |
unittest.main() |