From 38e0a2fe12f225a994c5b4b2e6f486280166cd94 Mon Sep 17 00:00:00 2001 From: Romain Dorgueil Date: Mon, 21 May 2018 12:30:43 +0200 Subject: [PATCH] Adds create_reader/create_writer tests. --- tests/test_registry.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 tests/test_registry.py diff --git a/tests/test_registry.py b/tests/test_registry.py new file mode 100644 index 0000000..dfff5eb --- /dev/null +++ b/tests/test_registry.py @@ -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)