Test validate_struct_list_type

Author: hombit

src/nested_pandas/series/utils.py

@@ -119,7 +119,7 @@ def align_struct_list_offsets(array: pa.StructArray) -> pa.StructArray:
         )
     new_array = pa.StructArray.from_arrays(
         arrays=list_arrays,
-        type=array.type,
+        fields=array.type.fields,
     )
     return new_array
 
@@ -281,7 +281,7 @@ def validate_list_struct_type(t: pa.ListType) -> None:
         raise ValueError(f"Expected a StructType as a list value type, got {t.value_type}")
 
 
-def validate_struct_list_type(t: pa.ListType) -> None:
+def validate_struct_list_type(t: pa.StructType) -> None:
     """Raise a ValueError if not a struct-list-type."""
     if not pa.types.is_struct(t):
         raise ValueError(f"Expected a StructType, got {t}")

tests/nested_pandas/series/test_series_utils.py

@@ -12,6 +12,7 @@
     transpose_list_struct_type,
     transpose_struct_list_array,
     transpose_struct_list_type,
+    validate_struct_list_type,
 )
 
 
@@ -116,6 +117,26 @@ def test_align_chunked_struct_list_offsets():
     assert output_array.equals(input_array)
 
 
+def test_validate_struct_list_type():
+    """Test validate_struct_list_type function."""
+    with pytest.raises(ValueError):
+        validate_struct_list_type(pa.float64())
+
+    with pytest.raises(ValueError):
+        validate_struct_list_type(pa.list_(pa.struct({"a": pa.int64()})))
+
+    with pytest.raises(ValueError):
+        validate_struct_list_type(pa.struct({"a": pa.float64()}))
+
+    with pytest.raises(ValueError):
+        validate_struct_list_type(pa.struct({"a": pa.list_(pa.float64()), "b": pa.float64()}))
+
+    assert (
+        validate_struct_list_type(pa.struct({"a": pa.list_(pa.float64()), "b": pa.list_(pa.float64())}))
+        is None
+    )
+
+
 def test_transpose_struct_list_type():
     """Test transpose_struct_list_type function."""
     # Raises for wrong types