def test_remapping(self):
"""Test that threshold values can be rescaled
"""
cpt = CPT(test_filename)
cpt.normalise()
assert allclose(cpt.segments[0].lower_bound, 0)
assert allclose(cpt.segments[0].upper_bound, 1./3)
assert allclose(cpt.segments[1].lower_bound, 1./3)
assert allclose(cpt.segments[1].upper_bound, 2./3)
assert allclose(cpt.segments[2].lower_bound, 2./3)
assert allclose(cpt.segments[2].upper_bound, 1.0)
# Test that colours and flags are unchanged
assert allclose(cpt.segments[0].rgb_min, [0,0,0])
assert allclose(cpt.segments[0].rgb_dif, [85,85,85])
assert allclose(cpt.segments[1].rgb_min, [85,85,85])
assert allclose(cpt.segments[1].rgb_dif, [170-85,170-85,170-85])
assert allclose(cpt.segments[2].rgb_min, [170,170,170])
assert allclose(cpt.segments[2].rgb_dif, [85,85,85])
assert cpt.segments[0].color_segment_boundary == 'L'
assert cpt.segments[1].color_segment_boundary == ''
assert cpt.segments[2].color_segment_boundary == 'U'
cpt.rescale(-10, 20)
assert allclose(cpt.segments[0].lower_bound, -10)
assert allclose(cpt.segments[0].upper_bound, 0)
assert allclose(cpt.segments[1].lower_bound, 0)
assert allclose(cpt.segments[1].upper_bound, 10)
assert allclose(cpt.segments[2].lower_bound, 10)
assert allclose(cpt.segments[2].upper_bound, 20)
# Test that colours and flags are unchanged
assert allclose(cpt.segments[0].rgb_min, [0,0,0])
assert allclose(cpt.segments[0].rgb_dif, [85,85,85])
assert allclose(cpt.segments[1].rgb_min, [85,85,85])
assert allclose(cpt.segments[1].rgb_dif, [170-85,170-85,170-85])
assert allclose(cpt.segments[2].rgb_min, [170,170,170])
assert allclose(cpt.segments[2].rgb_dif, [85,85,85])
assert cpt.segments[0].color_segment_boundary == 'L'
assert cpt.segments[1].color_segment_boundary == ''
assert cpt.segments[2].color_segment_boundary == 'U'
