Skip to content

Commit 48fd726

Browse files
author
Evan Wright
committed
Simply and improve read_csv_infer_compression test
1 parent 6b5c454 commit 48fd726

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

pandas/io/tests/test_parsers.py

100644100755
+3-4
Original file line numberDiff line numberDiff line change
@@ -1072,17 +1072,16 @@ def test_read_csv_no_index_name(self):
10721072
tm.assert_frame_equal(df, df2)
10731073

10741074
def test_read_csv_infer_compression(self):
1075+
expected = self.read_csv(self.csv1, index_col=0, parse_dates=True)
1076+
10751077
inputs = [self.csv1, self.csv1 + '.gz',
10761078
self.csv1 + '.bz2', open(self.csv1)]
10771079

10781080
for f in inputs:
10791081
df = self.read_csv(f, index_col=0, parse_dates=True,
10801082
compression='infer')
10811083

1082-
self.assert_numpy_array_equal(df.columns, ['A', 'B', 'C', 'D'])
1083-
self.assertEqual(df.index.name, 'index')
1084-
self.assertIsInstance(df.index[0], (datetime, np.datetime64, Timestamp))
1085-
self.assertEqual(df.values.dtype, np.float64)
1084+
tm.assert_frame_equal(expected, df)
10861085

10871086
inputs[3].close()
10881087

0 commit comments

Comments
 (0)