Skip to content

Commit f207fd8

Browse files
committed
add unit test
1 parent 144d937 commit f207fd8

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

unit_tests/sources/file_based/file_types/test_csv_parser.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -658,6 +658,20 @@ def test_read_data_with_encoding_error(self) -> None:
658658
assert "encoding" in ate.value.message
659659
assert self._csv_reader._get_headers.called
660660

661+
def test_read_data_with_whitespace_in_header(self) -> None:
662+
self._stream_reader.open_file.return_value = (
663+
CsvFileBuilder()
664+
.with_data(
665+
[
666+
"header1 ,\theader2",
667+
"1,2",
668+
]
669+
)
670+
.build()
671+
)
672+
data_generator = self._read_data()
673+
assert list(data_generator) == [{"header1": "1", "header2": "2"}]
674+
661675
def _read_data(self) -> Generator[Dict[str, str], None, None]:
662676
data_generator = self._csv_reader.read_data(
663677
self._config,

0 commit comments

Comments
 (0)