def test_itercsv_emits_data_lines(): """itercsv should yield individual lines of CSV, with headers""" expected = [ b'Hello,World\r\n', b'1,2\r\n', b'3,4\r\n' ] assert list(itercsv(['Hello', 'World'], [[1, 2], [3, 4]])) == expected
def test_itercsv_always_emits_headers(): """itercsv should always emit headers, even where there is no data""" assert list(itercsv(['Hello', 'World'], [])) == [b'Hello,World\r\n']
def test_itercsv_emits_data_lines(): """itercsv should yield individual lines of CSV, with headers""" expected = [b"Hello,World\r\n", b"1,2\r\n", b"3,4\r\n"] assert list(itercsv(["Hello", "World"], [[1, 2], [3, 4]])) == expected
def test_itercsv_always_emits_headers(): """itercsv should always emit headers, even where there is no data""" assert list(itercsv(["Hello", "World"], [])) == [b"Hello,World\r\n"]