Skip to content

Commit 32f1fad

Browse files
authored
TEST: cut() with nullable Int64 dtype (#51384)
1 parent 4ee08b4 commit 32f1fad

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pandas/tests/reshape/test_cut.py

+15
Original file line numberDiff line numberDiff line change
@@ -744,3 +744,18 @@ def test_cut_bins_datetime_intervalindex():
744744
result = cut(Series([Timestamp("2022-02-26")]), bins=bins)
745745
expected = Categorical.from_codes([0], bins, ordered=True)
746746
tm.assert_categorical_equal(result.array, expected)
747+
748+
749+
def test_cut_with_nullable_int64():
750+
# GH 30787
751+
series = Series([0, 1, 2, 3, 4, pd.NA, 6, 7], dtype="Int64")
752+
bins = [0, 2, 4, 6, 8]
753+
intervals = IntervalIndex.from_breaks(bins)
754+
755+
expected = Series(
756+
Categorical.from_codes([-1, 0, 0, 1, 1, -1, 2, 3], intervals, ordered=True)
757+
)
758+
759+
result = cut(series, bins=bins)
760+
761+
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)