From becaff79cd1d6dcbb73e9c6402d9433471b652a1 Mon Sep 17 00:00:00 2001 From: gfyoung Date: Tue, 2 Aug 2016 06:37:42 -0400 Subject: [PATCH] TST: Add first line comment tests in read_csv Closes gh-4623. --- pandas/io/tests/parser/comment.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/pandas/io/tests/parser/comment.py b/pandas/io/tests/parser/comment.py index f7cd1e190ec16..9987a017cf985 100644 --- a/pandas/io/tests/parser/comment.py +++ b/pandas/io/tests/parser/comment.py @@ -104,3 +104,15 @@ def test_custom_comment_char(self): result = self.read_csv(StringIO(data), comment='#') expected = DataFrame({'a': [1, 4], 'b': [2, 5], 'c': [3, 6]}) tm.assert_frame_equal(result, expected) + + def test_commment_first_line(self): + # see gh-4623 + data = '# notes\na,b,c\n# more notes\n1,2,3' + + expected = DataFrame([[1, 2, 3]], columns=['a', 'b', 'c']) + result = self.read_csv(StringIO(data), comment='#') + tm.assert_frame_equal(result, expected) + + expected = DataFrame({0: ['a', '1'], 1: ['b', '2'], 2: ['c', '3']}) + result = self.read_csv(StringIO(data), comment='#', header=None) + tm.assert_frame_equal(result, expected)