[tests] simplify assertion

This commit is contained in:
Romain Dorgueil
2017-10-28 17:24:26 +02:00
parent 7d9b547a53
commit efb31126b4

View File

@ -44,18 +44,18 @@ def test_read_csv_from_file_kwargs(tmpdir):
CsvReader(path=filename, delimiter=','),
services=services,
) as context:
context.write(BEGIN, Bag(), END)
context.step()
output = context.get_buffer()
context.write_sync(())
assert context.get_buffer() == [
{
'a': 'a foo',
'b': 'b foo',
'c': 'c foo',
},
{
'a': 'a bar',
'b': 'b bar',
'c': 'c bar',
}
]
assert len(output) == 2
assert output[0] == {
'a': 'a foo',
'b': 'b foo',
'c': 'c foo',
}
assert output[1] == {
'a': 'a bar',
'b': 'b bar',
'c': 'c bar',
}