File tree 1 file changed +8
-8
lines changed
1 file changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -6322,16 +6322,16 @@ def test_to_csv_from_csv(self):
6322
6322
with ensure_clean() as path:
6323
6323
# GH 10833 (TimedeltaIndex formatting)
6324
6324
dt = pd.Timedelta(seconds=1)
6325
- df_orig = pd.DataFrame({'data ': list( range(10)) },
6326
- index=[i*dt for i in range(10)])
6327
- df_orig.index.rename('timestamp', inplace=True )
6328
- df_orig .to_csv(path)
6325
+ df = pd.DataFrame({'dt_data ': [i*dt for i in range(3)] },
6326
+ index=pd.Index( [i*dt for i in range(3)],
6327
+ name='dt_index') )
6328
+ df .to_csv(path)
6329
6329
6330
- df_test = pd.read_csv(path, index_col='timestamp ')
6331
- df_test .index = pd.to_timedelta(df_test .index)
6332
- df_test.index.rename('timestamp', inplace=True )
6330
+ result = pd.read_csv(path, index_col='dt_index ')
6331
+ result .index = pd.to_timedelta(result .index)
6332
+ result['dt_data'] = pd.to_timedelta(result['dt_data'] )
6333
6333
6334
- self.assertTrue(df_test.equal(df_orig) )
6334
+ assert_frame_equal(df, result, check_index_type=True )
6335
6335
6336
6336
def test_to_csv_cols_reordering(self):
6337
6337
# GH3454
You can’t perform that action at this time.
0 commit comments