Fix size passed to column constructors in _with_type_metadata (#8539)

Authors:
  - Ashwin Srinath (https://github.com/shwina)

Approvers:
  - Charles Blackmon-Luca (https://github.com/charlesbluca)
  - Marlene  (https://github.com/marlenezw)

URL: #8539

python/cudf/cudf/core/column/lists.py

@@ -246,7 +246,7 @@ def _with_type_metadata(
                     dtype.element_type
                 ),
                 mask=self.base_mask,
-                size=self.base_size,
+                size=self.size,
                 offset=self.offset,
                 null_count=self.null_count,
             )

python/cudf/cudf/core/column/struct.py

@@ -99,7 +99,7 @@ def _rename_fields(self, names):
         )
         return StructColumn(
             data=None,
-            size=self.base_size,
+            size=self.size,
             dtype=dtype,
             mask=self.base_mask,
             offset=self.offset,
@@ -122,7 +122,7 @@ def _with_type_metadata(self: StructColumn, dtype: Dtype) -> StructColumn:
                     for i, f in enumerate(dtype.fields.keys())
                 ),
                 mask=self.base_mask,
-                size=self.base_size,
+                size=self.size,
                 offset=self.offset,
                 null_count=self.null_count,
             )

python/cudf/cudf/tests/test_indexing.py

@@ -1401,6 +1401,10 @@ def test_iloc_with_lists(data, key):
     gsr = cudf.Series(data)
     assert_eq(psr.iloc[key], gsr.iloc[key])
 
+    pdf = pd.DataFrame({"a": data, "b": data})
+    gdf = cudf.DataFrame({"a": data, "b": data})
+    assert_eq(pdf.iloc[key], gdf.iloc[key])
+
 
 @pytest.mark.parametrize("key", [5, -10, "0", "a", np.array(5), np.array("a")])
 def test_loc_bad_key_type(key):