Skip to content

Commit af60064

Browse files
committed
fix test
1 parent f2ff9fa commit af60064

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

unit_tests/sources/declarative/decoders/test_composite_decoder.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,12 +51,14 @@ def generate_csv(
5151
{"id": "2", "name": "Alice", "age": "34"},
5252
{"id": "3", "name": "Bob", "age": "25"},
5353
]
54+
fieldnames = ["id", "name", "age"]
5455
if add_empty_strings:
5556
for row in data:
5657
row["gender"] = ""
58+
fieldnames.append("gender")
5759

5860
output = StringIO()
59-
writer = csv.DictWriter(output, fieldnames=["id", "name", "age"], delimiter=delimiter)
61+
writer = csv.DictWriter(output, fieldnames=fieldnames, delimiter=delimiter)
6062
writer.writeheader()
6163
for row in data:
6264
writer.writerow(row)
@@ -282,7 +284,7 @@ def test_composite_raw_decoder_parse_empty_strings(requests_mock, set_empty_cell
282284
{"id": "3", "name": "Bob", "age": "25"},
283285
]
284286
for expected_record in expected_data:
285-
expected_recordp["gender"] = None if set_empty_cell_to_none else ""
287+
expected_record["gender"] = None if set_empty_cell_to_none else ""
286288

287289
parsed_records = list(composite_raw_decoder.decode(response))
288290
assert parsed_records == expected_data

0 commit comments

Comments
 (0)