tests yay (negentropy)

rs-station · Aug 11, 2024 · 0ed4a34 · 0ed4a34
1 parent 687ce86
commit 0ed4a34
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/meteor/validate.py b/meteor/validate.py
@@ -2,7 +2,7 @@
 from scipy.stats import differential_entropy
 
 
-def negentropy(samples: np.ndarray, tolerance: float = 1e-4) -> float:
+def negentropy(samples: np.ndarray, tolerance: float = 0.01) -> float:
     """
     Return negentropy (float) of X (numpy array)
 

diff --git a/test/unit/test_validate.py b/test/unit/test_validate.py
@@ -3,9 +3,18 @@
 from meteor import validate
 
 def test_negentropy_gaussian() -> None:
-    n_samples = 100
+    n_samples = 10000
     samples = np.random.normal(size=n_samples)
     negentropy = validate.negentropy(samples)
 
-    # negentropy should be small for a Gaussian sample
-    assert np.abs(negentropy) < 1e-5
+    # negentropy should be zero for a Gaussian sample
+    assert np.abs(negentropy) < 1e-2
+
+
+def test_negentropy_uniform() -> None:
+    n_samples = 1000000
+    samples = np.random.uniform(size=n_samples)
+    negentropy = validate.negentropy(samples)
+
+    uniform_negentropy = (1./2.) * np.log(np.pi * np.exp(1) / 6.0)
+    assert np.abs(negentropy - uniform_negentropy) < 1e-2