File tree 2 files changed +10
-4
lines changed
2 files changed +10
-4
lines changed Original file line number Diff line number Diff line change @@ -2097,17 +2097,23 @@ def test_set_index_cast_datetimeindex(self):
2097
2097
i = pd .DatetimeIndex (pd .tseries .tools .to_datetime (['2013-1-1 13:00' ,'2013-1-2 14:00' ], errors = "raise" )).tz_localize ('US/Pacific' )
2098
2098
df = DataFrame (np .random .randn (2 ,1 ),columns = ['A' ])
2099
2099
2100
- expected = Series (i )
2101
- self .assertTrue (expected .dtype == object )
2102
- self .assertTrue (i .equals (expected .values .values ))
2100
+ expected = Series (np .array ([pd .Timestamp ('2013-01-01 13:00:00-0800' , tz = 'US/Pacific' ),
2101
+ pd .Timestamp ('2013-01-02 14:00:00-0800' , tz = 'US/Pacific' )], dtype = "object" ))
2103
2102
2103
+ # convert index to series
2104
+ result = Series (i )
2105
+ assert_series_equal (result , expected )
2106
+
2107
+ # assignt to frame
2104
2108
df ['B' ] = i
2105
2109
result = df ['B' ]
2106
2110
assert_series_equal (result , expected )
2107
2111
2112
+ # keep the timezone
2108
2113
result = i .to_series (keep_tz = True )
2109
2114
assert_series_equal (result .reset_index (drop = True ), expected )
2110
2115
2116
+ # convert to utc
2111
2117
df ['C' ] = i .to_series ().reset_index (drop = True )
2112
2118
result = df ['C' ]
2113
2119
comp = DatetimeIndex (expected .values ).copy ()
Original file line number Diff line number Diff line change @@ -750,7 +750,7 @@ def to_series(self, keep_tz=False):
750
750
def _to_embed (self , keep_tz = False ):
751
751
""" return an array repr of this object, potentially casting to object """
752
752
if keep_tz and self .tz is not None and str (self .tz ) != 'UTC' :
753
- return self .asobject
753
+ return self .asobject . values
754
754
return self .values
755
755
756
756
@property
You can’t perform that action at this time.
0 commit comments