Skip to content

Commit 04ab607

Browse files
committed
TST: ensure DataFrame.tail(0) preserves dtypes
1 parent 2d6c51f commit 04ab607

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pandas/tests/frame/methods/test_head_tail.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
from pandas import DataFrame
44
import pandas._testing as tm
55

6+
from pandas.core.series import Series
7+
68

79
def test_head_tail_generic(index, frame_or_series):
810
# GH#5370
@@ -55,3 +57,16 @@ def test_head_tail_empty():
5557
empty_df = DataFrame()
5658
tm.assert_frame_equal(empty_df.tail(), empty_df)
5759
tm.assert_frame_equal(empty_df.head(), empty_df)
60+
61+
def test_tail_zero_preserves_dtypes():
62+
df = DataFrame(
63+
{
64+
"A": Series([1, 2, 3], dtype="int64"),
65+
"B": Series([1.0, 2.0, 3.0], dtype="float64"),
66+
"C": Series(["a", "b", "c"], dtype="string"),
67+
}
68+
)
69+
70+
result = df.tail(0)
71+
72+
assert result.dtypes.to_dict() == df.dtypes.to_dict()

0 commit comments

Comments
 (0)