Skip to content

Raise error in read_csv when arguments header and prefix both are not None #31383

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 17 commits into from
Feb 3, 2020
Merged
29 changes: 15 additions & 14 deletions pandas/io/parsers.py
Original file line number Diff line number Diff line change
Expand Up @@ -1400,16 +1400,21 @@ def __init__(self, kwds):
"when specifying a multi-index header"
)

# GH 16338
elif self.header is not None and not is_integer(self.header):
raise ValueError("header must be integer or list of integers")

# GH 27779
elif self.header is not None and self.header < 0:
raise ValueError(
"Passing negative integer to header is invalid. "
"For no header, use header=None instead"
)
elif self.header:
# GH 16338
if not is_integer(self.header):
raise ValueError("header must be integer or list of integers")
# GH 27394
elif self.prefix:
raise ValueError(
"Argument prefix must be None if argument header is not None"
)
# GH 27779
elif self.header < 0:
raise ValueError(
"Passing negative integer to header is invalid. "
"For no header, use header=None instead"
)

self._name_processed = False

Expand Down Expand Up @@ -1888,10 +1893,6 @@ def __init__(self, src, **kwds):
if self._reader.header is None:
self.names = None
else:
if self.prefix:
raise ValueError(
"Argument prefix must be None if argument header is not None"
)
if len(self._reader.header) > 1:
# we have a multi index in the columns
(
Expand Down
7 changes: 4 additions & 3 deletions pandas/tests/io/parser/test_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -2040,12 +2040,13 @@ def test_read_csv_memory_growth_chunksize(all_parsers):
pass


def test_read_csv_raises_on_header_prefix():
def test_read_csv_raises_on_header_prefix(all_parsers):
# gh-27394
msg = "Argument prefix must be None if argument header is not None"
parser = all_parsers
msg = ("Argument prefix must be None if argument header is not None")
s = StringIO("0,1\n2,3")
with pytest.raises(ValueError, match=msg):
read_csv(s, header=0, prefix="_X")
parser.read_csv(s, header=0, prefix = "_X")


def test_read_table_equivalency_to_read_csv(all_parsers):
Expand Down