fix #833: marked tests to be skipped if a necessary optional dependency is missing

Author: gdementen

larray/tests/common.py

@@ -11,6 +11,18 @@
     import xlwings as xw
 except ImportError:
     xw = None
+try:
+    import tables
+except ImportError:
+    tables = None
+try:
+    import xlrd
+except ImportError:
+    xlrd = None
+try:
+    import xlsxwriter
+except ImportError:
+    xlsxwriter = None
 
 from larray import Array, isnan, asarray, Metadata
 
@@ -134,6 +146,10 @@ def meta():
 
 
 needs_xlwings = pytest.mark.skipif(xw is None, reason="xlwings is required for this test")
+needs_pytables = pytest.mark.skipif(tables is None, reason="pytables is required for this test")
+needs_xlrd = pytest.mark.skipif(xlrd is None, reason="xlrd is required for this test")
+needs_xlsxwriter = pytest.mark.skipif(xlsxwriter is None, reason="xlsxwriter is required for this test")
+
 needs_python35 = pytest.mark.skipif(sys.version_info < (3, 5), reason="Python 3.5 is required for this test")
 needs_python36 = pytest.mark.skipif(sys.version_info < (3, 6), reason="Python 3.6 is required for this test")
 needs_python37 = pytest.mark.skipif(sys.version_info < (3, 7), reason="Python 3.7 is required for this test")

larray/tests/test_array.py

@@ -10,14 +10,10 @@
 import pandas as pd
 from collections import OrderedDict
 
-try:
-    import xlwings as xw
-except ImportError:
-    xw = None
-
-from larray.tests.common import (inputpath, assert_array_equal, assert_array_nan_equal, assert_larray_equiv,
-                                 tmp_path, meta, needs_xlwings, needs_python35, needs_python36, needs_python37,
-                                 assert_larray_equal)
+from larray.tests.common import (inputpath, tmp_path, meta,
+                                 assert_array_equal, assert_array_nan_equal, assert_larray_equiv, assert_larray_equal,
+                                 needs_xlwings, needs_pytables, needs_xlsxwriter, needs_xlrd,
+                                 needs_python35, needs_python36, needs_python37)
 from larray import (Array, LArray, Axis, LGroup, union, zeros, zeros_like, ndtest, empty, ones, eye, diag, stack,
                     clip, exp, where, X, mean, isnan, round, read_hdf, read_csv, read_eurostat, read_excel,
                     from_lists, from_string, open_excel, from_frame, sequence, nan, IGroup)
@@ -116,6 +112,7 @@ def test_read_set_update_delete_metadata(meta, tmpdir):
     assert meta2 == meta
 
 
+@needs_pytables
 def test_metadata_hdf(meta, tmpdir):
     key = 'meta'
     fname = os.path.join(tmpdir.strpath, 'test_metadata.hdf')
@@ -3092,6 +3089,7 @@ def test_extend(small_array):
     assert small_array.shape == (3, 16)
 
 
+@needs_pytables
 def test_hdf_roundtrip(tmpdir, meta):
     a = ndtest((2, 3), meta=meta)
     fpath = tmp_path(tmpdir, 'test.h5')
@@ -3336,6 +3334,7 @@ def test_read_excel_xlwings():
     assert_array_equal(bad4, good2)
 
 
+@needs_xlrd
 def test_read_excel_pandas():
     arr = read_excel(inputpath('test.xlsx'), '1d', engine='xlrd')
     assert_array_equal(arr, io_1d)
@@ -3906,6 +3905,7 @@ def test_to_csv(tmpdir):
         assert f.readlines() == result
 
 
+@needs_xlsxwriter
 def test_to_excel_xlsxwriter(tmpdir):
     fpath = tmp_path(tmpdir, 'test_to_excel_xlsxwriter.xlsx')
 

larray/tests/test_axis.py

@@ -5,7 +5,7 @@
 import os.path
 import numpy as np
 
-from larray.tests.common import assert_array_equal, assert_nparray_equal
+from larray.tests.common import assert_array_equal, assert_nparray_equal, needs_pytables
 from larray import Axis, LGroup, IGroup, read_hdf, X
 from larray.core.axis import AxisReference
 
