From 53ca86203ba0629d430eaa0ff4852b3181f1c855 Mon Sep 17 00:00:00 2001 From: Ashwin Srinath Date: Thu, 21 Apr 2022 14:43:27 -0400 Subject: [PATCH] Also test leaf values --- python/cudf/cudf/tests/test_list.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/cudf/cudf/tests/test_list.py b/python/cudf/cudf/tests/test_list.py index 4b7a1e03511..c21e1a0f61f 100644 --- a/python/cudf/cudf/tests/test_list.py +++ b/python/cudf/cudf/tests/test_list.py @@ -765,7 +765,9 @@ def test_list_astype(): s = cudf.Series([[1, 2], [3, 4]]) s2 = s.list.astype("float64") assert s2.dtype == cudf.ListDtype("float64") + assert_eq(s.list.leaves.astype("float64"), s2.list.leaves) s = cudf.Series([[[1, 2], [3]], [[5, 6], None]]) s2 = s.list.astype("string") assert s2.dtype == cudf.ListDtype(cudf.ListDtype("string")) + assert_eq(s.list.leaves.astype("string"), s2.list.leaves)