Skip to content

Commit d39c474

Browse files
author
Egor
committed
BUG: #10833 Reworked test in accordance with the code style
1 parent 8ac1489 commit d39c474

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

pandas/tests/test_frame.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6322,16 +6322,16 @@ def test_to_csv_from_csv(self):
63226322
with ensure_clean() as path:
63236323
# GH 10833 (TimedeltaIndex formatting)
63246324
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)
63296329

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'])
63336333

6334-
self.assertTrue(df_test.equal(df_orig))
6334+
assert_frame_equal(df, result, check_index_type=True)
63356335

63366336
def test_to_csv_cols_reordering(self):
63376337
# GH3454

0 commit comments

Comments
 (0)