Skip to content

Commit

Permalink
try catch with print on the set_output test
Browse files Browse the repository at this point in the history
  • Loading branch information
Martin ROYER committed Jul 23, 2024
1 parent 032e85d commit 5fdf944
Showing 1 changed file with 12 additions and 9 deletions.
21 changes: 12 additions & 9 deletions src/python/test/test_sklearn_rips_persistence.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,12 +61,15 @@ def test_distance_matrix_rips_persistence_of_points_on_a_circle():


def test_set_output():
NB_PC = 5
point_clouds = [points.sphere(n_samples=random.randint(100, 150), ambient_dim=2) for _ in range(NB_PC)]

rips = RipsPersistence(homology_dimensions=[0, 2], n_jobs=-2)
diags_pandas = rips.set_output(transform="pandas").fit_transform(point_clouds)
assert 'H0' == diags_pandas.columns[0]
assert 'H2' == diags_pandas.columns[1]
assert len(diags_pandas.index) == NB_PC

try:
import pandas
NB_PC = 5
point_clouds = [points.sphere(n_samples=random.randint(100, 150), ambient_dim=2) for _ in range(NB_PC)]

rips = RipsPersistence(homology_dimensions=[0, 2], n_jobs=-2)
diags_pandas = rips.set_output(transform="pandas").fit_transform(point_clouds)
assert 'H0' == diags_pandas.columns[0]
assert 'H2' == diags_pandas.columns[1]
assert len(diags_pandas.index) == NB_PC
except ImportError:
print("Missing pandas, skipping set_output test")

0 comments on commit 5fdf944

Please sign in to comment.