diff options
author | Marc Glisse <marc.glisse@inria.fr> | 2020-05-19 20:46:38 +0200 |
---|---|---|
committer | Marc Glisse <marc.glisse@inria.fr> | 2020-05-19 20:46:38 +0200 |
commit | 7d9b2278891095b380fd051100bf4f09f00fddc9 (patch) | |
tree | d381675b9ff150417cc25ffebb73a8bb7c8f0f4a /src/python | |
parent | bee990ed74dec6d3257b89e1fcceb24b089d5c00 (diff) | |
parent | bb9b6b2a58d3b31a0e25d473339f2bde6430a52d (diff) |
Merge branch 'dtmdensity' into tomato2
Diffstat (limited to 'src/python')
-rwxr-xr-x | src/python/test/test_dtm.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/python/test/test_dtm.py b/src/python/test/test_dtm.py index 8d400c7e..0a52279e 100755 --- a/src/python/test/test_dtm.py +++ b/src/python/test/test_dtm.py @@ -84,5 +84,6 @@ def test_density(): assert density == pytest.approx(expected) distances = [[0, 1], [2, 0], [1, 3]] density = DTMDensity(metric="neighbors", dim=1).fit_transform(distances) - expected = numpy.array([2.0, 1.0, 0.5]) + assert density == pytest.approx(expected) + density = DTMDensity(weights=[0.5, 0.5], metric="neighbors", dim=1).fit_transform(distances) assert density == pytest.approx(expected) |