From 767c205c8172d82c6d88297d0a52a39a5a273ecc Mon Sep 17 00:00:00 2001 From: Alex Dewar Date: Fri, 24 Jan 2025 09:23:42 +0000 Subject: [PATCH 1/2] Also re-export `read_to_list` and `read_to_dict` --- csvy/__init__.py | 2 ++ 1 file changed, 2 insertions(+) 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 From 2490ca8692224dcbbe5fe1c3757585cf65d3a5ad Mon Sep 17 00:00:00 2001 From: Alex Dewar Date: Fri, 24 Jan 2025 09:28:50 +0000 Subject: [PATCH 2/2] Import readers from root module in tests --- tests/test_read.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) 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": ","}