1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
# Licensed under a 3-clause BSD style license - see LICENSE.rst
import numpy as np
from numpy.testing import assert_allclose
from ..ui import scale_image
from ...tests.helper import pytest
DATA = np.array([0, 1., 2.])
DATASCL = 0.5 * DATA
class TestImageScaling(object):
def test_linear(self):
"""Test linear scaling."""
img = scale_image(DATA, scale='linear')
assert_allclose(img, DATASCL, atol=0, rtol=1.e-5)
def test_sqrt(self):
"""Test sqrt scaling."""
img = scale_image(DATA, scale='sqrt')
assert_allclose(img, np.sqrt(DATASCL), atol=0, rtol=1.e-5)
def test_power(self):
"""Test power scaling."""
power = 3.0
img = scale_image(DATA, scale='power', power=power)
assert_allclose(img, DATASCL ** power, atol=0, rtol=1.e-5)
def test_log(self):
"""Test log10 scaling."""
img = scale_image(DATA, scale='log')
ref = np.log10(1000 * DATASCL + 1.0) / np.log10(1001.0)
assert_allclose(img, ref, atol=0, rtol=1.e-5)
def test_asinh(self):
"""Test asinh scaling."""
a = 0.1
img = scale_image(DATA, scale='asinh', asinh_a=a)
ref = np.arcsinh(DATASCL / a) / np.arcsinh(1. / a)
assert_allclose(img, ref, atol=0, rtol=1.e-5)
def test_min(self):
"""Test linear scaling."""
img = scale_image(DATA, scale='linear', min_cut=1.)
assert_allclose(img, [0., 0., 1.], atol=0, rtol=1.e-5)
def test_percent(self):
"""Test percent keywords."""
img = scale_image(DATA, scale='linear', percent=99.)
assert_allclose(img, DATASCL, atol=0, rtol=1.e-5)
img2 = scale_image(DATA, scale='linear', min_percent=0.5,
max_percent=99.5)
assert_allclose(img, img2, atol=0, rtol=1.e-5)
def test_invalid_stretch(self):
"""Test invalid stretch keyword."""
with pytest.raises(ValueError):
scale_image(DATA, scale='invalid')
|
