dycw-utilities 0.112.9__tar.gz → 0.112.10__tar.gz

{dycw_utilities-0.112.9 → dycw_utilities-0.112.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.112.9
+Version: 0.112.10
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12
@@ -24,7 +24,7 @@ Provides-Extra: zzz-test-altair
 Requires-Dist: altair<5.6,>=5.5.0; extra == 'zzz-test-altair'
 Requires-Dist: atomicwrites<1.5,>=1.4.1; extra == 'zzz-test-altair'
 Requires-Dist: img2pdf<0.7,>=0.6.0; extra == 'zzz-test-altair'
-Requires-Dist: polars-lts-cpu<1.29,>=1.28.1; extra == 'zzz-test-altair'
+Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-altair'
 Requires-Dist: vl-convert-python<1.8,>=1.7.0; extra == 'zzz-test-altair'
 Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-altair'
 Provides-Extra: zzz-test-astor
@@ -48,7 +48,7 @@ Provides-Extra: zzz-test-cvxpy
 Requires-Dist: cvxpy<1.7,>=1.6.5; extra == 'zzz-test-cvxpy'
 Provides-Extra: zzz-test-dataclasses
 Requires-Dist: orjson<3.11,>=3.10.15; extra == 'zzz-test-dataclasses'
-Requires-Dist: polars-lts-cpu<1.29,>=1.28.1; extra == 'zzz-test-dataclasses'
+Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-dataclasses'
 Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-dataclasses'
 Provides-Extra: zzz-test-datetime
 Requires-Dist: tzlocal<5.4,>=5.3.1; extra == 'zzz-test-datetime'
@@ -70,7 +70,7 @@ Provides-Extra: zzz-test-getpass
 Provides-Extra: zzz-test-git
 Provides-Extra: zzz-test-hashlib
 Requires-Dist: orjson<3.11,>=3.10.15; extra == 'zzz-test-hashlib'
-Requires-Dist: polars-lts-cpu<1.29,>=1.28.1; extra == 'zzz-test-hashlib'
+Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-hashlib'
 Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-hashlib'
 Provides-Extra: zzz-test-http
 Requires-Dist: atomicwrites<1.5,>=1.4.1; extra == 'zzz-test-http'
@@ -92,12 +92,12 @@ Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-hypothesis'
 Provides-Extra: zzz-test-ipython
 Requires-Dist: ipython<9.1,>=9.0.1; extra == 'zzz-test-ipython'
 Provides-Extra: zzz-test-iterables
-Requires-Dist: polars-lts-cpu<1.29,>=1.28.1; extra == 'zzz-test-iterables'
+Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-iterables'
 Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-iterables'
 Provides-Extra: zzz-test-jupyter
 Requires-Dist: jupyterlab<4.3,>=4.2.0; extra == 'zzz-test-jupyter'
 Requires-Dist: pandas<2.3,>=2.2.2; extra == 'zzz-test-jupyter'
-Requires-Dist: polars-lts-cpu<1.29,>=1.28.1; extra == 'zzz-test-jupyter'
+Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-jupyter'
 Provides-Extra: zzz-test-logging
 Requires-Dist: atomicwrites<1.5,>=1.4.1; extra == 'zzz-test-logging'
 Requires-Dist: coloredlogs<15.1,>=15.0.1; extra == 'zzz-test-logging'
@@ -121,13 +121,13 @@ Requires-Dist: more-itertools<10.8,>=10.7.0; extra == 'zzz-test-more-itertools'
 Provides-Extra: zzz-test-numpy
 Requires-Dist: numpy<2.3,>=2.2.5; extra == 'zzz-test-numpy'
 Provides-Extra: zzz-test-operator
-Requires-Dist: polars-lts-cpu<1.29,>=1.28.1; extra == 'zzz-test-operator'
+Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-operator'
 Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-operator'
 Provides-Extra: zzz-test-optuna
 Requires-Dist: optuna<4.4,>=4.3.0; extra == 'zzz-test-optuna'
 Provides-Extra: zzz-test-orjson
 Requires-Dist: orjson<3.11,>=3.10.15; extra == 'zzz-test-orjson'
-Requires-Dist: polars-lts-cpu<1.29,>=1.28.1; extra == 'zzz-test-orjson'
+Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-orjson'
 Requires-Dist: rich<14.1,>=14.0.0; extra == 'zzz-test-orjson'
 Requires-Dist: tzlocal<5.4,>=5.3.1; extra == 'zzz-test-orjson'
 Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-orjson'
@@ -138,7 +138,7 @@ Requires-Dist: atomicwrites<1.5,>=1.4.1; extra == 'zzz-test-pickle'
 Provides-Extra: zzz-test-platform
 Provides-Extra: zzz-test-polars
 Requires-Dist: dacite<1.10,>=1.9.2; extra == 'zzz-test-polars'
-Requires-Dist: polars-lts-cpu<1.29,>=1.28.1; extra == 'zzz-test-polars'
+Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-polars'
 Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-polars'
 Provides-Extra: zzz-test-pqdm
 Requires-Dist: pqdm<0.3,>=0.2.0; extra == 'zzz-test-pqdm'
@@ -164,7 +164,7 @@ Provides-Extra: zzz-test-random
 Provides-Extra: zzz-test-re
 Provides-Extra: zzz-test-redis
 Requires-Dist: orjson<3.11,>=3.10.15; extra == 'zzz-test-redis'
-Requires-Dist: polars-lts-cpu<1.29,>=1.28.1; extra == 'zzz-test-redis'
+Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-redis'
 Requires-Dist: redis<5.3,>=5.2.1; extra == 'zzz-test-redis'
 Requires-Dist: rich<14.1,>=14.0.0; extra == 'zzz-test-redis'
 Requires-Dist: tenacity<9.0,>=8.5.0; extra == 'zzz-test-redis'
@@ -192,7 +192,7 @@ Requires-Dist: aiosqlite<0.22,>=0.21.0; extra == 'zzz-test-sqlalchemy-polars'
 Requires-Dist: asyncpg<0.31,>=0.30.0; extra == 'zzz-test-sqlalchemy-polars'
 Requires-Dist: greenlet<3.3,>=3.2.0; extra == 'zzz-test-sqlalchemy-polars'
 Requires-Dist: nest-asyncio<1.7,>=1.6.0; extra == 'zzz-test-sqlalchemy-polars'
-Requires-Dist: polars-lts-cpu<1.29,>=1.28.1; extra == 'zzz-test-sqlalchemy-polars'
+Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-sqlalchemy-polars'
 Requires-Dist: sqlalchemy<2.1,>=2.0.40; extra == 'zzz-test-sqlalchemy-polars'
 Requires-Dist: tenacity<9.0,>=8.5.0; extra == 'zzz-test-sqlalchemy-polars'
 Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-sqlalchemy-polars'
@@ -217,7 +217,7 @@ Requires-Dist: tzlocal<5.4,>=5.3.1; extra == 'zzz-test-traceback'
 Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-traceback'
 Provides-Extra: zzz-test-types
 Provides-Extra: zzz-test-typing
-Requires-Dist: polars-lts-cpu<1.29,>=1.28.1; extra == 'zzz-test-typing'
+Requires-Dist: polars-lts-cpu<1.30,>=1.29.0; extra == 'zzz-test-typing'
 Requires-Dist: whenever<0.8,>=0.7.3; extra == 'zzz-test-typing'
 Provides-Extra: zzz-test-tzlocal
 Requires-Dist: tzlocal<5.4,>=5.3.1; extra == 'zzz-test-tzlocal'

{dycw_utilities-0.112.9 → dycw_utilities-0.112.10}/pyproject.toml

@@ -40,7 +40,7 @@ dev = [
   "optuna >= 4.3.0, < 4.4",
   "orjson >= 3.10.18, < 3.11",
   "pathvalidate >= 3.2.3, < 3.3",
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
   "polars-ols >= 0.3.5, < 0.4",
   "pqdm >= 0.2.0, < 0.3",
   "psycopg2-binary >= 2.9.10, < 2.10",  # for sqlalchemy
@@ -93,7 +93,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.112.9"
+version = "0.112.10"
 
 [project.optional-dependencies]
 test = [
@@ -116,7 +116,7 @@ zzz-test-altair = [
   "altair >= 5.5.0, < 5.6",
   "atomicwrites >= 1.4.1, < 1.5",
   "img2pdf >= 0.6.0, < 0.7",
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
   "vl-convert-python >= 1.7.0, < 1.8",
   "whenever >= 0.7.3, < 0.8",
 ]
@@ -138,7 +138,7 @@ zzz-test-cryptography = ["cryptography >= 44.0.2, < 44.1"]
 zzz-test-cvxpy = ["cvxpy >= 1.6.5, < 1.7"]
 zzz-test-dataclasses = [
   "orjson >= 3.10.15, < 3.11",
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
   "whenever >= 0.7.3, < 0.8",
 ]
 zzz-test-datetime = [
@@ -162,7 +162,7 @@ zzz-test-functools = []
 zzz-test-getpass = []
 zzz-test-git = []
 zzz-test-hashlib = [
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
   "orjson >= 3.10.15, < 3.11",
   "whenever >= 0.7.3, < 0.8",
 ]
@@ -187,13 +187,13 @@ zzz-test-hypothesis = [
 ]
 zzz-test-ipython = ["ipython >= 9.0.1, < 9.1"]
 zzz-test-iterables = [
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
   "whenever >= 0.7.3, < 0.8",
 ]
 zzz-test-jupyter = [
   "jupyterlab >= 4.2.0, < 4.3",
   "pandas >= 2.2.2, < 2.3",
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
 ]
 zzz-test-logging = [
   "atomicwrites >= 1.4.1, < 1.5",
@@ -215,13 +215,13 @@ zzz-test-modules = []
 zzz-test-more-itertools = ["more-itertools >= 10.7.0, < 10.8"]
 zzz-test-numpy = ["numpy >= 2.2.5, < 2.3"]
 zzz-test-operator = [
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
   "whenever >= 0.7.3, < 0.8",
 ]
 zzz-test-optuna = ["optuna >= 4.3.0, < 4.4"]
 zzz-test-orjson = [
   "orjson >= 3.10.15, < 3.11",
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
   "rich >= 14.0.0, < 14.1",
   "tzlocal >= 5.3.1, < 5.4",
   "whenever >= 0.7.3, < 0.8",
@@ -232,7 +232,7 @@ zzz-test-pickle = ["atomicwrites >= 1.4.1, < 1.5"]
 zzz-test-platform = []
 zzz-test-polars = [
   "dacite >= 1.9.2, < 1.10",
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
   "whenever >= 0.7.3, < 0.8",
 ]
 zzz-test-pqdm = ["pqdm >= 0.2.0, < 0.3"]
@@ -260,7 +260,7 @@ zzz-test-random = []
 zzz-test-re = []
 zzz-test-redis = [
   "orjson >= 3.10.15, < 3.11",
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
   "redis >= 5.2.1, < 5.3",
   "rich >= 14.0.0, < 14.1",
   "tenacity >= 8.5.0, < 9.0",  # limited by luigi
@@ -289,7 +289,7 @@ zzz-test-sqlalchemy-polars = [
   "asyncpg >= 0.30.0, < 0.31",  # for sqlalchemy async
   "greenlet >= 3.2.0, < 3.3",  # for sqlalchemy async
   "nest-asyncio >= 1.6.0, < 1.7",  # for sqlalchemy async
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
   "sqlalchemy >= 2.0.40, < 2.1",
   "tenacity >= 8.5.0, < 9.0",  # limited by luigi
   "whenever >= 0.7.3, < 0.8",
@@ -317,7 +317,7 @@ zzz-test-traceback = [
 ]
 zzz-test-types = []
 zzz-test-typing = [
-  "polars-lts-cpu >= 1.28.1, < 1.29",
+  "polars-lts-cpu >= 1.29.0, < 1.30",
   "whenever >= 0.7.3, < 0.8",
 ]
 zzz-test-tzlocal = ["tzlocal >= 5.3.1, < 5.4"]
@@ -336,7 +336,7 @@ zzz-test-zoneinfo = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.112.9"
+current_version = "0.112.10"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.112.9 → dycw_utilities-0.112.10}/src/tests/test_numpy.py

@@ -5,7 +5,8 @@ from typing import TYPE_CHECKING, Any, Literal
 
 import numpy as np
 from hypothesis import assume, given
-from hypothesis.strategies import DataObject, data, floats, integers, none
+from hypothesis.extra.numpy import array_shapes
+from hypothesis.strategies import DataObject, data, floats, integers, none, sampled_from
 from numpy import (
     arange,
     array,
@@ -36,6 +37,7 @@ from utilities.numpy import (
     FlatN0MultipleError,
     NDArrayF,
     NDArrayI,
+    ShapeLike,
     ShiftError,
     SigmoidError,
     _BoxCarLocationsError,
@@ -44,6 +46,7 @@ from utilities.numpy import (
     adjust_frequencies,
     array_indexer,
     as_int,
+    bernoulli,
     boxcar,
     discretize,
     fillna,
@@ -285,6 +288,15 @@ class TestDiscretize:
         assert_equal(result, expected)
 
 
+class TestBernoulli:
+    @given(case=sampled_from([(0.0, False), (1.0, True)]), size=array_shapes())
+    def test_main(self, *, case: tuple[float, bool], size: ShapeLike) -> None:
+        true, expected = case
+        result = bernoulli(true=true, size=size)
+        assert (result == expected).all()
+        assert result.shape == size
+
+
 class TestBoxCar:
     @given(
         locs=pairs(floats(-10.0, 10.0), sorted=True),

{dycw_utilities-0.112.9 → dycw_utilities-0.112.10}/src/tests/test_polars.py

@@ -50,6 +50,10 @@ from polars import (
     int_range,
     lit,
 )
+from polars._typing import (
+    IntoExprColumn,  # pyright: ignore[reportPrivateImportUsage]
+    SchemaDict,  # pyright: ignore[reportPrivateImportUsage]
+)
 from polars.testing import assert_frame_equal, assert_series_equal
 from pytest import raises
 
@@ -67,6 +71,7 @@ from utilities.numpy import DEFAULT_RNG
 from utilities.pathlib import PWD
 from utilities.polars import (
     AppendDataClassError,
+    BooleanValueCountsError,
     ColumnsToDictError,
     DatetimeHongKong,
     DatetimeTokyo,
@@ -113,8 +118,11 @@ from utilities.polars import (
     adjust_frequencies,
     append_dataclass,
     are_frames_equal,
+    bernoulli,
+    boolean_value_counts,
     ceil_datetime,
     check_polars_dataframe,
+    choice,
     collect_series,
     columns_to_dict,
     concat_series,
@@ -126,6 +134,7 @@ from utilities.polars import (
     drop_null_struct_series,
     ensure_data_type,
     ensure_expr_or_series,
+    ensure_expr_or_series_many,
     finite_ewm_mean,
     floor_datetime,
     get_data_type_or_series_time_zone,
@@ -160,11 +169,12 @@ from utilities.tzdata import HongKong, Tokyo, USCentral, USEastern
 from utilities.zoneinfo import UTC, get_time_zone_name
 
 if TYPE_CHECKING:
-    from collections.abc import Callable, Mapping, Sequence
+    from collections.abc import Callable, Iterable, Mapping, Sequence
     from zoneinfo import ZoneInfo
 
     from polars._typing import (
         IntoExprColumn,  # pyright: ignore[reportPrivateImportUsage]
+        PolarsDataType,  # pyright: ignore[reportPrivateImportUsage]
         SchemaDict,  # pyright: ignore[reportPrivateImportUsage]
     )
     from polars.datatypes import DataTypeClass
@@ -342,6 +352,87 @@ class TestAreFramesEqual:
         assert result is expected
 
 
+class TestBernoulli:
+    @given(length=hypothesis.strategies.integers(0, 10))
+    def test_int(self, *, length: int) -> None:
+        series = bernoulli(length)
+        self._assert(series, length)
+
+    @given(length=hypothesis.strategies.integers(0, 10))
+    def test_series(self, *, length: int) -> None:
+        orig = int_range(end=length, eager=True)
+        series = bernoulli(orig)
+        self._assert(series, length)
+
+    @given(length=hypothesis.strategies.integers(0, 10))
+    def test_dataframe(self, *, length: int) -> None:
+        df = int_range(end=length, eager=True).to_frame()
+        series = bernoulli(df)
+        self._assert(series, length)
+
+    def _assert(self, series: Series, length: int, /) -> None:
+        assert series.dtype == Boolean
+        assert series.len() == length
+        assert series.is_not_null().all()
+
+
+class TestBooleanValueCounts:
+    df: ClassVar[DataFrame] = DataFrame(
+        data=[
+            (False, False),
+            (True, None),
+            (True, True),
+            (True, None),
+            (False, True),
+            (None, True),
+            (False, False),
+            (False, True),
+            (False, False),
+            (None, True),
+        ],
+        schema={"x": Boolean, "y": Boolean},
+        orient="row",
+    )
+    schema: ClassVar[SchemaDict] = {
+        "name": String,
+        "true": UInt32,
+        "false": UInt32,
+        "null": UInt32,
+        "total": UInt32,
+        "true (%)": Float64,
+        "false (%)": Float64,
+        "null (%)": Float64,
+    }
+
+    def test_series(self) -> None:
+        result = boolean_value_counts(self.df["x"], "x")
+        check_polars_dataframe(result, height=1, schema_list=self.schema)
+
+    def test_dataframe(self) -> None:
+        result = boolean_value_counts(
+            self.df,
+            "x",
+            "y",
+            (col("x") & col("y")).alias("x_and_y"),
+            x_or_y=col("x") | col("y"),
+        )
+        check_polars_dataframe(result, height=4, schema_list=self.schema)
+
+    def test_empty(self) -> None:
+        result = boolean_value_counts(self.df[:0], "x")
+        check_polars_dataframe(result, height=1, schema_list=self.schema)
+        for column in ["true", "false", "null", "total"]:
+            assert (result[column] == 0).all()
+        for column in ["true (%)", "false (%)", "null (%)"]:
+            assert result[column].is_nan().all()
+
+    def test_error(self) -> None:
+        with raises(
+            BooleanValueCountsError, match="Column 'z' must be Boolean; got Int64"
+        ):
+            _ = boolean_value_counts(self.df, col("x").cast(Int64).alias("z"))
+
+
 class TestCeilDateTime:
     start: ClassVar[dt.datetime] = dt.datetime(2000, 1, 1, 0, 0, tzinfo=UTC)
     end: ClassVar[dt.datetime] = dt.datetime(2000, 1, 1, 0, 3, tzinfo=UTC)
@@ -634,6 +725,47 @@ class TestCheckPolarsDataFrameSchemaSubset:
             _check_polars_dataframe_schema_subset(df, schema_inc)
 
 
+class TestChoice:
+    @given(length=hypothesis.strategies.integers(0, 10))
+    def test_int_with_bool(self, *, length: int) -> None:
+        elements = [True, False, None]
+        series = choice(length, elements, dtype=Boolean)
+        self._assert(series, length, elements, dtype=Boolean)
+
+    @given(length=hypothesis.strategies.integers(0, 10))
+    def test_int_with_str(self, *, length: int) -> None:
+        elements = ["A", "B", "C"]
+        series = choice(length, elements, dtype=String)
+        self._assert(series, length, elements, dtype=String)
+
+    @given(length=hypothesis.strategies.integers(0, 10))
+    def test_series(self, *, length: int) -> None:
+        orig = int_range(end=length, eager=True)
+        elements = ["A", "B", "C"]
+        series = choice(orig, elements, dtype=String)
+        self._assert(series, length, elements, dtype=String)
+
+    @given(length=hypothesis.strategies.integers(0, 10))
+    def test_dataframe(self, *, length: int) -> None:
+        df = int_range(end=length, eager=True).to_frame()
+        elements = ["A", "B", "C"]
+        series = choice(df, elements, dtype=String)
+        self._assert(series, length, elements, dtype=String)
+
+    def _assert(
+        self,
+        series: Series,
+        length: int,
+        elements: Iterable[Any],
+        /,
+        *,
+        dtype: PolarsDataType = Float64,
+    ) -> None:
+        assert series.dtype == dtype
+        assert series.len() == length
+        assert series.is_in(list(elements)).all()
+
+
 class TestCollectSeries:
     def test_main(self) -> None:
         expr = int_range(end=10)
@@ -1185,6 +1317,15 @@ class TestEnsureExprOrSeries:
         assert isinstance(result, Expr | Series)
 
 
+class TestEnsureExprOrSeriesMany:
+    @given(column=sampled_from(["column", col("column"), int_range(end=10)]))
+    def test_main(self, *, column: IntoExprColumn) -> None:
+        result = ensure_expr_or_series_many(column, column=column)
+        assert len(result) == 2
+        for r in result:
+            assert isinstance(r, Expr | Series)
+
+
 class TestFiniteEWMMean:
     alpha_0_75_values: ClassVar[list[float]] = [
         -8.269850726503885,
@@ -1472,8 +1613,7 @@ class TestIntegers:
     )
     def test_int(self, *, length: int, high: int) -> None:
         series = utilities.polars.integers(length, high)
-        assert series.len() == length
-        assert series.is_between(0, high, closed="left").all()
+        self._assert(series, length, high)
 
     @given(
         length=hypothesis.strategies.integers(0, 10),
@@ -1482,8 +1622,7 @@ class TestIntegers:
     def test_series(self, *, length: int, high: int) -> None:
         orig = int_range(end=length, eager=True)
         series = utilities.polars.integers(orig, high)
-        assert series.len() == length
-        assert series.is_between(0, high, closed="left").all()
+        self._assert(series, length, high)
 
     @given(
         length=hypothesis.strategies.integers(0, 10),
@@ -1492,6 +1631,10 @@ class TestIntegers:
     def test_dataframe(self, *, length: int, high: int) -> None:
         df = int_range(end=length, eager=True).to_frame()
         series = utilities.polars.integers(df, high)
+        self._assert(series, length, high)
+
+    def _assert(self, series: Series, length: int, high: int, /) -> None:
+        assert series.dtype == Int64
         assert series.len() == length
         assert series.is_between(0, high, closed="left").all()
 
@@ -1714,20 +1857,22 @@ class TestNormal:
     @given(length=hypothesis.strategies.integers(0, 10))
     def test_int(self, *, length: int) -> None:
         series = normal(length)
-        assert series.len() == length
-        assert series.is_finite().all()
+        self._assert(series, length)
 
     @given(length=hypothesis.strategies.integers(0, 10))
     def test_series(self, *, length: int) -> None:
         orig = int_range(end=length, eager=True)
         series = normal(orig)
-        assert series.len() == length
-        assert series.is_finite().all()
+        self._assert(series, length)
 
     @given(length=hypothesis.strategies.integers(0, 10))
     def test_dataframe(self, *, length: int) -> None:
         df = int_range(end=length, eager=True).to_frame()
         series = normal(df)
+        self._assert(series, length)
+
+    def _assert(self, series: Series, length: int, /) -> None:
+        assert series.dtype == Float64
         assert series.len() == length
         assert series.is_finite().all()
 
@@ -1927,6 +2072,7 @@ class TestUniform:
         series = uniform(length, low=low, high=high)
         assert series.len() == length
         assert series.is_between(low, high).all()
+        self._assert(series, length, low, high)
 
     @given(
         length=hypothesis.strategies.integers(0, 10),
@@ -1938,6 +2084,7 @@ class TestUniform:
         series = uniform(orig, low=low, high=high)
         assert series.len() == length
         assert series.is_between(low, high).all()
+        self._assert(series, length, low, high)
 
     @given(
         length=hypothesis.strategies.integers(0, 10),
@@ -1947,6 +2094,10 @@ class TestUniform:
         low, high = bounds
         df = int_range(end=length, eager=True).to_frame()
         series = uniform(df, low=low, high=high)
+        self._assert(series, length, low, high)
+
+    def _assert(self, series: Series, length: int, low: float, high: float, /) -> None:
+        assert series.dtype == Float64
         assert series.len() == length
         assert series.is_between(low, high).all()
 

{dycw_utilities-0.112.9 → dycw_utilities-0.112.10}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.112.9"
+__version__ = "0.112.10"

{dycw_utilities-0.112.9 → dycw_utilities-0.112.10}/src/utilities/numpy.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 
+from collections.abc import Sequence
 from dataclasses import dataclass
 from functools import partial, reduce
 from itertools import repeat
-from typing import TYPE_CHECKING, Any, overload, override
+from typing import TYPE_CHECKING, Any, SupportsIndex, overload, override
 
 import numpy as np
 from numpy import (
@@ -46,6 +47,9 @@ if TYPE_CHECKING:
     from utilities.types import MaybeIterable
 
 
+type ShapeLike = SupportsIndex | Sequence[SupportsIndex]
+
+
 ##
 
 
@@ -136,6 +140,19 @@ class AsIntError(Exception): ...
 ##
 
 
+def bernoulli(
+    *, true: float = 0.5, seed: int | None = None, size: ShapeLike = ()
+) -> NDArrayB:
+    """Return a set of Bernoulli random variates."""
+    from numpy.random import default_rng
+
+    rng = default_rng(seed=seed)
+    return rng.binomial(1, true, size=size).astype(bool)
+
+
+##
+
+
 def boxcar(
     array: NDArrayF,
     /,
@@ -980,11 +997,13 @@ __all__ = [
     "NDArrayF",
     "NDArrayI",
     "NDArrayO",
+    "ShapeLike",
     "ShiftError",
     "SigmoidError",
     "adjust_frequencies",
     "array_indexer",
     "as_int",
+    "bernoulli",
     "boxcar",
     "datetime64D",
     "datetime64M",