scikit-image

1
import numpy as np
2
from skimage.measure import shannon_entropy
3

4
from skimage._shared.testing import assert_almost_equal
5

6

7
def test_shannon_ones():
8
    img = np.ones((10, 10))
9
    res = shannon_entropy(img, base=np.e)
10
    assert_almost_equal(res, 0.0)
11

12

13
def test_shannon_all_unique():
14
    img = np.arange(64)
15
    res = shannon_entropy(img, base=2)
16
    assert_almost_equal(res, np.log(64) / np.log(2))
17