Restore NaN assert for zero waveforms

AxFoundation · Nov 8, 2023 · f056106 · f056106
1 parent 7c6f6b8
commit f056106
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_hitlet.py b/tests/test_hitlet.py
@@ -493,6 +493,9 @@ def test_conditional_entropy(data, size_template_and_ind_max_template):
         assert math.isclose(e1, e2, rel_tol=10**-3,
                             abs_tol=10**-3), f"Test 3.: Entropy function: {e1}, entropy test: {e2}"
 
+    else:
+        assert np.isnan(e1), f'Hitlet entropy is {e1}, but expected np.nan'
+
 
 def _align_compute_entropy(data, template):
     ind_max_data = np.argmax(data)