diff --git a/csvy/__init__.py b/csvy/__init__.py index 52d6d40..ff4de16 100644 --- a/csvy/__init__.py +++ b/csvy/__init__.py @@ -6,6 +6,8 @@ read_metadata, read_to_array, read_to_dataframe, + read_to_dict, + read_to_list, read_to_polars, ) from .writers import Writer, write, write_header # noqa: F401 diff --git a/tests/test_read.py b/tests/test_read.py index 406e75d..21cf9af 100644 --- a/tests/test_read.py +++ b/tests/test_read.py @@ -79,7 +79,7 @@ def test_read_to_dataframe(data_path): """Test the read_to_dataframe function.""" import pandas as pd - from csvy.readers import read_to_dataframe + from csvy import read_to_dataframe data, header = read_to_dataframe(data_path) assert isinstance(data, pd.DataFrame) @@ -100,7 +100,7 @@ def test_read_to_polars(data_path): import polars as pl from polars.testing import assert_frame_equal - from csvy.readers import read_to_polars + from csvy import read_to_polars lazy_data, header = read_to_polars(data_path) assert isinstance(lazy_data, pl.LazyFrame) @@ -124,7 +124,7 @@ def test_read_to_polars(data_path): def test_read_to_list(array_data_path): """Test the read_to_list function.""" - from csvy.readers import read_to_list + from csvy import read_to_list data, header = read_to_list(array_data_path, csv_options={"delimiter": ","}) assert isinstance(data, list) @@ -136,7 +136,7 @@ def test_read_to_list(array_data_path): def test_read_to_dict_with_default_column_names(array_data_path): """Test the read_to_list function.""" - from csvy.readers import read_to_dict + from csvy import read_to_dict data, header = read_to_dict(array_data_path, csv_options={"delimiter": ","}) @@ -149,7 +149,7 @@ def test_read_to_dict_with_default_column_names(array_data_path): def test_read_to_dict_with_custom_column_names(array_data_path): """Test the read_to_list function.""" - from csvy.readers import read_to_dict + from csvy import read_to_dict column_names = ["A", "B", "C", "D"] data, header = read_to_dict( @@ -165,7 +165,7 @@ def test_read_to_dict_with_custom_column_names(array_data_path): def test_read_to_dict_with_row_based_column_names(data_path): """Test the read_to_list function.""" - from csvy.readers import read_to_dict + from csvy import read_to_dict data, header = read_to_dict( data_path, column_names=0, csv_options={"delimiter": ","}