From 9c0b0d348b448d68d28fbab661626ab2263e77d0 Mon Sep 17 00:00:00 2001 From: Manu Date: Wed, 23 Feb 2022 19:31:48 +0100 Subject: A test of gudhi.representations.Entropy has been added --- src/python/test/test_representations.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) (limited to 'src/python/test/test_representations.py') diff --git a/src/python/test/test_representations.py b/src/python/test/test_representations.py index 6a3dddc4..4a455bb6 100755 --- a/src/python/test/test_representations.py +++ b/src/python/test/test_representations.py @@ -160,8 +160,18 @@ def test_entropy_miscalculation(): l = l/sum(l) return -np.dot(l, np.log(l)) sce = Entropy(mode="scalar") - assert [[pe_max(diag_ex)]] == sce.fit_transform([diag_ex]) - + assert [[pe(diag_ex)]] == sce.fit_transform([diag_ex]) + sce = Entropy(mode="vector", resolution=4, normalized=False) + pef = [-1/4*np.log(1/4)-1/4*np.log(1/4)-1/2*np.log(1/2), + -1/4*np.log(1/4)-1/4*np.log(1/4)-1/2*np.log(1/2), + -1/2*np.log(1/2), + 0.0] + assert all(([pef] == sce.fit_transform([diag_ex]))[0]) + sce = Entropy(mode="vector", resolution=4, normalized=True) + pefN = (sce.fit_transform([diag_ex]))[0] + area = np.linalg.norm(pefN, ord=1) + assert area==1 + def test_kernel_empty_diagrams(): empty_diag = np.empty(shape = [0, 2]) assert SlicedWassersteinDistance(num_directions=100)(empty_diag, empty_diag) == 0. -- cgit v1.2.3