Adds create_reader/create_writer tests.

This commit is contained in:
Romain Dorgueil
2018-05-21 12:30:43 +02:00
parent 68de8368b8
commit 38e0a2fe12

18
tests/test_registry.py Normal file
View File

@ -0,0 +1,18 @@
from bonobo.nodes import CsvReader, JsonReader, CsvWriter, JsonWriter
from bonobo import create_reader, create_writer
def test_create_reader():
t = create_reader('foo.csv')
assert isinstance(t, CsvReader)
t = create_reader('foo.txt', format='json')
assert isinstance(t, JsonReader)
def test_create_writer():
t = create_writer('foo.csv')
assert isinstance(t, CsvWriter)
t = create_writer('foo.txt', format='json')
assert isinstance(t, JsonWriter)