test(cursor): skips test using numpy, pandas if not available

Author: Brooke-white

test/integration/test_pandas.py

@@ -1,7 +1,6 @@
+from test.utils import numpy_only, pandas_only
 from warnings import filterwarnings
 
-import numpy as np  # type: ignore
-import pandas as pd  # type: ignore
 import pytest  # type: ignore
 
 import redshift_connector
@@ -30,7 +29,11 @@ def fin():
     return con
 
 
+@pandas_only
 def test_fetch_dataframe(db_table):
+    import numpy as np
+    import pandas as pd
+
     df = pd.DataFrame(
         np.array(
             [
@@ -53,7 +56,11 @@ def test_fetch_dataframe(db_table):
         assert result.columns[0] == "bookname"
 
 
+@pandas_only
 def test_write_dataframe(db_table):
+    import numpy as np
+    import pandas as pd
+
     df = pd.DataFrame(
         np.array(
             [
@@ -70,6 +77,7 @@ def test_write_dataframe(db_table):
         assert len(np.array(result)) == 2
 
 
+@numpy_only
 def test_fetch_numpyarray(db_table):
     with db_table.cursor() as cursor:
         cursor.executemany(

test/unit/test_cursor.py

@@ -1,4 +1,5 @@
 import typing
+from test.utils import pandas_only
 
 import pytest  # type: ignore
 from pytest_mock import mocker
@@ -11,6 +12,7 @@
 ]
 
 
+@pandas_only
 @pytest.mark.parametrize("_input", test_warn_response_data)
 def test_fetch_dataframe_warns_user(_input, mocker):
     data, exp_warning_msg = _input
@@ -21,6 +23,7 @@ def test_fetch_dataframe_warns_user(_input, mocker):
         mock_cursor.fetch_dataframe(1)
 
 
+@pandas_only
 def test_fetch_dataframe_no_results(mocker):
     mock_cursor: Cursor = Cursor.__new__(Cursor)
     mocker.patch("redshift_connector.Cursor._getDescription", return_value=["test"])

test/utils/__init__.py

@@ -0,0 +1 @@
+from .decorators import numpy_only, pandas_only

test/utils/decorators.py

@@ -0,0 +1,24 @@
+import pytest
+
+
+def is_numpy_installed() -> bool:
+    try:
+        import numpy
+
+        return True
+    except ModuleNotFoundError:
+        return False
+
+
+def is_pandas_installed() -> bool:
+    try:
+        import pandas
+
+        return True
+    except ModuleNotFoundError:
+        return False
+
+
+numpy_only: pytest.mark = pytest.mark.skipif(not is_numpy_installed(), reason="requires numpy")
+
+pandas_only: pytest.mark = pytest.mark.skipif(not is_pandas_installed(), reason="requires pandas")