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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.112.9
+Version: 0.112.11
 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.11}/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.11"
 
 [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.11"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.112.9 → dycw_utilities-0.112.11}/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),