diff --git a/python/arrow_odbc/__init__.py b/python/arrow_odbc/__init__.py index bb8d657..152f5b2 100644 --- a/python/arrow_odbc/__init__.py +++ b/python/arrow_odbc/__init__.py @@ -1,6 +1,6 @@ from .error import Error from .reader import BatchReader, read_arrow_batches_from_odbc -from .writer import BatchWriter, insert_into_table +from .writer import BatchWriter, insert_into_table, from_table_to_db __all__ = [ "BatchReader", diff --git a/tests/test_arrow_odbc.py b/tests/test_arrow_odbc.py index 1fa16aa..a9da96e 100644 --- a/tests/test_arrow_odbc.py +++ b/tests/test_arrow_odbc.py @@ -11,8 +11,12 @@ from pytest import raises -from arrow_odbc import read_arrow_batches_from_odbc, Error -from arrow_odbc.writer import insert_into_table, from_table_to_db +from arrow_odbc import ( + insert_into_table, + from_table_to_db, + read_arrow_batches_from_odbc, + Error, +) MSSQL = "Driver={ODBC Driver 17 for SQL Server};Server=localhost;UID=SA;PWD=My@Test@Password1;"