MCPcopy
hub / github.com/pandas-dev/pandas / test_usecols_str

Method test_usecols_str

pandas/tests/io/excel/test_readers.py:286–335  ·  view source on GitHub ↗
(self, request, engine, read_ext, df_ref)

Source from the content-addressed store, hash-verified

284 tm.assert_frame_equal(df2, expected)
285
286 def test_usecols_str(self, request, engine, read_ext, df_ref):
287 xfail_datetimes_with_pyxlsb(engine, request)
288
289 expected = df_ref[["A", "B", "C"]]
290 adjust_expected(expected, read_ext, engine)
291
292 df2 = pd.read_excel(
293 "test1" + read_ext, sheet_name="Sheet1", index_col=0, usecols="A:D"
294 )
295 df3 = pd.read_excel(
296 "test1" + read_ext,
297 sheet_name="Sheet2",
298 skiprows=[1],
299 index_col=0,
300 usecols="A:D",
301 )
302
303 # TODO add index to xls, read xls ignores index name ?
304 tm.assert_frame_equal(df2, expected)
305 tm.assert_frame_equal(df3, expected)
306
307 expected = df_ref[["B", "C"]]
308 adjust_expected(expected, read_ext, engine)
309
310 df2 = pd.read_excel(
311 "test1" + read_ext, sheet_name="Sheet1", index_col=0, usecols="A,C,D"
312 )
313 df3 = pd.read_excel(
314 "test1" + read_ext,
315 sheet_name="Sheet2",
316 skiprows=[1],
317 index_col=0,
318 usecols="A,C,D",
319 )
320 # TODO add index to xls file
321 tm.assert_frame_equal(df2, expected)
322 tm.assert_frame_equal(df3, expected)
323
324 df2 = pd.read_excel(
325 "test1" + read_ext, sheet_name="Sheet1", index_col=0, usecols="A,C:D"
326 )
327 df3 = pd.read_excel(
328 "test1" + read_ext,
329 sheet_name="Sheet2",
330 skiprows=[1],
331 index_col=0,
332 usecols="A,C:D",
333 )
334 tm.assert_frame_equal(df2, expected)
335 tm.assert_frame_equal(df3, expected)
336
337 @pytest.mark.parametrize(
338 "usecols", [[0, 1, 3], [0, 3, 1], [1, 0, 3], [1, 3, 0], [3, 0, 1], [3, 1, 0]]

Callers

nothing calls this directly

Calls 2

adjust_expectedFunction · 0.85

Tested by

no test coverage detected