Move test_cut for nullable int

Author: Daniel Saxton

pandas/tests/arrays/test_integer.py

@@ -1061,19 +1061,6 @@ def test_value_counts_na():
     tm.assert_series_equal(result, expected)
 
 
-@pytest.mark.parametrize("bins", [3, [0, 5, 15]])
-@pytest.mark.parametrize("right", [True, False])
-@pytest.mark.parametrize("include_lowest", [True, False])
-def test_cut(bins, right, include_lowest):
-    a = np.random.randint(0, 10, size=50).astype(object)
-    a[::2] = np.nan
-    result = pd.cut(
-        pd.array(a, dtype="Int64"), bins, right=right, include_lowest=include_lowest
-    )
-    expected = pd.cut(a, bins, right=right, include_lowest=include_lowest)
-    tm.assert_categorical_equal(result, expected)
-
-
 def test_array_setitem_nullable_boolean_mask():
     # GH 31446
     ser = pd.Series([1, 2], dtype="Int64")

pandas/tests/reshape/test_cut.py

@@ -612,3 +612,16 @@ def test_cut_incorrect_labels(labels):
     msg = "Bin labels must either be False, None or passed in as a list-like argument"
     with pytest.raises(ValueError, match=msg):
         cut(values, 4, labels=labels)
+
+
+@pytest.mark.parametrize("bins", [3, [0, 5, 15]])
+@pytest.mark.parametrize("right", [True, False])
+@pytest.mark.parametrize("include_lowest", [True, False])
+def test_cut_nullable_integer(bins, right, include_lowest):
+    a = np.random.randint(0, 10, size=50).astype(object)
+    a[::2] = np.nan
+    result = cut(
+        pd.array(a, dtype="Int64"), bins, right=right, include_lowest=include_lowest
+    )
+    expected = cut(a, bins, right=right, include_lowest=include_lowest)
+    tm.assert_categorical_equal(result, expected)