dycw-utilities 0.153.10__tar.gz → 0.153.12__tar.gz

{dycw_utilities-0.153.10 → dycw_utilities-0.153.12}/PKG-INFO

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

{dycw_utilities-0.153.10 → dycw_utilities-0.153.12}/pyproject.toml

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

{dycw_utilities-0.153.10 → dycw_utilities-0.153.12}/src/tests/test_polars.py

@@ -2420,11 +2420,21 @@ class TestReplaceTimeZone:
 
 
 class TestRoundToFloat:
-    @mark.parametrize(("x", "y", "expected"), tests.test_math.TestRoundToFloat.cases)
-    def test_main(self, *, x: float, y: float, expected: float) -> None:
+    @mark.parametrize(("x", "y", "exp_value"), tests.test_math.TestRoundToFloat.cases)
+    def test_main(self, *, x: float, y: float, exp_value: float) -> None:
         series = Series(name="x", values=[x], dtype=Float64)
-        result = round_to_float(series, y).item()
-        assert result == expected
+        result = round_to_float(series, y)
+        expected = Series(name="x", values=[exp_value], dtype=Float64)
+        assert_series_equal(result, expected, check_exact=True)
+
+    def test_dataframe_name(self) -> None:
+        df = (
+            Series(name="x", values=[1.234], dtype=Float64)
+            .to_frame()
+            .with_columns(round_to_float("x", 0.1))
+        )
+        expected = Series(name="x", values=[1.2], dtype=Float64).to_frame()
+        assert_frame_equal(df, expected)
 
 
 class TestSerializeAndDeserializeDataFrame:

{dycw_utilities-0.153.10 → dycw_utilities-0.153.12}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.153.10"
+__version__ = "0.153.12"

{dycw_utilities-0.153.10 → dycw_utilities-0.153.12}/src/utilities/polars.py

@@ -2383,7 +2383,7 @@ def round_to_float(
 ) -> ExprOrSeries:
     """Round a column to the nearest multiple of another float."""
     x = ensure_expr_or_series(x)
-    return y * (x / y).round(mode=mode)
+    return (x / y).round(mode=mode) * y
 
 
 ##

{dycw_utilities-0.153.10 → dycw_utilities-0.153.12}/src/utilities/sqlalchemy.py

@@ -18,6 +18,7 @@ from itertools import chain
 from math import floor
 from operator import ge, le
 from re import search
+from socket import gaierror
 from typing import (
     TYPE_CHECKING,
     Any,
@@ -127,7 +128,7 @@ def check_connect(engine: Engine, /) -> bool:
     try:
         with engine.connect() as conn:
             return bool(conn.execute(_SELECT).scalar_one())
-    except (OperationalError, ProgrammingError):
+    except (gaierror, OperationalError, ProgrammingError):
         return False
 
 
@@ -138,7 +139,7 @@ async def check_connect_async(
     try:
         async with timeout_td(timeout), engine.connect() as conn:
             return bool((await conn.execute(_SELECT)).scalar_one())
-    except (OperationalError, ProgrammingError, TimeoutError):
+    except (gaierror, OperationalError, ProgrammingError, TimeoutError):
         return False