Skip to content
This repository was archived by the owner on Aug 9, 2023. It is now read-only.

Commit 4003a70

Browse files
author
aCampello
committed
Add new test that it re-loads the correct class
1 parent 3ce111b commit 4003a70

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_clustering.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,8 @@ def test_full_pipeline(reducer, cluster_reduced, tmp_path):
3131
# Asserts all coordinates of the loaded points are equal
3232
assert (cluster_new.embedded_points != cluster.embedded_points).sum() == 0
3333
assert (cluster_new.reduced_points != cluster.reduced_points).sum() == 0
34+
assert cluster_new.reducer_class.__class__ == cluster.reducer_class.__class__
35+
assert cluster_new.clustering_class.__class__ == cluster.clustering_class.__class__
3436

3537

3638
@pytest.mark.parametrize("reducer", ["tsne", "umap"])

0 commit comments

Comments
 (0)