Fix future warning for neighbors in tests

dwnorton · May 14, 2020 · 8e999a9 · 8e999a9
1 parent 058de9b
commit 8e999a9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/scanpy/tests/test_neighbors.py b/scanpy/tests/test_neighbors.py
@@ -1,3 +1,5 @@
+import warnings
+
 import numpy as np
 import pytest
 from anndata import AnnData
@@ -159,6 +161,9 @@ def test_restore_n_neighbors(neigh, conv):
     neigh.compute_neighbors(method='gauss', n_neighbors=n_neighbors)
 
     ad = AnnData(np.array(X))
-    ad.uns['neighbors'] = dict(connectivities=conv(neigh.connectivities))
+    # Allow deprecated usage for now
+    with warnings.catch_warnings():
+        warnings.filterwarnings("ignore", category=FutureWarning, module="anndata")
+        ad.uns['neighbors'] = dict(connectivities=conv(neigh.connectivities))
     neigh_restored = Neighbors(ad)
     assert neigh_restored.n_neighbors == 1