summaryrefslogtreecommitdiff
path: root/src/python/test/test_tomato.py
diff options
context:
space:
mode:
authorMarc Glisse <marc.glisse@inria.fr>2021-04-15 20:41:49 +0200
committerMarc Glisse <marc.glisse@inria.fr>2021-04-15 20:41:49 +0200
commitd2f12c8563d9f7aa1f8ead0da0034796a88704e9 (patch)
tree7f436a82dece398befeadb437d54459b9806e76d /src/python/test/test_tomato.py
parent7e05e915adc1be285e04eb00d3ab7ba1b797f38d (diff)
Compare lists with ==
numpy/core/_asarray.py:102: VisibleDeprecationWarning: Creating an ndarray from ragged nested sequences (which is a list-or-tuple of lists-or-tuples-or ndarrays with different lengths or shapes) is deprecated. If you meant to do this, you must specify 'dtype=object' when creating the ndarray.
Diffstat (limited to 'src/python/test/test_tomato.py')
-rwxr-xr-xsrc/python/test/test_tomato.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/test/test_tomato.py b/src/python/test/test_tomato.py
index ecab03c4..c571f799 100755
--- a/src/python/test/test_tomato.py
+++ b/src/python/test/test_tomato.py
@@ -37,7 +37,7 @@ def test_tomato_1():
t = Tomato(metric="euclidean", graph_type="radius", r=4.7, k=4)
t.fit(a)
assert t.max_weight_per_cc_.size == 2
- assert np.array_equal(t.neighbors_, [[0, 1, 2], [0, 1, 2], [0, 1, 2], [3, 4, 5, 6], [3, 4, 5], [3, 4, 5], [3, 6]])
+ assert t.neighbors_ == [[0, 1, 2], [0, 1, 2], [0, 1, 2], [3, 4, 5, 6], [3, 4, 5], [3, 4, 5], [3, 6]]
t.plot_diagram()
t = Tomato(graph_type="radius", r=4.7, k=4, symmetrize_graph=True)