@@ -486,6 +486,7 @@ def test_contains():
     assert age[['2', '7']] not in agg
 
 
+@needs_pytables
 def test_h5_io(tmpdir):
     age = Axis('age=0..10')
     lipro = Axis('lipro=P01..P05')

larray/tests/test_group.py

@@ -5,7 +5,7 @@
 import os.path
 import numpy as np
 
-from larray.tests.common import assert_array_equal
+from larray.tests.common import assert_array_equal, needs_pytables
 from larray import Axis, IGroup, LGroup, LSet, ndtest, read_hdf
 from larray.util.oset import OrderedSet
 
@@ -186,6 +186,7 @@ def test_to_lset_lgroup():
     assert res.key == OrderedSet(['c'])
 
 
+@needs_pytables
 def test_h5_io_lgroup(tmpdir):
     fpath = os.path.join(str(tmpdir), 'lgroups.h5')
     age.to_hdf(fpath)
@@ -426,6 +427,7 @@ def test_to_lset_igroup():
     assert res.key == OrderedSet(['c'])
 
 
+@needs_pytables
 def test_h5_io_igroup(tmpdir):
     fpath = os.path.join(str(tmpdir), 'igroups.h5')
     age.to_hdf(fpath)

larray/tests/test_session.py

@@ -7,7 +7,8 @@
 import pandas as pd
 import pytest
 
-from larray.tests.common import assert_array_nan_equal, inputpath, tmp_path, meta, needs_xlwings
+from larray.tests.common import (assert_array_nan_equal, inputpath, tmp_path, meta,
+                                 needs_xlwings, needs_pytables, needs_xlrd)
 from larray import (Session, Axis, Array, Group, isnan, zeros_like, ndtest, ones_like, ones, full,
                     local_arrays, global_arrays, arrays)
 from larray.util.misc import pickle, PY2
@@ -54,22 +55,27 @@ def test_init_session(meta):
     s = Session(b, b024, a, a01, a2=a2, anonymous=anonymous, ano01=ano01, c=c, d=d, e=e, f=f, g=g, h=h)
     assert s.names == ['a', 'a01', 'a2', 'ano01', 'anonymous', 'b', 'b024', 'c', 'd', 'e', 'f', 'g', 'h']
 
-    s = Session(inputpath('test_session.h5'))
-    assert s.names == ['e', 'f', 'g']
-
-    # this needs xlwings installed
-    # s = Session('test_session_ef.xlsx')
-    # assertEqual(s.names, ['e', 'f'])
-
     # TODO: format autodetection does not work in this case
     # s = Session('test_session_csv')
-    # assertEqual(s.names, ['e', 'f', 'g'])
+    # assert s.names == ['e', 'f', 'g']
 
     # metadata
     s = Session(b, b024, a, a01, a2=a2, anonymous=anonymous, ano01=ano01, c=c, d=d, e=e, f=f, g=g, h=h, meta=meta)
     assert s.meta == meta
 
 
+@needs_xlwings
+def test_init_session_xlsx():
+    s = Session(inputpath('demography_eurostat.xlsx'))
+    assert s.names == ['births', 'deaths', 'immigration', 'pop', 'pop_benelux']
+
+
+@needs_pytables
+def test_init_session_hdf():
+    s = Session(inputpath('test_session.h5'))
+    assert s.names == ['e', 'f', 'g']
+
+
 def test_getitem(session):
     assert session['a'] is a
     assert session['a2'] is a2
@@ -219,11 +225,13 @@ def _test_io(fpath, session, meta, engine):
         assert s.meta == meta
 
 
+@needs_pytables
 def test_h5_io(tmpdir, session, meta):
     fpath = tmp_path(tmpdir, 'test_session.h5')
     _test_io(fpath, session, meta, engine='pandas_hdf')
 
 
+@needs_xlrd
 def test_xlsx_pandas_io(tmpdir, session, meta):
     fpath = tmp_path(tmpdir, 'test_session.xlsx')
     _test_io(fpath, session, meta, engine='pandas_excel')