Skip to content

Commit 0079188

Browse files
Added a test case
1 parent 4d6c510 commit 0079188

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

pandas/tests/io/parser/test_common.py

+19
Original file line numberDiff line numberDiff line change
@@ -2135,3 +2135,22 @@ def test_no_header_two_extra_columns(all_parsers):
21352135
parser = all_parsers
21362136
df = parser.read_csv(stream, header=None, names=column_names, index_col=False)
21372137
tm.assert_frame_equal(df, ref)
2138+
2139+
2140+
def test_read_csv_names_types(all_parsers):
2141+
# GH 34946
2142+
data = """\
2143+
1,2,3
2144+
4,5,6
2145+
7,8,9
2146+
10,11,12\n"""
2147+
parser = all_parsers
2148+
msg = "Names should have consistent ordering. Consider a list instead."
2149+
names = "QAZ"
2150+
with pytest.raises(ValueError, match=msg):
2151+
parser.read_csv(StringIO(data), names=set(names))
2152+
2153+
ref = DataFrame(data={"Q": [1, 4, 7, 10], "A": [2, 5, 8, 11], "Z": [3, 6, 9, 12]})
2154+
for valid_type_converter in (list, tuple):
2155+
df = parser.read_csv(StringIO(data), names=valid_type_converter(names))
2156+
tm.assert_frame_equal(df, ref)

0 commit comments

Comments
 (0)