Skip to content

Commit becaff7

Browse files
committed
TST: Add first line comment tests in read_csv
Closes pandas-devgh-4623.
1 parent 299fb75 commit becaff7

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

pandas/io/tests/parser/comment.py

+12
Original file line numberDiff line numberDiff line change
@@ -104,3 +104,15 @@ def test_custom_comment_char(self):
104104
result = self.read_csv(StringIO(data), comment='#')
105105
expected = DataFrame({'a': [1, 4], 'b': [2, 5], 'c': [3, 6]})
106106
tm.assert_frame_equal(result, expected)
107+
108+
def test_commment_first_line(self):
109+
# see gh-4623
110+
data = '# notes\na,b,c\n# more notes\n1,2,3'
111+
112+
expected = DataFrame([[1, 2, 3]], columns=['a', 'b', 'c'])
113+
result = self.read_csv(StringIO(data), comment='#')
114+
tm.assert_frame_equal(result, expected)
115+
116+
expected = DataFrame({0: ['a', '1'], 1: ['b', '2'], 2: ['c', '3']})
117+
result = self.read_csv(StringIO(data), comment='#', header=None)
118+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)