From e26ef5007c2c69576d948e3e6fab54b210ae1fea Mon Sep 17 00:00:00 2001 From: Binayak Date: Mon, 25 Nov 2024 13:49:18 +0800 Subject: [PATCH] update test --- tests/io/test_pyarrow_stats.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/io/test_pyarrow_stats.py b/tests/io/test_pyarrow_stats.py index 241dd800ac..788891711e 100644 --- a/tests/io/test_pyarrow_stats.py +++ b/tests/io/test_pyarrow_stats.py @@ -711,12 +711,13 @@ def test_read_missing_statistics() -> None: # expect only "strings" column values to be reflected in the # upper_bound, lower_bound and null_value_counts props of datafile + string_col_idx = 1 assert len(datafile.lower_bounds) == 1 - assert datafile.lower_bounds[1].decode() == "aaaaaaaaaaaaaaaa" + assert datafile.lower_bounds[string_col_idx].decode() == "aaaaaaaaaaaaaaaa" assert len(datafile.upper_bounds) == 1 - assert datafile.upper_bounds[1].decode() == "zzzzzzzzzzzzzzz{" + assert datafile.upper_bounds[string_col_idx].decode() == "zzzzzzzzzzzzzzz{" assert len(datafile.null_value_counts) == 1 - assert datafile.null_value_counts[1] == 1 + assert datafile.null_value_counts[string_col_idx] == 1 # This is commented out for now because write_to_dataset drops the partition