Skip to content

Commit 468d68b

Browse files
author
Pankaj Pandey
committed
Move test to CParserTests
The behavior is only applicable on the `c` engine.
1 parent 6547762 commit 468d68b

File tree

2 files changed

+29
-25
lines changed

2 files changed

+29
-25
lines changed

pandas/tests/io/parser/c_parser_only.py

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@
77
further arguments when parsing.
88
"""
99

10+
import sys
11+
1012
import pytest
1113
import numpy as np
1214

@@ -417,3 +419,30 @@ def test_data_after_quote(self):
417419
expected = DataFrame({'a': ['1', 'ba']})
418420

419421
tm.assert_frame_equal(result, expected)
422+
423+
@tm.capture_stderr
424+
def test_comment_whitespace_delimited(self):
425+
test_input = """\
426+
1 2
427+
2 2 3
428+
3 2 3 # 3 fields
429+
4 2 3# 3 fields
430+
5 2 # 2 fields
431+
6 2# 2 fields
432+
7 # 1 field, NaN
433+
8# 1 field, NaN
434+
9 2 3 # skipped line
435+
# comment"""
436+
df = self.read_csv(StringIO(test_input), comment='#', header=None,
437+
delimiter='\\s+', skiprows=0,
438+
error_bad_lines=False)
439+
error = sys.stderr.getvalue()
440+
# skipped lines 2, 3, 4, 9
441+
for line_num in (2, 3, 4, 9):
442+
assert 'Skipping line {}'.format(line_num) in error, error
443+
expected = DataFrame([[1, 2],
444+
[5, 2],
445+
[6, 2],
446+
[7, np.nan],
447+
[8, np.nan]])
448+
tm.assert_frame_equal(df, expected)

pandas/tests/io/parser/comment.py

Lines changed: 0 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,6 @@
55
for all of the parsers defined in parsers.py
66
"""
77

8-
import sys
9-
108
import numpy as np
119
import pandas.util.testing as tm
1210

@@ -118,26 +116,3 @@ def test_commment_first_line(self):
118116
expected = DataFrame({0: ['a', '1'], 1: ['b', '2'], 2: ['c', '3']})
119117
result = self.read_csv(StringIO(data), comment='#', header=None)
120118
tm.assert_frame_equal(result, expected)
121-
122-
@tm.capture_stderr
123-
def test_comment_whitespace_delimited(self):
124-
test_input = """\
125-
1 2
126-
2 2 3
127-
3 2 3 # 3 fields
128-
4 2 3# 3 fields
129-
5 2 # 2 fields
130-
6 2# 2 fields
131-
7 # 1 field, NaN
132-
8# 1 field, NaN
133-
9 2 3 # skipped line
134-
# comment"""
135-
df = self.read_csv(StringIO(test_input), comment='#', header=None,
136-
delimiter='\\s+', skiprows=0,
137-
error_bad_lines=False)
138-
error = sys.stderr.getvalue()
139-
# skipped lines 2, 3, 4, 9
140-
for line_num in (2, 3, 4, 9):
141-
assert 'Skipping line {}'.format(line_num) in error, error
142-
expected = DataFrame([[1, 2], [5, 2], [6, 2], [7, np.nan], [8, np.nan]])
143-
tm.assert_frame_equal(df, expected)

0 commit comments

Comments
 (0)