from pandas import DataFrame, Series import pandas._testing as tm class TestToFrame: def test_to_frame(self, datetime_series): datetime_series.name = None rs = datetime_series.to_frame() xp = DataFrame(datetime_series.values, index=datetime_series.index) tm.assert_frame_equal(rs, xp) datetime_series.name = "testname" rs = datetime_series.to_frame() xp = DataFrame( {"testname": datetime_series.values}, index=datetime_series.index ) tm.assert_frame_equal(rs, xp) rs = datetime_series.to_frame(name="testdifferent") xp = DataFrame( {"testdifferent": datetime_series.values}, index=datetime_series.index ) tm.assert_frame_equal(rs, xp) def test_to_frame_expanddim(self): # GH#9762 class SubclassedSeries(Series): @property def _constructor_expanddim(self): return SubclassedFrame class SubclassedFrame(DataFrame): pass ser = SubclassedSeries([1, 2, 3], name="X") result = ser.to_frame() assert isinstance(result, SubclassedFrame) expected = SubclassedFrame({"X": [1, 2, 3]}) tm.assert_frame_equal(result, expected)