From f7cd9e645d211d45f67e59c461e2b758d9d98735 Mon Sep 17 00:00:00 2001 From: Thomas Li <47963215+lithomas1@users.noreply.github.com> Date: Wed, 26 Jun 2024 09:15:50 -0700 Subject: [PATCH] cleanup utils --- python/cudf/cudf/pylibcudf_tests/common/utils.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/python/cudf/cudf/pylibcudf_tests/common/utils.py b/python/cudf/cudf/pylibcudf_tests/common/utils.py index 4bc2911c4fe..d5693750002 100644 --- a/python/cudf/cudf/pylibcudf_tests/common/utils.py +++ b/python/cudf/cudf/pylibcudf_tests/common/utils.py @@ -65,8 +65,6 @@ def assert_column_eq( if isinstance(rhs, pa.ChunkedArray): rhs = rhs.combine_chunks() - print(lhs) - print(rhs) assert lhs.equals(rhs) @@ -219,10 +217,8 @@ def sink_to_str(sink): NUMERIC_PA_TYPES + STRING_PA_TYPES + BOOL_PA_TYPES - # exclude nested list/struct cases - # since not all tests work with them yet - + LIST_PA_TYPES # [:1] - + DEFAULT_PA_STRUCT_TESTING_TYPES # [:1] + + LIST_PA_TYPES + + DEFAULT_PA_STRUCT_TESTING_TYPES ) -ALL_PA_TYPES = DEFAULT_PA_TYPES # + LIST_PA_TYPES[1:] + DEFAULT_PA_STRUCT_TESTING_TYPES[1:] +ALL_PA_TYPES = DEFAULT_PA_TYPES