PySDM 3.0.0rc1__tar.gz → 3.0.0rc2__tar.gz

{pysdm-3.0.0rc1/pysdm.egg-info → pysdm-3.0.0rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pysdm
-Version: 3.0.0rc1
+Version: 3.0.0rc2
 Summary: Pythonic particle-based (super-droplet) warm-rain/aqueous-chemistry cloud microphysics package with box, parcel & 1D/2D prescribed-flow examples in Python, Julia and Matlab
 Author-email: "https://github.com/open-atmos/PySDM/graphs/contributors" <sylwester.arabas@agh.edu.pl>
 License-Expression: GPL-3.0-only

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/PySDM/environments/kinematic_1d.py

@@ -8,7 +8,7 @@ import numpy as np
 from PySDM.environments.impl.moist import Moist
 
 from PySDM.impl import arakawa_c
-from PySDM.initialisation.equilibrate_wet_radii import equilibrate_wet_radii
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 from PySDM.environments.impl import register_environment
 
 

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/PySDM/environments/kinematic_2d.py

@@ -7,7 +7,7 @@ import numpy as np
 
 from PySDM.environments.impl.moist import Moist
 from PySDM.impl.mesh import Mesh
-from PySDM.initialisation.equilibrate_wet_radii import (
+from PySDM.initialisation.hygroscopic_equilibrium import (
     default_rtol,
     equilibrate_wet_radii,
 )

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/PySDM/environments/parcel.py

@@ -8,7 +8,7 @@ import numpy as np
 
 from PySDM.environments.impl.moist import Moist
 from PySDM.impl.mesh import Mesh
-from PySDM.initialisation.equilibrate_wet_radii import (
+from PySDM.initialisation.hygroscopic_equilibrium import (
     default_rtol,
     equilibrate_wet_radii,
 )

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/PySDM/initialisation/__init__.py

@@ -5,7 +5,6 @@ wet radii equilibration
 
 from . import sampling, spectra
 from .discretise_multiplicities import discretise_multiplicities
-from .equilibrate_wet_radii import equilibrate_wet_radii
 from .init_fall_momenta import init_fall_momenta
 
 from . import aerosol_composition  # isort:skip

pysdm-3.0.0rc2/PySDM/initialisation/hygroscopic_equilibrium.py

@@ -0,0 +1,198 @@
+"""
+Koehler-curve equilibrium in unsaturated conditions
+"""
+
+import numba
+import numpy as np
+
+from ..backends.impl_numba.conf import JIT_FLAGS
+from ..backends.impl_numba.toms748 import toms748_solve
+from ..backends.impl_numba.warnings import warn
+
+default_rtol = 1e-5
+default_max_iters = 64
+
+# pylint: disable=too-many-locals,too-many-arguments
+
+
+def _solve_equilibrium_radii(
+    *,
+    radii_in: np.ndarray,
+    get_bounds,
+    get_args,
+    minfun,
+    environment,
+    kappa,
+    f_org: np.ndarray,
+    cell_id: np.ndarray,
+    rtol: float,
+    max_iters: int,
+    skip_fa_lt_zero: bool,
+    RH_range: tuple = (0, 1),
+):
+    T = environment["T"].to_ndarray()
+    RH = environment["RH"].to_ndarray()
+    formulae = environment.particulator.formulae
+    within_tolerance = formulae.trivia.within_tolerance
+    jit_flags = {**JIT_FLAGS, **{"fastmath": formulae.fastmath}}
+
+    if cell_id is None:
+        cell_id = np.zeros_like(radii_in, dtype=int)
+    if f_org is None:
+        f_org = np.zeros_like(radii_in, dtype=float)
+
+    @numba.njit(**jit_flags)
+    def impl(radii_in, iters, T, RH, cell_id, kappa, f_org):
+        radii_out = np.empty_like(radii_in)
+        for i in numba.prange(len(radii_in)):  # pylint: disable=not-an-iterable
+            cid = cell_id[i]
+            a, b = get_bounds(radii_in[i], T[cid], kappa[i])
+
+            if not a < b:
+                radii_out[i] = radii_in[i]
+                iters[i] = 0
+                continue
+
+            RH_i = np.maximum(RH_range[0], np.minimum(RH_range[1], RH[cid]))
+            args = get_args(T[cid], RH_i, kappa[i], radii_in[i], f_org[i])
+            fa, fb = minfun(a, *args), minfun(b, *args)
+
+            if skip_fa_lt_zero and fa < 0:
+                radii_out[i] = radii_in[i]
+                iters[i] = 0
+                continue
+
+            radii_out[i], iters[i] = toms748_solve(
+                minfun,
+                args,
+                a,
+                b,
+                fa,
+                fb,
+                rtol=rtol,
+                max_iter=max_iters,
+                within_tolerance=within_tolerance,
+            )
+            if iters[i] == -1:
+                warn(
+                    msg="failed to find equilibrium particle size",
+                    file=__file__,
+                    context=(
+                        "i",
+                        i,
+                        "r",
+                        radii_in[i],
+                        "T",
+                        T[cid],
+                        "RH",
+                        RH_i,
+                        "f_org",
+                        f_org[i],
+                        "kappa",
+                        kappa[i],
+                    ),
+                )
+        return radii_out
+
+    iters = np.empty_like(radii_in, dtype=int)
+    radii_out = impl(radii_in, iters, T, RH, cell_id, kappa, f_org)
+    assert (iters != max_iters).all() and (iters != -1).all()
+    return radii_out
+
+
+def equilibrate_dry_radii(
+    *,
+    r_wet: np.ndarray,
+    environment,
+    kappa: np.ndarray,
+    f_org: np.ndarray = None,
+    cell_id: np.ndarray = None,
+    rtol=default_rtol,
+    max_iters=default_max_iters,
+):
+    formulae = environment.particulator.formulae
+    sigma = formulae.surface_tension.sigma
+    phys_volume = formulae.trivia.volume
+    const = environment.particulator.formulae.constants
+    RH_eq = formulae.hygroscopicity.RH_eq
+    jit_flags = {**JIT_FLAGS, **{"fastmath": formulae.fastmath}}
+
+    @numba.njit(**{**jit_flags, "parallel": False})
+    def get_args(T_i, RH_i, kappa, r_wet, f_org):
+        return T_i, RH_i, kappa, r_wet, f_org
+
+    @numba.njit(**{**jit_flags, "parallel": False})
+    def get_bounds(r_wet, _, __):
+        return 0.0, r_wet
+
+    @numba.njit(**{**jit_flags, "parallel": False})
+    def minfun_dry(r_dry, temperature, relative_humidity, kappa, r_wet, f_org):
+        r_dry_3 = r_dry**3
+        sgm = sigma(
+            temperature, phys_volume(radius=r_wet), const.PI_4_3 * r_dry_3, f_org
+        )
+        return relative_humidity - RH_eq(r_wet, temperature, kappa, r_dry_3, sgm)
+
+    return _solve_equilibrium_radii(
+        radii_in=r_wet,
+        get_args=get_args,
+        get_bounds=get_bounds,
+        minfun=minfun_dry,
+        environment=environment,
+        kappa=kappa,
+        f_org=f_org,
+        cell_id=cell_id,
+        rtol=rtol,
+        max_iters=max_iters,
+        skip_fa_lt_zero=False,
+    )
+
+
+def equilibrate_wet_radii(
+    *,
+    r_dry: np.ndarray,
+    environment,
+    kappa_times_dry_volume: np.ndarray,
+    f_org: np.ndarray = None,
+    cell_id: np.ndarray = None,
+    rtol=default_rtol,
+    max_iters=default_max_iters,
+):
+    formulae = environment.particulator.formulae
+    sigma = formulae.surface_tension.sigma
+    phys_volume = formulae.trivia.volume
+    const = environment.particulator.formulae.constants
+    RH_eq = formulae.hygroscopicity.RH_eq
+    r_cr = formulae.hygroscopicity.r_cr
+    jit_flags = {**JIT_FLAGS, **{"fastmath": formulae.fastmath}}
+
+    @numba.njit(**{**jit_flags, "parallel": False})
+    def get_args(T_i, RH_i, kappa, r_dry, f_org):
+        return T_i, RH_i, kappa, r_dry**3, f_org
+
+    @numba.njit(**{**jit_flags, "parallel": False})
+    def get_bounds(r_dry, T_i, kappa):
+        a = r_dry
+        b = r_cr(kappa, r_dry**3, T_i, const.sgm_w)
+        return a, b
+
+    @numba.njit(**{**jit_flags, "parallel": False})
+    def minfun_wet(r_wet, temperature, relative_humidity, kappa, r_dry_3, f_org):
+        sgm = sigma(
+            temperature, phys_volume(radius=r_wet), const.PI_4_3 * r_dry_3, f_org
+        )
+        return relative_humidity - RH_eq(r_wet, temperature, kappa, r_dry_3, sgm)
+
+    return _solve_equilibrium_radii(
+        radii_in=r_dry,
+        get_args=get_args,
+        get_bounds=get_bounds,
+        minfun=minfun_wet,
+        environment=environment,
+        kappa=kappa_times_dry_volume / phys_volume(radius=r_dry),
+        f_org=f_org,
+        cell_id=cell_id,
+        rtol=rtol,
+        max_iters=max_iters,
+        skip_fa_lt_zero=True,
+    )

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/docs/markdown/pysdm_landing.md

@@ -386,7 +386,7 @@ si = pyimport("PySDM.physics").si
 spectral_sampling = pyimport("PySDM.initialisation.sampling").spectral_sampling
 discretise_multiplicities = pyimport("PySDM.initialisation").discretise_multiplicities
 Lognormal = pyimport("PySDM.initialisation.spectra").Lognormal
-equilibrate_wet_radii = pyimport("PySDM.initialisation").equilibrate_wet_radii
+equilibrate_wet_radii = pyimport("PySDM.initialisation.hygroscopic_equilibrium").equilibrate_wet_radii
 CPU = pyimport("PySDM.backends").CPU
 AmbientThermodynamics = pyimport("PySDM.dynamics").AmbientThermodynamics
 Condensation = pyimport("PySDM.dynamics").Condensation
@@ -467,7 +467,7 @@ si = py.importlib.import_module('PySDM.physics').si;
 spectral_sampling = py.importlib.import_module('PySDM.initialisation.sampling').spectral_sampling;
 discretise_multiplicities = py.importlib.import_module('PySDM.initialisation').discretise_multiplicities;
 Lognormal = py.importlib.import_module('PySDM.initialisation.spectra').Lognormal;
-equilibrate_wet_radii = py.importlib.import_module('PySDM.initialisation').equilibrate_wet_radii;
+equilibrate_wet_radii = py.importlib.import_module('PySDM.initialisation.hygroscopic_equilibrium').equilibrate_wet_radii;
 CPU = py.importlib.import_module('PySDM.backends').CPU;
 AmbientThermodynamics = py.importlib.import_module('PySDM.dynamics').AmbientThermodynamics;
 Condensation = py.importlib.import_module('PySDM.dynamics').Condensation;
@@ -568,7 +568,8 @@ saveas(gcf, "parcel.png");
 ```Python
 from matplotlib import pyplot
 from PySDM.physics import si
-from PySDM.initialisation import discretise_multiplicities, equilibrate_wet_radii
+from PySDM.initialisation import discretise_multiplicities
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 from PySDM.initialisation.spectra import Lognormal
 from PySDM.initialisation.sampling import spectral_sampling
 from PySDM.backends import CPU

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/examples/PySDM_examples/Abdul_Razzak_Ghan_2000/run_ARG_parcel.py

@@ -8,7 +8,7 @@ from PySDM import products as PySDM_products
 from PySDM.backends import CPU
 from PySDM.dynamics import AmbientThermodynamics, Condensation
 from PySDM.environments import Parcel
-from PySDM.initialisation import equilibrate_wet_radii
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 from PySDM.initialisation.sampling.spectral_sampling import ConstantMultiplicity
 from PySDM.physics import si
 

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/examples/PySDM_examples/Arabas_and_Shima_2017/simulation.py

@@ -5,7 +5,7 @@ from PySDM.backends import CPU
 from PySDM.builder import Builder
 from PySDM.dynamics import AmbientThermodynamics, Condensation
 from PySDM.environments import Parcel
-from PySDM.initialisation import equilibrate_wet_radii
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 from PySDM.physics import constants as const
 
 

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/examples/PySDM_examples/Arabas_et_al_2025/aida.ipynb

@@ -69,7 +69,7 @@
     "from PySDM import products as PySDM_products\n",
     "from PySDM.initialisation.sampling import spectral_sampling\n",
     "from PySDM.initialisation.spectra import Lognormal\n",
-    "from PySDM.initialisation import equilibrate_wet_radii\n",
+    "from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii\n",
     "\n",
     "from PySDM_examples.utils import BasicSimulation"
    ],

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/examples/PySDM_examples/Grabowski_and_Pawlowska_2023/simulation.py

@@ -6,7 +6,7 @@ from PySDM.backends import CPU
 from PySDM.backends.impl_numba.test_helpers import scipy_ode_condensation_solver
 from PySDM.dynamics import AmbientThermodynamics, Condensation
 from PySDM.environments import Parcel
-from PySDM.initialisation import equilibrate_wet_radii
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 from PySDM.initialisation.sampling.spectral_sampling import ConstantMultiplicity
 from PySDM.physics import si
 

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/examples/PySDM_examples/Graf_et_al_2019/figure_4.ipynb

@@ -64,7 +64,8 @@
     "from PySDM.dynamics import AmbientThermodynamics, Condensation\n",
     "from PySDM.initialisation.spectra import Lognormal\n",
     "from PySDM.initialisation.sampling import spectral_sampling\n",
-    "from PySDM.initialisation import equilibrate_wet_radii, discretise_multiplicities\n",
+    "from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii\n",
+    "from PySDM.initialisation import discretise_multiplicities\n",
     "from PySDM import products"
    ]
   },

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/examples/PySDM_examples/Lowe_et_al_2019/simulation.py

@@ -5,7 +5,7 @@ import PySDM.products as PySDM_products
 from PySDM import Builder
 from PySDM.dynamics import AmbientThermodynamics, Condensation
 from PySDM.environments import Parcel
-from PySDM.initialisation import equilibrate_wet_radii
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 from PySDM.initialisation.spectra import Sum
 
 

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/examples/PySDM_examples/Niedermeier_et_al_2014/simulation.py

@@ -9,7 +9,7 @@ from PySDM import Builder
 from PySDM.backends import CPU
 from PySDM.dynamics import AmbientThermodynamics, Condensation, Freezing
 from PySDM.environments import Parcel
-from PySDM.initialisation import equilibrate_wet_radii
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 from PySDM.products import AmbientTemperature, IceWaterContent, ParcelDisplacement
 
 

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/examples/PySDM_examples/Pyrcel/simulation.py

@@ -6,7 +6,7 @@ from PySDM.backends import CPU
 from PySDM.backends.impl_numba.test_helpers import scipy_ode_condensation_solver
 from PySDM.dynamics import AmbientThermodynamics, Condensation
 from PySDM.environments import Parcel
-from PySDM.initialisation import equilibrate_wet_radii
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 from PySDM.initialisation.sampling.spectral_sampling import ConstantMultiplicity
 from PySDM.physics import si
 

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/examples/PySDM_examples/Spichtinger_et_al_2023/simulation.py

@@ -12,7 +12,8 @@ from PySDM.dynamics import (
     VapourDepositionOnIce,
 )
 from PySDM.environments import Parcel
-from PySDM.initialisation import discretise_multiplicities, equilibrate_wet_radii
+from PySDM.initialisation import discretise_multiplicities
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 
 
 class Simulation(BasicSimulation):

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2/pysdm.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pysdm
-Version: 3.0.0rc1
+Version: 3.0.0rc2
 Summary: Pythonic particle-based (super-droplet) warm-rain/aqueous-chemistry cloud microphysics package with box, parcel & 1D/2D prescribed-flow examples in Python, Julia and Matlab
 Author-email: "https://github.com/open-atmos/PySDM/graphs/contributors" <sylwester.arabas@agh.edu.pl>
 License-Expression: GPL-3.0-only

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/pysdm.egg-info/SOURCES.txt

@@ -204,7 +204,7 @@ PySDM/impl/particle_attributes_factory.py
 PySDM/impl/wall_timer.py
 PySDM/initialisation/__init__.py
 PySDM/initialisation/discretise_multiplicities.py
-PySDM/initialisation/equilibrate_wet_radii.py
+PySDM/initialisation/hygroscopic_equilibrium.py
 PySDM/initialisation/init_fall_momenta.py
 PySDM/initialisation/aerosol_composition/__init__.py
 PySDM/initialisation/aerosol_composition/dry_aerosol.py
@@ -910,7 +910,7 @@ tests/unit_tests/impl/test_particle_attributes.py
 tests/unit_tests/initialisation/__init__.py
 tests/unit_tests/initialisation/test_aerosol_init.py
 tests/unit_tests/initialisation/test_discretise_multiplicities.py
-tests/unit_tests/initialisation/test_equilibrate_wet_radii.py
+tests/unit_tests/initialisation/test_hygroscopic_equilibrium.py
 tests/unit_tests/initialisation/test_init_fall_momenta.py
 tests/unit_tests/initialisation/test_spatial_discretisation.py
 tests/unit_tests/initialisation/test_spectra_lognormal.py

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/tests/smoke_tests/box/partmc/test_dry_wet_equilibration.py

@@ -12,7 +12,7 @@ import PyPartMC
 from PySDM import Builder
 from PySDM.backends import CPU
 from PySDM.environments import Box
-from PySDM.initialisation import equilibrate_wet_radii
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 from PySDM.initialisation.spectra import Lognormal
 from PySDM.physics import si
 

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/tests/smoke_tests/parcel_c/abdul_razzak_ghan_2000/test_single_supersaturation_peak.py

@@ -11,7 +11,7 @@ from PySDM.backends import CPU
 from PySDM.backends.impl_numba.test_helpers import scipy_ode_condensation_solver
 from PySDM.dynamics import AmbientThermodynamics, Condensation
 from PySDM.environments import Parcel
-from PySDM.initialisation import equilibrate_wet_radii
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 from PySDM.initialisation.sampling.spectral_sampling import ConstantMultiplicity
 from PySDM.initialisation.spectra import Lognormal
 from PySDM.physics import si

{pysdm-3.0.0rc1 → pysdm-3.0.0rc2}/tests/unit_tests/dynamics/condensation/test_parcel_sanity_checks.py

@@ -10,7 +10,8 @@ from PySDM import Builder, Formulae, products
 from PySDM.backends import CPU, GPU
 from PySDM.dynamics import AmbientThermodynamics, Condensation
 from PySDM.environments import Parcel
-from PySDM.initialisation import discretise_multiplicities, equilibrate_wet_radii
+from PySDM.initialisation import discretise_multiplicities
+from PySDM.initialisation.hygroscopic_equilibrium import equilibrate_wet_radii
 from PySDM.initialisation.sampling import spectral_sampling
 from PySDM.initialisation.spectra import Lognormal
 from PySDM.physics import si

pysdm-3.0.0rc2/tests/unit_tests/initialisation/test_hygroscopic_equilibrium.py

@@ -0,0 +1,169 @@
+# pylint: disable=missing-module-docstring,missing-class-docstring,missing-function-docstring
+import numpy as np
+import pytest
+from matplotlib import pyplot
+
+from PySDM import Formulae, Builder
+from PySDM.backends import CPU
+from PySDM.initialisation.hygroscopic_equilibrium import (
+    equilibrate_wet_radii,
+    equilibrate_dry_radii,
+)
+from PySDM.physics import constants_defaults as const
+from PySDM.physics import si
+from PySDM.environments import Box
+
+
+class TestHygroscopicEquilibrium:
+    @staticmethod
+    @pytest.mark.parametrize("r_dry", (pytest.param(2.4e-09), pytest.param(2.5e-09)))
+    @pytest.mark.parametrize(
+        "surface_tension",
+        ("Constant", "CompressedFilmOvadnevaite", "SzyszkowskiLangmuir"),
+    )
+    def test_equilibrate_wet_radii(r_dry, surface_tension, plot=False):
+        # Arrange
+        T = 280.0
+        RH = 0.9
+        f_org = 0.607
+        kappa = 0.356
+
+        class Particulator:  # pylint: disable=too-few-public-methods
+            formulae = Formulae(
+                surface_tension=surface_tension,
+                constants={
+                    "sgm_org": 40 * si.mN / si.m,
+                    "delta_min": 0.1 * si.nm,
+                    "RUEHL_nu_org": 7.47e-05,
+                    "RUEHL_A0": 2.5e-19 * si.m**2,
+                    "RUEHL_C0": 1e-5,
+                    "RUEHL_sgm_min": 40 * si.mN / si.m,
+                },
+            )
+
+        class Env:  # pylint: disable=too-few-public-methods
+            particulator = Particulator()
+            thermo = {
+                "T": CPU.Storage.from_ndarray(np.full(1, T)),
+                "RH": CPU.Storage.from_ndarray(np.full(1, RH)),
+            }
+
+            def __getitem__(self, item):
+                return self.thermo[item]
+
+        r_dry_arr = np.full(1, r_dry)
+
+        # Plot
+        r_wet = np.logspace(np.log(0.9 * r_dry), np.log(10 * si.nm), base=np.e, num=100)
+        sigma = Env.particulator.formulae.surface_tension.sigma(
+            T,
+            Env.particulator.formulae.trivia.volume(r_wet),
+            Env.particulator.formulae.trivia.volume(r_dry),
+            f_org,
+        )
+        RH_eq = Env.particulator.formulae.hygroscopicity.RH_eq(
+            r_wet, T, kappa, r_dry**3, sigma
+        )
+
+        pyplot.plot(r_wet / si.nm, (RH_eq - 1) * 100, label="RH_eq")
+        pyplot.axhline((RH - 1) * 100, color="orange", label="RH")
+        pyplot.axvline(r_dry / si.nm, label="a", color="red")
+        pyplot.axvline(
+            Env.particulator.formulae.hygroscopicity.r_cr(
+                kappa, r_dry**3, T, const.sgm_w
+            )
+            / si.nm,
+            color="green",
+            label="b",
+        )
+        pyplot.grid()
+        pyplot.xscale("log")
+        pyplot.xlabel("Wet radius [nm]")
+        pyplot.xlim(r_wet[0] / si.nm, r_wet[-1] / si.nm)
+        pyplot.ylabel("Equilibrium supersaturation [%]")
+        pyplot.legend()
+        if plot:
+            pyplot.show()
+        else:
+            pyplot.clf()
+
+        # Act
+        env = Env()
+        r_wet = equilibrate_wet_radii(
+            r_dry=r_dry_arr,
+            environment=env,
+            kappa_times_dry_volume=Env.particulator.formulae.trivia.volume(r_dry_arr)
+            * kappa,
+            f_org=np.full_like(r_dry_arr, f_org),
+        )
+
+        # Assert
+        assert (r_wet >= r_dry_arr).all()
+
+    @staticmethod
+    @pytest.mark.parametrize(
+        "kappas, temperature, relative_humidity",
+        (
+            pytest.param([0.0], 300 * si.K, 0.95, marks=pytest.mark.xfail(strict=True)),
+            ([0.5, 1.0, 1.5], 300 * si.K, 0.95),
+            ([0.5, 1.0, 1.5], 250 * si.K, 0.75),
+        ),
+    )
+    def test_equilibrate_dry_radii(
+        kappas, temperature, relative_humidity, backend_instance, plot=False
+    ):
+        # arrange
+        r_wet = np.logspace(-8, -3, num=10)
+        builder = Builder(
+            environment=Box(dv=np.nan, dt=np.nan),
+            backend=backend_instance,
+            n_sd=len(r_wet),
+        )
+        builder.particulator.environment["T"] = temperature
+        builder.particulator.environment["RH"] = relative_humidity
+
+        # act
+        r_dry = {}
+        for kappa in kappas:
+            r_dry[kappa] = equilibrate_dry_radii(
+                r_wet=r_wet,
+                environment=builder.particulator.environment,
+                kappa=kappa if kappa is np.ndarray else np.full_like(r_wet, kappa),
+            )
+
+        # plot
+        for kappa in kappas:
+            pyplot.loglog(r_wet, r_dry[kappa], label=f"{kappa=}")
+        pyplot.gca().set(
+            ylim=(r_wet[0], r_wet[-1]),
+            xlabel="wet radius [m]",
+            ylabel="dry radius [m]",
+            title=f"{temperature=} {relative_humidity=}",
+        )
+        pyplot.grid()
+        pyplot.legend()
+        if plot:
+            pyplot.show()
+        else:
+            pyplot.clf()
+
+        # assert
+        assert (np.diff(kappas) > 0).all()
+        kappa_prev = None
+        for kappa in kappas:
+            assert (r_dry[kappa] < r_wet).all()
+
+            if kappa_prev is not None:
+                assert (r_dry[kappa] < r_dry[kappa_prev]).all()
+            kappa_prev = kappa
+
+            np.testing.assert_allclose(
+                rtol=1e-5,
+                desired=r_wet,
+                actual=equilibrate_wet_radii(
+                    r_dry=r_dry[kappa],
+                    kappa_times_dry_volume=kappa
+                    * builder.particulator.formulae.trivia.volume(r_dry[kappa]),
+                    environment=builder.particulator.environment,
+                ),
+            )