def test_read_dict():
schema = CsvSchema(
header=['a', 'b'],
as_dict=True,
)
assert schema.read_line('1,2') == {'a': '1', 'b': '2'}
assert schema.read_line('3') == {'a': '3', 'b': None}
def test_lineno():
schema = CsvSchema()
assert schema.lineno == -1
schema.read_line('a,b')
assert schema.lineno == 0
schema.read_line('1,2')
assert schema.lineno == 1
schema.reset()
assert schema.lineno == -1
def test_read_list():
schema = CsvSchema(header=False)
assert schema.read_line('1,2') == ['1', '2']
def test_replace_header():
schema = CsvSchema(header=True, replace_header={'X': 'x'})
schema.read_line('X,y')
assert schema.header == ('x', 'y')
def test_no_header():
schema = CsvSchema(header=False)
assert schema.header is None
schema.read_line('1,2')
assert schema.header is None
def test_provide_header():
schema = CsvSchema(header=['a', 'b'])
assert schema.header == ('a', 'b')
schema.read_line('1,2')
assert schema.header == ('a', 'b')
def test_read_header():
schema = CsvSchema()
assert schema.header is None
assert schema.read_line('a,b') is None
assert schema.header == ('a', 'b')