add: tests

Author: shizuo-kaji

tests/test_mod.py

@@ -0,0 +1,24 @@
+import numpy as np
+
+import cripser
+import tcripser
+
+
+def test_cripser_module_on_3d_hole():
+    arr = np.load('sample/3d_hole.npy').astype(np.float64)
+    ph = cripser.computePH(arr, maxdim=2)
+    assert ph.ndim == 2 and ph.shape[1] == 9
+    dims = set(ph[:, 0].astype(int))
+    # Expect H0 and H2 features on a 3D hole dataset
+    assert 0 in dims
+    assert 2 in dims
+
+
+def test_tcripser_module_on_3d_hole():
+    arr = np.load('sample/3d_hole.npy').astype(np.float64)
+    ph = tcripser.computePH(arr, maxdim=2)
+    assert ph.ndim == 2 and ph.shape[1] == 9
+    dims = set(ph[:, 0].astype(int))
+    # Expect non-empty result and some higher-dimensional features
+    assert len(dims) >= 1
+    assert any(d in dims for d in (1, 2))

tests/test_utils.py

@@ -0,0 +1,34 @@
+import numpy as np
+
+from cripser import (
+    compute_ph,
+    to_gudhi_diagrams,
+    to_gudhi_persistence,
+    group_by_dim,
+)
+
+
+def test_utils_converters_basic():
+    # Simple 2D constant image ensures an infinite H0 bar
+    arr = np.zeros((3, 3), dtype=np.float64)
+    ph = compute_ph(arr, maxdim=1)
+
+    # Base shape checks
+    assert ph.ndim == 2 and ph.shape[1] == 9
+
+    # Diagrams
+    dgms = to_gudhi_diagrams(ph, maxdim=1)
+    assert isinstance(dgms, list) and len(dgms) == 2
+    assert dgms[0].ndim == 2 and dgms[0].shape[1] == 2
+    # Expect at least one infinite death in H0
+    assert np.isinf(dgms[0][:, 1]).any()
+
+    # Persistence list
+    pers = to_gudhi_persistence(ph)
+    assert any(d == 0 and np.isinf(bd[1]) for d, bd in pers)
+
+    # Group by dimension
+    groups = group_by_dim(ph)
+    assert len(groups) >= 1
+    if len(groups[0]) > 0:
+        assert np.all(groups[0][:, 0] == 0)