Also test leaf values

rapidsai · Apr 21, 2022 · 53ca862 · 53ca862
1 parent 0cf1b71
commit 53ca862
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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)