From 35d5ac4e6bb79ec41b35c0df611207b9cd578f49 Mon Sep 17 00:00:00 2001 From: Marc Glisse Date: Tue, 19 May 2020 18:10:16 +0200 Subject: Test with explicit weights and remove duplicated assignment --- src/python/test/test_dtm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/python/test/test_dtm.py') 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) -- cgit v1.2.3