diff options
| author | Vincent Rouvreau <10407034+VincentRouvreau@users.noreply.github.com> | 2023-01-05 10:34:00 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-05 10:34:00 +0100 |
| commit | b6171a12cfdeb26aa782a9dfe5d58177ca2239fb (patch) | |
| tree | 6089ddf51ed57c5592ba5843d1b5777cd251f114 /src/python/test/test_representations.py | |
| parent | 5b38d86d386c0a30eac401a35a5a737e7e49e68b (diff) | |
| parent | 86689f89bf896e41683fd7b1a4568f2b34ea505d (diff) | |
Merge pull request #782 from mglisse/sklearn-getparams
Fix get_params for vector_methods
Diffstat (limited to 'src/python/test/test_representations.py')
| -rwxr-xr-x | src/python/test/test_representations.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/python/test/test_representations.py b/src/python/test/test_representations.py index ae0362f8..f4ffbdc1 100755 --- a/src/python/test/test_representations.py +++ b/src/python/test/test_representations.py @@ -249,7 +249,7 @@ def test_landscape_nan_range(): dgm = np.array([[2., 6.], [3., 5.]]) lds = Landscape(num_landscapes=2, resolution=9, sample_range=[np.nan, 6.]) lds_dgm = lds(dgm) - assert (lds.sample_range[0] == 2) & (lds.sample_range[1] == 6) + assert (lds.sample_range_fixed[0] == 2) & (lds.sample_range_fixed[1] == 6) assert lds.new_resolution == 10 def test_endpoints(): @@ -263,3 +263,7 @@ def test_endpoints(): vec = BettiCurve(resolution=None) vec.fit(diags) assert np.equal(vec.grid_, [-np.inf, 2., 3.]).all() + +def test_get_params(): + for vec in [ Landscape(), Silhouette(), BettiCurve(), Entropy(mode="vector") ]: + vec.get_params() |