Skip to content

Commit 7ad78e3

Browse files
committed
test with no head and pyarrow
1 parent 98d9050 commit 7ad78e3

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

pandas/tests/io/parser/test_header.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -684,3 +684,21 @@ def test_header_delim_whitespace(all_parsers):
684684
result = parser.read_csv(StringIO(data), delim_whitespace=True)
685685
expected = DataFrame({"a,b": ["1,2", "3,4"]})
686686
tm.assert_frame_equal(result, expected)
687+
688+
689+
def test_usecols_no_header_pyarrow(all_parsers):
690+
parser = all_parsers
691+
data = """
692+
a,i,x
693+
b,j,y
694+
"""
695+
result = parser.read_csv(
696+
StringIO(data),
697+
header=None,
698+
usecols=[0, 1],
699+
dtype="object",
700+
dtype_backend="pyarrow",
701+
engine="pyarrow",
702+
)
703+
expected = DataFrame([["a", "i"], ["b", "j"]])
704+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)