Minor test change.

This commit is contained in:
Romain Dorgueil
2017-10-29 12:13:52 +01:00
parent f9c809f626
commit cac6920040

View File

@ -20,10 +20,10 @@ def test_write_json_ioformat_arg0(tmpdir):
@pytest.mark.parametrize('add_kwargs', ( @pytest.mark.parametrize('add_kwargs', (
{}, {},
{ {
'ioformat': settings.IOFORMAT_KWARGS, 'ioformat': settings.IOFORMAT_KWARGS,
}, },
)) ))
def test_write_json_kwargs(tmpdir, add_kwargs): def test_write_json_kwargs(tmpdir, add_kwargs):
fs, filename, services = json_tester.get_services_for_writer(tmpdir) fs, filename, services = json_tester.get_services_for_writer(tmpdir)
@ -55,8 +55,10 @@ def test_write_stream_json(tmpdir):
with BufferingNodeExecutionContext(LdjsonWriter(filename), with BufferingNodeExecutionContext(LdjsonWriter(filename),
services=services) as context: services=services) as context:
context.write_sync({'foo': 'bar'}) context.write_sync(
context.write_sync({'baz': 'boz'}) {'foo': 'bar'},
{'baz': 'boz'},
)
expected = '''{"foo": "bar"}\n{"baz": "boz"}\n''' expected = '''{"foo": "bar"}\n{"baz": "boz"}\n'''
with fs.open(filename) as fin: with fs.open(filename) as fin: