Skip to content

Commit 00e315a

Browse files
author
GitHub Actions
committed
Add tests for scalar row + slice columns dtype preservation
1 parent e0162cc commit 00e315a

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

pandas/tests/indexing/test_iloc.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1548,3 +1548,11 @@ def test_setitem_pyarrow_int_series(self):
15481548

15491549
expected = Series([7, 8, 3], dtype="int64[pyarrow]")
15501550
tm.assert_series_equal(ser, expected)
1551+
1552+
1553+
def test_iloc_scalar_row_slice_columns_dtype(self):
1554+
# GH 63071
1555+
df = DataFrame([["a", 1.0, 2.0], ["b", 3.0, 4.0]])
1556+
result = df.iloc[0, 1:]
1557+
expected = Series([1.0, 2.0], index=[1, 2], dtype=float, name=0)
1558+
tm.assert_series_equal(result, expected)

pandas/tests/indexing/test_loc.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,14 @@ def test_loc_dtype():
6969
tm.assert_series_equal(result, expected)
7070

7171

72+
def test_loc_scalar_row_slice_columns_dtype():
73+
# GH 63071
74+
df = DataFrame([["a", 1.0, 2.0], ["b", 3.0, 4.0]])
75+
result = df.loc[0, 1:]
76+
expected = Series([1.0, 2.0], index=[1, 2], dtype=float, name=0)
77+
tm.assert_series_equal(result, expected)
78+
79+
7280
class TestLoc:
7381
def test_none_values_on_string_columns(self, using_infer_string):
7482
# Issue #32218

0 commit comments

Comments
 (0)