STYLE: remove use in pd.api.dtype in test_ufunc.py (pandas-dev#39203)

Author: nofarmish

pandas/tests/frame/test_ufunc.py

@@ -3,6 +3,7 @@
 
 import pandas as pd
 import pandas._testing as tm
+from pandas.api.types import is_extension_array_dtype
 
 dtypes = [
     "int64",
@@ -26,7 +27,7 @@ def test_unary_unary(dtype):
 @pytest.mark.parametrize("dtype", dtypes)
 def test_unary_binary(request, dtype):
     # unary input, binary output
-    if pd.api.types.is_extension_array_dtype(dtype) or isinstance(dtype, dict):
+    if is_extension_array_dtype(dtype) or isinstance(dtype, dict):
         request.node.add_marker(
             pytest.mark.xfail(
                 reason="Extension / mixed with multiple outputs not implemented."
@@ -61,9 +62,9 @@ def test_binary_input_dispatch_binop(dtype):
 @pytest.mark.parametrize("dtype_b", dtypes)
 def test_binary_input_aligns_columns(request, dtype_a, dtype_b):
     if (
-        pd.api.types.is_extension_array_dtype(dtype_a)
+        is_extension_array_dtype(dtype_a)
         or isinstance(dtype_a, dict)
-        or pd.api.types.is_extension_array_dtype(dtype_b)
+        or is_extension_array_dtype(dtype_b)
         or isinstance(dtype_b, dict)
     ):
         request.node.add_marker(
@@ -89,7 +90,7 @@ def test_binary_input_aligns_columns(request, dtype_a, dtype_b):
 
 @pytest.mark.parametrize("dtype", dtypes)
 def test_binary_input_aligns_index(request, dtype):
-    if pd.api.types.is_extension_array_dtype(dtype) or isinstance(dtype, dict):
+    if is_extension_array_dtype(dtype) or isinstance(dtype, dict):
         request.node.add_marker(
             pytest.mark.xfail(
                 reason="Extension / mixed with multiple inputs not implemented."