From d3d2a22145948574d1c3e7ec87c69c3375e23f9f Mon Sep 17 00:00:00 2001
From: Romain Dorgueil <romain@dorgueil.net>
Date: Wed, 28 Dec 2016 12:29:15 +0100
Subject: [PATCH] Exports CSV reader/writer to root package (#2)

---
 bonobo/__init__.py    | 4 +++-
 bonobo/io/__init__.py | 3 +++
 tests/io/test_csv.py  | 3 +--
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/bonobo/__init__.py b/bonobo/__init__.py
index 6832651..d70e30a 100644
--- a/bonobo/__init__.py
+++ b/bonobo/__init__.py
@@ -23,7 +23,7 @@ import os
 import sys
 
 from .core import *
-from .io import FileReader, FileWriter, JsonReader, JsonWriter
+from .io import CsvReader, CsvWriter, FileReader, FileWriter, JsonReader, JsonWriter
 from .util import *
 
 PY35 = (sys.version_info >= (3, 5))
@@ -36,6 +36,8 @@ with open(os.path.realpath(os.path.join(os.path.dirname(__file__), '../version.t
 
 __all__ = [
     'Bag',
+    'CsvReader',
+    'CsvWriter',
     'FileReader',
     'FileWriter',
     'Graph',
diff --git a/bonobo/io/__init__.py b/bonobo/io/__init__.py
index 36addbc..f814c31 100644
--- a/bonobo/io/__init__.py
+++ b/bonobo/io/__init__.py
@@ -2,8 +2,11 @@
 
 from .file import FileReader, FileWriter
 from .json import JsonReader, JsonWriter
+from .csv import CsvReader, CsvWriter
 
 __all__ = [
+    'CsvReader',
+    'CsvWriter',
     'FileReader',
     'FileWriter',
     'JsonReader',
diff --git a/tests/io/test_csv.py b/tests/io/test_csv.py
index 96def96..aa3ca8f 100644
--- a/tests/io/test_csv.py
+++ b/tests/io/test_csv.py
@@ -1,8 +1,7 @@
 import pytest
 
-from bonobo import Bag
+from bonobo import Bag, CsvReader, CsvWriter
 from bonobo.core.contexts import ComponentExecutionContext
-from bonobo.io.csv import CsvReader, CsvWriter
 from bonobo.util.testing import CapturingComponentExecutionContext
 from bonobo.util.tokens import BEGIN, END