From 649b8ea4bc8a3df099f4070337db7377cfb1096d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Guillermo=20Bern=C3=A1rdez?= Date: Fri, 7 Mar 2025 19:27:26 -0800 Subject: [PATCH] Coverage --- test/data/utils/test_data_utils.py | 7 ++++++- topobench/data/utils/__init__.py | 2 ++ 2 files changed, 8 insertions(+), 1 deletion(-) 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",