diff options
| author | Marc Glisse <marc.glisse@inria.fr> | 2020-04-20 12:16:15 +0200 |
|---|---|---|
| committer | Marc Glisse <marc.glisse@inria.fr> | 2020-04-20 12:16:15 +0200 |
| commit | 9ef7ba65367ab2ff92bf66b1b8166c5990530b76 (patch) | |
| tree | c9b814e942b05203fc0f56bf4628e1031d748bbb /src/python/test/test_knn.py | |
| parent | 1c1a99074049e4ff04fa28e7d6e1b6fc2067397a (diff) | |
Explicitly pass sort_results=True on some tests
Diffstat (limited to 'src/python/test/test_knn.py')
| -rwxr-xr-x | src/python/test/test_knn.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/python/test/test_knn.py b/src/python/test/test_knn.py index 415c9d48..a87ec212 100755 --- a/src/python/test/test_knn.py +++ b/src/python/test/test_knn.py @@ -54,12 +54,12 @@ def test_knn_explicit(): knn = KNearestNeighbors(2, metric="precomputed", return_index=True, return_distance=False) r = knn.fit_transform(dist) assert np.array_equal(r, [[0, 1], [1, 0], [2, 0]]) - knn = KNearestNeighbors(2, metric="precomputed", return_index=True, return_distance=True) + knn = KNearestNeighbors(2, metric="precomputed", return_index=True, return_distance=True, sort_results=True) r = knn.fit_transform(dist) assert np.array_equal(r[0], [[0, 1], [1, 0], [2, 0]]) assert np.array_equal(r[1], [[0, 3], [0, 1], [0, 1]]) # Second time in parallel - knn = KNearestNeighbors(2, metric="precomputed", return_index=True, return_distance=False, n_jobs=2) + knn = KNearestNeighbors(2, metric="precomputed", return_index=True, return_distance=False, n_jobs=2, sort_results=True) r = knn.fit_transform(dist) assert np.array_equal(r, [[0, 1], [1, 0], [2, 0]]) knn = KNearestNeighbors(2, metric="precomputed", return_index=True, return_distance=True, n_jobs=2) |