diff --git a/test/data/utils/test_data_utils.py b/test/data/utils/test_data_utils.py index 0aa10332..2bdf39c5 100644 --- a/test/data/utils/test_data_utils.py +++ b/test/data/utils/test_data_utils.py @@ -23,9 +23,14 @@ def setup_method(self): def test_get_complex_connectivity(self): """Test get_complex_connectivity.""" - out = get_complex_connectivity(self.complex, 2, neighborhoods=self.neighborhoods2) + out = get_complex_connectivity(self.complex, 3, neighborhoods=self.neighborhoods2) assert 'up_laplacian-0' in out.keys() + def test_get_combinatorial_complex_connectivity(self): + """Test get_complex_connectivity.""" + out = get_combinatorial_complex_connectivity(self.complex, 3) + assert 'adjacency_0' in out.keys() + def test_select_neighborhoods_of_interest(self): """Test select_neighborhoods_of_interest.""" connectivity = get_complex_connectivity(self.complex, 2) diff --git a/topobench/data/utils/__init__.py b/topobench/data/utils/__init__.py index 70126253..595de6e0 100644 --- a/topobench/data/utils/__init__.py +++ b/topobench/data/utils/__init__.py @@ -3,6 +3,7 @@ from .utils import ( ensure_serializable, # noqa: F401 generate_zero_sparse_connectivity, # noqa: F401 + get_combinatorial_complex_connectivity, # noqa: F401 get_complex_connectivity, # noqa: F401 get_routes_from_neighborhoods, # noqa: F401 load_cell_complex_dataset, # noqa: F401 @@ -13,6 +14,7 @@ ) utils_functions = [ + "get_combinatorial_complex_connectivity", "get_complex_connectivity", "get_routes_from_neighborhoods", "generate_zero_sparse_connectivity",