Allow tests to pass if fitsio not installed

rmjarvis · Jun 24, 2024 · 519eb41 · 519eb41
1 parent 84db2f5
commit 519eb41
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_gg.py b/tests/test_gg.py
@@ -200,7 +200,7 @@ def test_direct():
     with assert_raises(OSError):
         treecorr.NNCorrelation.from_file(ascii_name)
     # And gives error if not a valid treecorr output file.
-    with assert_raises(OSError):
+    with assert_raises((OSError, ImportError)):
         treecorr.Corr2.from_file(config['file_name'])
     with assert_raises(OSError):
         treecorr.GGCorrelation.from_file(config['file_name'])

diff --git a/tests/test_ggg.py b/tests/test_ggg.py
@@ -386,7 +386,7 @@ def test_direct_logruv():
     with assert_raises(OSError):
         treecorr.KKKCorrelation.from_file(ascii_name)
     # And gives error if not a valid treecorr output file.
-    with assert_raises(OSError):
+    with assert_raises((OSError, ImportError)):
         treecorr.Corr3.from_file(config['file_name'])
     with assert_raises(OSError):
         treecorr.GGGCorrelation.from_file(config['file_name'])

diff --git a/tests/test_kkk.py b/tests/test_kkk.py
@@ -237,7 +237,7 @@ def test_direct_logruv():
     with assert_raises(OSError):
         treecorr.GGGCorrelation.from_file(ascii_name)
     # And gives error if not a valid treecorr output file.
-    with assert_raises(OSError):
+    with assert_raises((OSError, ImportError)):
         treecorr.KKKCorrelation.from_file(config['file_name'])
     with assert_raises(OSError):
         treecorr.KKKCorrelation.from_file('invalid_file')