handle missing anndata better

KrishnaswamyLab · Jul 11, 2018 · f791b71 · f791b71
1 parent d232b77
commit f791b71
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,6 +3,7 @@ __pycache__
 build
 dist
 *egg-info
+.coverage
 
 #syncthing
 .syncthing.*

diff --git a/test/test_data.py b/test/test_data.py
@@ -12,12 +12,15 @@
     squareform,
     pdist,
 )
+import warnings
+
 try:
     import anndata
-except SyntaxError:
+except (ImportError, SyntaxError):
     # python2 support is missing
-    import sys
-    print("Warning: failed to import anndata", file=sys.stderr)
+    with warnings.catch_warnings():
+        warnings.filterwarnings("always")
+        warnings.warn("Warning: failed to import anndata", ImportWarning)
     pass
 
 #####################################################