From e2f6fb7f200f2949093cc43a913e9198c335c7da Mon Sep 17 00:00:00 2001
From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com>
Date: Tue, 28 Nov 2023 11:07:06 -0800
Subject: [PATCH] Avoid indirect imports

---
 pandas/tests/extension/test_masked.py              | 12 +++++++++---
 pandas/tests/indexes/interval/test_constructors.py |  5 +++--
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/pandas/tests/extension/test_masked.py b/pandas/tests/extension/test_masked.py
index a5e8b2ed1efe5..be4077d921a9e 100644
--- a/pandas/tests/extension/test_masked.py
+++ b/pandas/tests/extension/test_masked.py
@@ -24,6 +24,12 @@
 )
 from pandas.compat.numpy import np_version_gt2
 
+from pandas.core.dtypes.common import (
+    is_float_dtype,
+    is_signed_integer_dtype,
+    is_unsigned_integer_dtype,
+)
+
 import pandas as pd
 import pandas._testing as tm
 from pandas.core.arrays.boolean import BooleanDtype
@@ -281,7 +287,7 @@ def check_reduce(self, ser: pd.Series, op_name: str, skipna: bool):
         tm.assert_almost_equal(result, expected)
 
     def _get_expected_reduction_dtype(self, arr, op_name: str, skipna: bool):
-        if tm.is_float_dtype(arr.dtype):
+        if is_float_dtype(arr.dtype):
             cmp_dtype = arr.dtype.name
         elif op_name in ["mean", "median", "var", "std", "skew"]:
             cmp_dtype = "Float64"
@@ -289,7 +295,7 @@ def _get_expected_reduction_dtype(self, arr, op_name: str, skipna: bool):
             cmp_dtype = arr.dtype.name
         elif arr.dtype in ["Int64", "UInt64"]:
             cmp_dtype = arr.dtype.name
-        elif tm.is_signed_integer_dtype(arr.dtype):
+        elif is_signed_integer_dtype(arr.dtype):
             # TODO: Why does Window Numpy 2.0 dtype depend on skipna?
             cmp_dtype = (
                 "Int32"
@@ -297,7 +303,7 @@ def _get_expected_reduction_dtype(self, arr, op_name: str, skipna: bool):
                 or not IS64
                 else "Int64"
             )
-        elif tm.is_unsigned_integer_dtype(arr.dtype):
+        elif is_unsigned_integer_dtype(arr.dtype):
             cmp_dtype = (
                 "UInt32"
                 if (is_platform_windows() and (not np_version_gt2 or not skipna))
diff --git a/pandas/tests/indexes/interval/test_constructors.py b/pandas/tests/indexes/interval/test_constructors.py
index 078a0e06e0ed7..778c07b46e57c 100644
--- a/pandas/tests/indexes/interval/test_constructors.py
+++ b/pandas/tests/indexes/interval/test_constructors.py
@@ -3,6 +3,7 @@
 import numpy as np
 import pytest
 
+from pandas.core.dtypes.common import is_unsigned_integer_dtype
 from pandas.core.dtypes.dtypes import IntervalDtype
 
 from pandas import (
@@ -330,7 +331,7 @@ def get_kwargs_from_breaks(self, breaks, closed="right"):
         converts intervals in breaks format to a dictionary of kwargs to
         specific to the format expected by IntervalIndex.from_tuples
         """
-        if tm.is_unsigned_integer_dtype(breaks):
+        if is_unsigned_integer_dtype(breaks):
             pytest.skip(f"{breaks.dtype} not relevant IntervalIndex.from_tuples tests")
 
         if len(breaks) == 0:
@@ -388,7 +389,7 @@ def get_kwargs_from_breaks(self, breaks, closed="right"):
         converts intervals in breaks format to a dictionary of kwargs to
         specific to the format expected by the IntervalIndex/Index constructors
         """
-        if tm.is_unsigned_integer_dtype(breaks):
+        if is_unsigned_integer_dtype(breaks):
             pytest.skip(f"{breaks.dtype} not relevant for class constructor tests")
 
         if len(breaks) == 0: