dycw-utilities 0.109.22__tar.gz → 0.109.23__tar.gz

{dycw_utilities-0.109.22 → dycw_utilities-0.109.23}/PKG-INFO

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

{dycw_utilities-0.109.22 → dycw_utilities-0.109.23}/pyproject.toml

@@ -93,7 +93,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.109.22"
+version = "0.109.23"
 
 [project.optional-dependencies]
 test = [
@@ -336,7 +336,7 @@ zzz-test-zoneinfo = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.109.22"
+current_version = "0.109.23"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.109.22 → dycw_utilities-0.109.23}/src/tests/test_polars.py

@@ -129,6 +129,7 @@ from utilities.polars import (
     finite_ewm_mean,
     floor_datetime,
     get_data_type_or_series_time_zone,
+    get_expr_name,
     get_frequency_spectrum,
     get_series_number_of_decimals,
     insert_after,
@@ -1331,6 +1332,22 @@ class TestGetDataTypeOrSeriesTimeZone:
             _ = get_data_type_or_series_time_zone(Datetime)
 
 
+class TestGetExprName:
+    @given(n=hypothesis.strategies.integers(0, 10), name=text_ascii())
+    def test_series(self, *, n: int, name: str) -> None:
+        sr = int_range(n, eager=True)
+        expr = lit(None, dtype=Boolean).alias(name)
+        result = get_expr_name(sr, expr)
+        assert result == name
+
+    @given(n=hypothesis.strategies.integers(0, 10), name=text_ascii())
+    def test_df(self, *, n: int, name: str) -> None:
+        df = int_range(n, eager=True).to_frame()
+        expr = lit(None, dtype=Boolean).alias(name)
+        result = get_expr_name(df, expr)
+        assert result == name
+
+
 class TestGetFrequencySpectrum:
     def test_main(self) -> None:
         n = 1000

{dycw_utilities-0.109.22 → dycw_utilities-0.109.23}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.109.22"
+__version__ = "0.109.23"

{dycw_utilities-0.109.22 → dycw_utilities-0.109.23}/src/utilities/polars.py

@@ -1228,6 +1228,21 @@ class _GetDataTypeOrSeriesTimeZoneNotZonedError(GetDataTypeOrSeriesTimeZoneError
 ##
 
 
+def get_expr_name(obj: Series | DataFrame, expr: Expr, /) -> str:
+    """Get the name of an expression."""
+    match obj:
+        case Series() as series:
+            return get_expr_name(series.to_frame(), expr)
+        case DataFrame() as df:
+            selected = df.select(expr)
+            return one(selected.columns)
+        case _ as never:
+            assert_never(never)
+
+
+##
+
+
 def get_frequency_spectrum(series: Series, /, *, d: int = 1) -> DataFrame:
     """Get the frequency spectrum."""
     import utilities.numpy
@@ -1970,6 +1985,7 @@ __all__ = [
     "finite_ewm_mean",
     "floor_datetime",
     "get_data_type_or_series_time_zone",
+    "get_expr_name",
     "get_frequency_spectrum",
     "get_series_number_of_decimals",
     "insert_after",