dycw-utilities 0.155.2__tar.gz → 0.155.3__tar.gz

{dycw_utilities-0.155.2 → dycw_utilities-0.155.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.155.2
+Version: 0.155.3
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

{dycw_utilities-0.155.2 → dycw_utilities-0.155.3}/pyproject.toml

@@ -227,7 +227,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.155.2"
+version = "0.155.3"
 
 [project.entry-points.pytest11]
 pytest-randomly = "utilities.pytest_plugins.pytest_randomly"
@@ -259,7 +259,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.155.2"
+current_version = "0.155.3"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.155.2 → dycw_utilities-0.155.3}/src/tests/test_polars.py

@@ -1041,6 +1041,23 @@ class TestDataClassToDataFrame:
         df = dataclass_to_dataframe(objs)
         check_polars_dataframe(df, height=len(objs), schema_list={"x": Duration})
 
+    @given(data=data())
+    def test_multiple_periods(self, *, data: DataObject) -> None:
+        @dataclass(kw_only=True, slots=True)
+        class Example:
+            x: DatePeriod
+            y: DatePeriod
+
+        objs = data.draw(
+            lists(builds(Example, x=date_periods(), y=date_periods()), min_size=1)
+        )
+        df = dataclass_to_dataframe(objs, globalns=globals())
+        check_polars_dataframe(
+            df,
+            height=len(objs),
+            schema_list={"x": DatePeriodDType, "y": DatePeriodDType},
+        )
+
     @given(data=data())
     def test_nested(self, *, data: DataObject) -> None:
         @dataclass(kw_only=True, slots=True)

{dycw_utilities-0.155.2 → dycw_utilities-0.155.3}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.155.2"
+__version__ = "0.155.3"

{dycw_utilities-0.155.2 → dycw_utilities-0.155.3}/src/utilities/polars.py

@@ -1066,7 +1066,10 @@ def _dataclass_to_dataframe_cast(series: Series, /) -> Series:
         end = _dataclass_to_dataframe_cast(
             series.map_elements(lambda x: x["end"], return_dtype=Object)
         ).alias("end")
-        return concat_series(start, end).select(x=struct(start=start, end=end))["x"]
+        name = series.name
+        return concat_series(start, end).select(
+            struct(start=start, end=end).alias(name)
+        )[name]
     raise NotImplementedError(series)  # pragma: no cover