TST: Use more pytest.importorskip (#54377)

* TST: Use more pytest.importorskip

* Fix xml

* Final cleanup

* Fix condition

* Fix downstream

* cleanup after matplotlib

Author: mroeschke

pandas/tests/arrays/interval/test_interval.py

@@ -1,8 +1,6 @@
 import numpy as np
 import pytest
 
-import pandas.util._test_decorators as td
-
 import pandas as pd
 from pandas import (
     Index,
@@ -249,12 +247,8 @@ def test_min_max(self, left_right_dtypes, index_or_series_or_array):
 # Arrow interaction
 
 
-pyarrow_skip = td.skip_if_no("pyarrow")
-
-
-@pyarrow_skip
 def test_arrow_extension_type():
-    import pyarrow as pa
+    pa = pytest.importorskip("pyarrow")
 
     from pandas.core.arrays.arrow.extension_types import ArrowIntervalType
 
@@ -269,9 +263,8 @@ def test_arrow_extension_type():
     assert hash(p1) != hash(p3)
 
 
-@pyarrow_skip
 def test_arrow_array():
-    import pyarrow as pa
+    pa = pytest.importorskip("pyarrow")
 
     from pandas.core.arrays.arrow.extension_types import ArrowIntervalType
 
@@ -299,9 +292,8 @@ def test_arrow_array():
         pa.array(intervals, type=ArrowIntervalType(pa.float64(), "left"))
 
 
-@pyarrow_skip
 def test_arrow_array_missing():
-    import pyarrow as pa
+    pa = pytest.importorskip("pyarrow")
 
     from pandas.core.arrays.arrow.extension_types import ArrowIntervalType
 
@@ -329,14 +321,13 @@ def test_arrow_array_missing():
     assert result.storage.equals(expected)
 
 
-@pyarrow_skip
 @pytest.mark.parametrize(
     "breaks",
     [[0.0, 1.0, 2.0, 3.0], date_range("2017", periods=4, freq="D")],
     ids=["float", "datetime64[ns]"],
 )
 def test_arrow_table_roundtrip(breaks):
-    import pyarrow as pa
+    pa = pytest.importorskip("pyarrow")
 
     from pandas.core.arrays.arrow.extension_types import ArrowIntervalType
 
@@ -363,14 +354,13 @@ def test_arrow_table_roundtrip(breaks):
     tm.assert_frame_equal(result, expected[0:0])
 
 
-@pyarrow_skip
 @pytest.mark.parametrize(
     "breaks",
     [[0.0, 1.0, 2.0, 3.0], date_range("2017", periods=4, freq="D")],
     ids=["float", "datetime64[ns]"],
 )
 def test_arrow_table_roundtrip_without_metadata(breaks):
-    import pyarrow as pa
+    pa = pytest.importorskip("pyarrow")
 
     arr = IntervalArray.from_breaks(breaks)
     arr[1] = None
@@ -386,12 +376,11 @@ def test_arrow_table_roundtrip_without_metadata(breaks):
     tm.assert_frame_equal(result, df)
 
 
-@pyarrow_skip
 def test_from_arrow_from_raw_struct_array():
     # in case pyarrow lost the Interval extension type (eg on parquet roundtrip
     # with datetime64[ns] subtype, see GH-45881), still allow conversion
     # from arrow to IntervalArray
-    import pyarrow as pa
+    pa = pytest.importorskip("pyarrow")
 
     arr = pa.array([{"left": 0, "right": 1}, {"left": 1, "right": 2}])
     dtype = pd.IntervalDtype(np.dtype("int64"), closed="neither")

pandas/tests/arrays/string_/test_string.py

@@ -5,8 +5,6 @@
 import numpy as np
 import pytest
 
-import pandas.util._test_decorators as td
-
 from pandas.core.dtypes.common import is_dtype_equal
 
 import pandas as pd
@@ -420,10 +418,9 @@ def test_arrow_array(dtype):
     assert arr.equals(expected)
 
 
-@td.skip_if_no("pyarrow")
 def test_arrow_roundtrip(dtype, string_storage2):
     # roundtrip possible from arrow 1.0.0
-    import pyarrow as pa
+    pa = pytest.importorskip("pyarrow")
 
     data = pd.array(["a", "b", None], dtype=dtype)
     df = pd.DataFrame({"a": data})
@@ -438,10 +435,9 @@ def test_arrow_roundtrip(dtype, string_storage2):
     assert result.loc[2, "a"] is pd.NA
 
 
-@td.skip_if_no("pyarrow")
 def test_arrow_load_from_zero_chunks(dtype, string_storage2):
     # GH-41040
-    import pyarrow as pa
+    pa = pytest.importorskip("pyarrow")
 
     data = pd.array([], dtype=dtype)
     df = pd.DataFrame({"a": data})

pandas/tests/frame/test_api.py

@@ -7,10 +7,7 @@
 
 from pandas._config.config import option_context
 
-from pandas.util._test_decorators import (
-    async_mark,
-    skip_if_no,
-)
+from pandas.util._test_decorators import async_mark
 
 import pandas as pd
 from pandas import (
@@ -373,9 +370,9 @@ def test_constructor_expanddim(self):
         with pytest.raises(AttributeError, match=msg):
             df._constructor_expanddim(np.arange(27).reshape(3, 3, 3))
 
-    @skip_if_no("jinja2")
     def test_inspect_getmembers(self):
         # GH38740
+        pytest.importorskip("jinja2")
         df = DataFrame()
         msg = "DataFrame._data is deprecated"
         with tm.assert_produces_warning(

pandas/tests/frame/test_ufunc.py

@@ -4,8 +4,6 @@
 import numpy as np
 import pytest
 
-import pandas.util._test_decorators as td
-
 import pandas as pd
 import pandas._testing as tm
 from pandas.api.types import is_extension_array_dtype
@@ -247,18 +245,14 @@ def test_alignment_deprecation_enforced():
         np.add(s2, df1)
 
 
-@td.skip_if_no("numba")
 def test_alignment_deprecation_many_inputs_enforced():
     # Enforced in 2.0
     # https://github.com/pandas-dev/pandas/issues/39184
     # test that the deprecation also works with > 2 inputs -> using a numba
     # written ufunc for this because numpy itself doesn't have such ufuncs
-    from numba import (
-        float64,
-        vectorize,
-    )
+    numba = pytest.importorskip("numba")
 
-    @vectorize([float64(float64, float64, float64)])
+    @numba.vectorize([numba.float64(numba.float64, numba.float64, numba.float64)])
     def my_ufunc(x, y, z):
         return x + y + z
 

pandas/tests/generic/test_to_xarray.py

@@ -1,8 +1,6 @@
 import numpy as np
 import pytest
 
-import pandas.util._test_decorators as td
-
 from pandas import (
     Categorical,
     DataFrame,
@@ -12,8 +10,9 @@
 )
 import pandas._testing as tm
 
+pytest.importorskip("xarray")
+
 
-@td.skip_if_no("xarray")
 class TestDataFrameToXArray:
     @pytest.fixture
     def df(self):
@@ -84,7 +83,6 @@ def test_to_xarray_with_multiindex(self, df):
         tm.assert_frame_equal(result, expected)
 
 
-@td.skip_if_no("xarray")
 class TestSeriesToXArray:
     def test_to_xarray_index_types(self, index_flat):
         index = index_flat