PySDM 2.54__tar.gz → 2.58__tar.gz

{PySDM-2.54 → PySDM-2.58}/.github/workflows/tests+artifacts+pypi.yml

@@ -192,13 +192,15 @@ jobs:
           echo toml_ci_md5=$(cat pyproject.toml setup.py examples/pyproject.toml examples/setup.py tests/devops_tests/requirements.txt .github/workflows/tests+artifacts+pypi.yml \
           | python -c "import hashlib;print(hashlib.md5(open(0,'rb').read()).hexdigest())") >> $GITHUB_ENV
       - uses: actions/cache@v4
+        id: cache
         with:
           path: ${{ env.pip_user_site }}
           key: examples_pip_cache-${{ matrix.platform }}-${{ matrix.python-version }}-${{ env.toml_ci_md5 }}
+          lookup-only: true
       - if: steps.cache.outputs.cache-hit != 'true'
         run: |
           python -m pip uninstall -y jupyterlab-server # https://github.com/pypa/pip/issues/6275
-          python -m pip install -e . ./examples
+          python -m pip install -e . -e ./examples
           python -m pip install -r tests/devops_tests/requirements.txt
       - if: steps.cache.outputs.cache-hit != 'true'
         uses: actions/cache/save@v4

{PySDM-2.54 → PySDM-2.58}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PySDM
-Version: 2.54
+Version: 2.58
 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
 Home-page: https://github.com/open-atmos/PySDM
 Author: https://github.com/open-atmos/PySDM/graphs/contributors
@@ -74,7 +74,7 @@ Description: # PySDM
         - [de Jong, Singer et al. 2023](https://doi.org/10.21105/joss.04968) (PySDM v2).
           
         PySDM includes an extension of the SDM scheme to represent collisional breakup described in [de Jong, Mackay et al. 2023](10.5194/gmd-16-4193-2023).   
-        For a list of talks and other materials on PySDM, see the [project wiki](https://github.com/open-atmos/PySDM/wiki).
+        For a list of talks and other materials on PySDM as well as a list of published papers featuring PySDM simulations, see the [project wiki](https://github.com/open-atmos/PySDM/wiki).
         
         A [pdoc-generated](https://pdoc3.github.io/pdoc) documentation of PySDM public API is maintained at: [https://open-atmos.github.io/PySDM](https://open-atmos.github.io/PySDM) 
         
@@ -846,6 +846,8 @@ Description: # PySDM
           https://yoctoyotta1024.github.io/CLEO/
         - droplets.jl (Julia)   
           https://github.com/emmacware/droplets.jl
+        - LacmoPy (Python/Numba)    
+          https://github.com/JanKBohrer/LacmoPy/blob/master/collision/all_or_nothing.py
         
         ### non-SDM probabilistic particle-based coagulation solvers
         

{PySDM-2.54 → PySDM-2.58}/PySDM/backends/impl_numba/methods/chemistry_methods.py

@@ -62,49 +62,48 @@ class ChemistryMethods(BackendMethods):
         droplet_volume,
         multiplicity,
     ):
-        for thread_id in numba.prange(n_threads):  # pylint: disable=not-an-iterable
-            for i in range(thread_id, n_cell, n_threads):
-                cell_id = cell_order[i]
-
-                cell_start = cell_start_arg[cell_id]
-                cell_end = cell_start_arg[cell_id + 1]
-                n_sd_in_cell = cell_end - cell_start
-                if n_sd_in_cell == 0:
-                    continue
-
-                super_droplet_ids = numba.typed.List()
-                for sd_id in idx[cell_start:cell_end]:
-                    if do_chemistry_flag.data[sd_id]:
-                        super_droplet_ids.append(sd_id)
-
-                if len(super_droplet_ids) == 0:
-                    return
-
-                for key, compound in GASEOUS_COMPOUNDS.items():
-                    ChemistryMethods.dissolution_body(
-                        super_droplet_ids=super_droplet_ids,
-                        mole_amounts=mole_amounts[key].data,
-                        env_mixing_ratio=env_mixing_ratio[compound][
-                            cell_id : cell_id + 1
-                        ],
-                        henrysConstant=self.HENRY_CONST.HENRY_CONST[compound].at(
-                            env_T[cell_id]
-                        ),
-                        env_p=env_p[cell_id],
-                        env_T=env_T[cell_id],
-                        env_rho_d=env_rho_d[cell_id],
-                        timestep=timestep,
-                        dv=dv,
-                        droplet_volume=droplet_volume.data,
-                        multiplicity=multiplicity.data,
-                        system_type=system_type,
-                        specific_gravity=self.specific_gravities[compound],
-                        alpha=MASS_ACCOMMODATION_COEFFICIENTS[compound],
-                        diffusion_const=DIFFUSION_CONST[compound],
-                        dissociation_factor=dissociation_factors[compound].data,
-                        radius=self.formulae.trivia.radius,
-                        const=self.formulae.constants,
-                    )
+        assert n_cell == 1
+        assert n_threads == 1
+        for i in range(n_cell):
+            cell_id = cell_order[i]
+
+            cell_start = cell_start_arg[cell_id]
+            cell_end = cell_start_arg[cell_id + 1]
+            n_sd_in_cell = cell_end - cell_start
+            if n_sd_in_cell == 0:
+                continue
+
+            super_droplet_ids = numba.typed.List()
+            for sd_id in idx[cell_start:cell_end]:
+                if do_chemistry_flag.data[sd_id]:
+                    super_droplet_ids.append(sd_id)
+
+            if len(super_droplet_ids) == 0:
+                return
+
+            for key, compound in GASEOUS_COMPOUNDS.items():
+                ChemistryMethods.dissolution_body(
+                    super_droplet_ids=super_droplet_ids,
+                    mole_amounts=mole_amounts[key].data,
+                    env_mixing_ratio=env_mixing_ratio[compound][cell_id : cell_id + 1],
+                    henrysConstant=self.HENRY_CONST.HENRY_CONST[compound].at(
+                        env_T[cell_id]
+                    ),
+                    env_p=env_p[cell_id],
+                    env_T=env_T[cell_id],
+                    env_rho_d=env_rho_d[cell_id],
+                    timestep=timestep,
+                    dv=dv,
+                    droplet_volume=droplet_volume.data,
+                    multiplicity=multiplicity.data,
+                    system_type=system_type,
+                    specific_gravity=self.specific_gravities[compound],
+                    alpha=MASS_ACCOMMODATION_COEFFICIENTS[compound],
+                    diffusion_const=DIFFUSION_CONST[compound],
+                    dissociation_factor=dissociation_factors[compound].data,
+                    radius=self.formulae.trivia.radius,
+                    const=self.formulae.constants,
+                )
 
     @staticmethod
     @numba.njit(**{**conf.JIT_FLAGS, **{"parallel": False}})

{PySDM-2.54 → PySDM-2.58}/PySDM/exporters/netcdf_exporter_1d.py

@@ -75,7 +75,9 @@ class NetCDFExporter_1d:  # pylint: disable=too-few-public-methods,too-many-inst
                 )
 
             n_dimensions = len(instance.shape)
-            if n_dimensions == 1:
+            if n_dimensions == 0:
+                dimensions = ("time",)
+            elif n_dimensions == 1:
                 dimensions = ("height", "time")
             elif n_dimensions == 2:
                 dim_name = name.replace(" ", "_") + "_bin_index"
@@ -94,7 +96,9 @@ class NetCDFExporter_1d:  # pylint: disable=too-few-public-methods,too-many-inst
     def _write_variables(self):
         for var in self.simulator.particulator.products.keys():
             n_dimensions = len(self.simulator.particulator.products[var].shape)
-            if n_dimensions == 1:
+            if n_dimensions == 0:
+                self.vars[var][:] = self.data[var][:]
+            elif n_dimensions == 1:
                 self.vars[var][:, :] = self.data[var][-self.nz_export :, :]
             elif n_dimensions == 2:
                 if self.n_save_spec == 0:

{PySDM-2.54 → PySDM-2.58}/PySDM.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PySDM
-Version: 2.54
+Version: 2.58
 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
 Home-page: https://github.com/open-atmos/PySDM
 Author: https://github.com/open-atmos/PySDM/graphs/contributors
@@ -74,7 +74,7 @@ Description: # PySDM
         - [de Jong, Singer et al. 2023](https://doi.org/10.21105/joss.04968) (PySDM v2).
           
         PySDM includes an extension of the SDM scheme to represent collisional breakup described in [de Jong, Mackay et al. 2023](10.5194/gmd-16-4193-2023).   
-        For a list of talks and other materials on PySDM, see the [project wiki](https://github.com/open-atmos/PySDM/wiki).
+        For a list of talks and other materials on PySDM as well as a list of published papers featuring PySDM simulations, see the [project wiki](https://github.com/open-atmos/PySDM/wiki).
         
         A [pdoc-generated](https://pdoc3.github.io/pdoc) documentation of PySDM public API is maintained at: [https://open-atmos.github.io/PySDM](https://open-atmos.github.io/PySDM) 
         
@@ -846,6 +846,8 @@ Description: # PySDM
           https://yoctoyotta1024.github.io/CLEO/
         - droplets.jl (Julia)   
           https://github.com/emmacware/droplets.jl
+        - LacmoPy (Python/Numba)    
+          https://github.com/JanKBohrer/LacmoPy/blob/master/collision/all_or_nothing.py
         
         ### non-SDM probabilistic particle-based coagulation solvers
         

{PySDM-2.54 → PySDM-2.58}/README.md

@@ -63,7 +63,7 @@ For an overview of PySDM features (and the preferred way to cite PySDM in papers
 - [de Jong, Singer et al. 2023](https://doi.org/10.21105/joss.04968) (PySDM v2).
   
 PySDM includes an extension of the SDM scheme to represent collisional breakup described in [de Jong, Mackay et al. 2023](10.5194/gmd-16-4193-2023).   
-For a list of talks and other materials on PySDM, see the [project wiki](https://github.com/open-atmos/PySDM/wiki).
+For a list of talks and other materials on PySDM as well as a list of published papers featuring PySDM simulations, see the [project wiki](https://github.com/open-atmos/PySDM/wiki).
 
 A [pdoc-generated](https://pdoc3.github.io/pdoc) documentation of PySDM public API is maintained at: [https://open-atmos.github.io/PySDM](https://open-atmos.github.io/PySDM) 
 
@@ -835,6 +835,8 @@ licence: [GPL v3](https://www.gnu.org/licenses/gpl-3.0.html)
   https://yoctoyotta1024.github.io/CLEO/
 - droplets.jl (Julia)   
   https://github.com/emmacware/droplets.jl
+- LacmoPy (Python/Numba)    
+  https://github.com/JanKBohrer/LacmoPy/blob/master/collision/all_or_nothing.py
 
 ### non-SDM probabilistic particle-based coagulation solvers
 

{PySDM-2.54 → PySDM-2.58}/examples/PySDM_examples/Shipway_and_Hill_2012/fig_1.ipynb

@@ -44,6 +44,15 @@
     "    pip_install_on_colab('PySDM-examples')"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 2,
@@ -85,6 +94,9 @@
     "    \"kappa\": .9,\n",
     "    \"particles_per_volume_STP\": 50 / si.cm**3\n",
     "}\n",
+    "if 'CI' in os.environ:\n",
+    "    common_params[\"t_max\"] = 10 * common_params[\"dt\"]\n",
+    "    common_params[\"n_sd_per_gridbox\"] = 16 \n",
     "\n",
     "output = {}\n",
     "settings = {}\n",
@@ -104,7 +116,7 @@
     "            precip=precip\n",
     "        )\n",
     "        simulation[key] = Simulation(settings[key])\n",
-    "        output[key] = simulation[key].run().products\n"
+    "        output[key] = simulation[key].run().products"
    ]
   },
   {
@@ -327,7 +339,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3.9.4 ('edjPySDM')",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -341,7 +353,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.4"
+   "version": "3.9.2"
   },
   "vscode": {
    "interpreter": {
@@ -350,5 +362,5 @@
   }
  },
  "nbformat": 4,
- "nbformat_minor": 1
+ "nbformat_minor": 4
 }

{PySDM-2.54 → PySDM-2.58}/examples/PySDM_examples/Shipway_and_Hill_2012/simulation.py

@@ -168,6 +168,7 @@ class Simulation:
                     PySDM_products.CoalescenceRatePerGridbox(
                         name="coalescence_rate",
                     ),
+                    PySDM_products.SurfacePrecipitation(),
                 ]
             )
         self.particulator = self.builder.build(
@@ -182,7 +183,9 @@ class Simulation:
         }
         self.output_products = {}
         for k, v in self.particulator.products.items():
-            if len(v.shape) == 1:
+            if len(v.shape) == 0:
+                self.output_products[k] = np.zeros(self.nt + 1)
+            elif len(v.shape) == 1:
                 self.output_products[k] = np.zeros((self.mesh.grid[-1], self.nt + 1))
             elif len(v.shape) == 2:
                 number_of_time_sections = len(self.save_spec_and_attr_times)
@@ -203,7 +206,10 @@ class Simulation:
         for k, v in self.particulator.products.items():
             if len(v.shape) > 1:
                 continue
-            self.output_products[k][:, step] = v.get()
+            if len(v.shape) == 1:
+                self.output_products[k][:, step] = v.get()
+            else:
+                self.output_products[k][step] = v.get()
 
     def save_spectrum(self, index):
         for k, v in self.particulator.products.items():

{PySDM-2.54 → PySDM-2.58}/examples/setup.py

@@ -32,7 +32,11 @@ setup(
         "seaborn",
         "numdifftools",
     ]
-    + (["pyvinecopulib", "vtk"] if platform.architecture()[0] != "32bit" else []),
+    + (
+        ["pyvinecopulib" + "==0.6.4" if CI else "", "vtk"]
+        if platform.architecture()[0] != "32bit"
+        else []
+    ),
     extras_require={
         "tests": [
             "pytest",

PySDM-2.58/tests/smoke_tests/kinematic_1d/shipway_and_hill_2012/test_1d_exporters.py

@@ -0,0 +1,159 @@
+# pylint: disable = missing-module-docstring,missing-class-docstring,missing-function-docstring
+import os
+import platform
+from collections import namedtuple
+from pathlib import Path
+from tempfile import TemporaryDirectory
+
+import numpy as np
+import pytest
+from open_atmos_jupyter_utils import TemporaryFile
+from PySDM_examples.Shipway_and_Hill_2012 import Settings, Simulation
+from PySDM_examples.utils import readVTK_1d
+
+from PySDM.exporters import NetCDFExporter_1d, VTKExporter_1d, readNetCDF_1d
+from PySDM.physics import si
+
+
+@pytest.fixture(name="simulation_1d")
+def simulation_1d_fixture():
+    n_sd_per_gridbox = 16
+    settings = Settings(
+        n_sd_per_gridbox=n_sd_per_gridbox,
+        dt=60 * si.s,
+        dz=200 * si.m,
+        precip=True,
+        rho_times_w_1=2 * si.m / si.s * si.kg / si.m**3,
+    )
+    settings.t_max = 20 * settings.dt
+    settings.save_spec_and_attr_times = [0 * si.min, 10 * si.min, 20 * si.min]
+    simulation = Simulation(settings)
+    results = simulation.run()
+    return namedtuple(
+        Path(__file__).stem + "_FixtureData", ("results, settings, simulation")
+    )(results=results, settings=settings, simulation=simulation)
+
+
+class Test2DExporters:
+    @staticmethod
+    @pytest.mark.parametrize(
+        "exclude_particle_reservoir",
+        (
+            False,
+            True,
+        ),
+    )
+    def test_netcdf_exporter_1d(simulation_1d, exclude_particle_reservoir):
+        # Arrange
+        data = simulation_1d.results.products
+        settings = simulation_1d.settings
+        nz_export = (
+            int(settings.z_max / settings.dz)
+            if exclude_particle_reservoir
+            else settings.nz
+        )
+
+        # Act
+        file = TemporaryFile(".nc")
+        netcdf_exporter = NetCDFExporter_1d(
+            data,
+            settings,
+            simulation_1d.simulation,
+            filename=file.absolute_path,
+            exclude_particle_reservoir=exclude_particle_reservoir,
+        )
+        netcdf_exporter.run()
+        data_from_file = readNetCDF_1d(file.absolute_path)
+
+        # Assert
+        assert data_from_file.products["time"].shape == data["t"].shape
+        assert data_from_file.products["height"].shape == data["z"][-nz_export:].shape
+        assert data_from_file.products["T"].shape == data["T"][-nz_export:, :].shape
+        assert (
+            data_from_file.products["dry spectrum"].shape
+            == data["dry spectrum"][-nz_export:, :, :].shape
+        )
+
+        assert np.amin(data_from_file.products["time"]) == np.amin(data["t"])
+        assert np.amax(data_from_file.products["time"]) == np.amax(data["t"])
+        assert np.amin(data_from_file.products["height"]) == np.amin(
+            data["z"][-nz_export:]
+        )
+        assert np.amax(data_from_file.products["height"]) == np.amax(
+            data["z"][-nz_export:]
+        )
+        assert data_from_file.products["rhod"].mean() == pytest.approx(
+            data["rhod"][-nz_export:, :].mean(), 1e-6
+        )
+        assert data_from_file.products["wet spectrum"].mean() == pytest.approx(
+            data["wet spectrum"][-nz_export:, :, :].mean(), 1e-6
+        )
+
+        assert data_from_file.settings["precip"] == settings.precip
+        assert data_from_file.settings["kappa"] == settings.kappa
+        assert (
+            data_from_file.settings["r_bins_edges"].size == settings.number_of_bins + 1
+        )
+
+    @staticmethod
+    @pytest.mark.skipif(
+        platform.architecture()[0] == "32bit", reason="Not available vtk module!"
+    )
+    @pytest.mark.parametrize(
+        "exclude_particle_reservoir",
+        (
+            False,
+            True,
+        ),
+    )
+    def test_vtk_exporter_1d(
+        simulation_1d, exclude_particle_reservoir
+    ):  # pylint: disable=too-many-locals
+        # Arrange
+        data = simulation_1d.results.attributes
+        settings = simulation_1d[1]
+        z0 = 0.0 if exclude_particle_reservoir else -settings.particle_reservoir_depth
+        exported_particles_indexes = {}
+        number_of_exported_particles = []
+        for i, t in enumerate(settings.save_spec_and_attr_times):
+            exported_particles_indexes[t] = np.where(
+                data["cell origin"][i][0]
+                >= int((z0 + settings.particle_reservoir_depth) / settings.dz)
+            )
+            number_of_exported_particles.append(exported_particles_indexes[t][0].size)
+
+        # Act
+        with TemporaryDirectory() as tmpdir_:
+            tmpdir = tmpdir_ + "/"
+            vtk_exporter = VTKExporter_1d(
+                data,
+                settings,
+                path=tmpdir,
+                exclude_particle_reservoir=exclude_particle_reservoir,
+            )
+            vtk_exporter.run()
+            written_files_list = os.listdir(tmpdir)
+            data_from_file = {}
+            for t in settings.save_spec_and_attr_times:
+                leading_zeros_in_filename = [
+                    "0" for i in range(len(str(settings.t_max)) - len(str(t)))
+                ]
+                filename = "time" + "".join(leading_zeros_in_filename) + str(t) + ".vtu"
+                data_from_file[t] = readVTK_1d(tmpdir + filename)
+
+        # Assert
+        for i, t in enumerate(settings.save_spec_and_attr_times):
+            filename_leading_zeros = "".join(
+                ["0" for i in range(len(str(settings.t_max)) - len(str(t)))]
+            )
+            filename = "time" + filename_leading_zeros + str(t) + ".vtu"
+            assert filename in written_files_list
+            assert z0 <= np.amin(data_from_file[t]["z"])
+            assert np.amax(data_from_file[t]["z"]) <= settings.z_max
+            assert data_from_file[t]["z"].size == number_of_exported_particles[i]
+            data_from_file[t].pop("z")
+            assert data.keys() == data_from_file[t].keys()
+            assert (
+                data["radius"][i][exported_particles_indexes[t]].mean()
+                == data_from_file[t]["radius"].mean()
+            )

{PySDM-2.54 → PySDM-2.58}/tests/smoke_tests/kinematic_1d/shipway_and_hill_2012/test_few_steps.py

@@ -23,7 +23,7 @@ def test_few_steps_no_precip(particle_reservoir_depth, plot=False):
         n_sd_per_gridbox=n_sd_per_gridbox,
         dt=30 * si.s,
         dz=60 * si.m,
-        precip=False,
+        precip=True,
         rho_times_w_1=2 * si.m / si.s * si.kg / si.m**3,
     )
     settings.particle_reservoir_depth = particle_reservoir_depth
@@ -80,6 +80,7 @@ def test_few_steps_no_precip(particle_reservoir_depth, plot=False):
     assert max(mean_profile_over_last_steps("activating")) == 0
     assert max(mean_profile_over_last_steps("ripening")) > 0
     assert max(mean_profile_over_last_steps("deactivating")) > 0
+    assert max(output["surface precipitation"]) == 0
 
 
 def test_fixed_thd():

PySDM-2.54/tests/smoke_tests/kinematic_1d/shipway_and_hill_2012/test_1d_exporters.py

@@ -1,146 +0,0 @@
-# pylint: disable = missing-module-docstring,missing-class-docstring,missing-function-docstring
-import os
-import platform
-from tempfile import TemporaryDirectory
-
-import numpy as np
-import pytest
-from open_atmos_jupyter_utils import TemporaryFile
-from PySDM_examples.Shipway_and_Hill_2012 import Settings, Simulation
-from PySDM_examples.utils import readVTK_1d
-
-from PySDM.exporters import NetCDFExporter_1d, VTKExporter_1d, readNetCDF_1d
-from PySDM.physics import si
-
-
-@pytest.fixture(name="simulation_1d")
-def simulation_1d_fixture():
-    n_sd_per_gridbox = 16
-    settings = Settings(
-        n_sd_per_gridbox=n_sd_per_gridbox,
-        dt=60 * si.s,
-        dz=200 * si.m,
-        precip=False,
-        rho_times_w_1=2 * si.m / si.s * si.kg / si.m**3,
-    )
-    settings.t_max = 20 * settings.dt
-    settings.save_spec_and_attr_times = [0 * si.min, 10 * si.min, 20 * si.min]
-    simulation = Simulation(settings)
-    results = simulation.run()
-    return results, settings, simulation
-
-
-@pytest.mark.parametrize(
-    "exclude_particle_reservoir",
-    (
-        False,
-        True,
-    ),
-)
-def test_netcdf_exporter_1d(simulation_1d, exclude_particle_reservoir):
-    # Arrange
-    data = simulation_1d[0].products
-    settings = simulation_1d[1]
-    simulation = simulation_1d[2]
-    nz_export = (
-        int(settings.z_max / settings.dz) if exclude_particle_reservoir else settings.nz
-    )
-
-    # Act
-    file = TemporaryFile(".nc")
-    netcdf_exporter = NetCDFExporter_1d(
-        data,
-        settings,
-        simulation,
-        filename=file.absolute_path,
-        exclude_particle_reservoir=exclude_particle_reservoir,
-    )
-    netcdf_exporter.run()
-    data_from_file = readNetCDF_1d(file.absolute_path)
-
-    # Assert
-    assert data_from_file.products["time"].shape == data["t"].shape
-    assert data_from_file.products["height"].shape == data["z"][-nz_export:].shape
-    assert data_from_file.products["T"].shape == data["T"][-nz_export:, :].shape
-    assert (
-        data_from_file.products["dry spectrum"].shape
-        == data["dry spectrum"][-nz_export:, :, :].shape
-    )
-
-    assert np.amin(data_from_file.products["time"]) == np.amin(data["t"])
-    assert np.amax(data_from_file.products["time"]) == np.amax(data["t"])
-    assert np.amin(data_from_file.products["height"]) == np.amin(data["z"][-nz_export:])
-    assert np.amax(data_from_file.products["height"]) == np.amax(data["z"][-nz_export:])
-    assert data_from_file.products["rhod"].mean() == pytest.approx(
-        data["rhod"][-nz_export:, :].mean(), 1e-6
-    )
-    assert data_from_file.products["wet spectrum"].mean() == pytest.approx(
-        data["wet spectrum"][-nz_export:, :, :].mean(), 1e-6
-    )
-
-    assert data_from_file.settings["precip"] == settings.precip
-    assert data_from_file.settings["kappa"] == settings.kappa
-    assert data_from_file.settings["r_bins_edges"].size == settings.number_of_bins + 1
-
-
-@pytest.mark.skipif(
-    platform.architecture()[0] == "32bit", reason="Not available vtk module!"
-)
-@pytest.mark.parametrize(
-    "exclude_particle_reservoir",
-    (
-        False,
-        True,
-    ),
-)
-def test_vtk_exporter_1d(
-    simulation_1d, exclude_particle_reservoir
-):  # pylint: disable=too-many-locals
-    # Arrange
-    data = simulation_1d[0].attributes
-    settings = simulation_1d[1]
-    z0 = 0.0 if exclude_particle_reservoir else -settings.particle_reservoir_depth
-    exported_particles_indexes = {}
-    number_of_exported_particles = []
-    for i, t in enumerate(settings.save_spec_and_attr_times):
-        exported_particles_indexes[t] = np.where(
-            data["cell origin"][i][0]
-            >= int((z0 + settings.particle_reservoir_depth) / settings.dz)
-        )
-        number_of_exported_particles.append(exported_particles_indexes[t][0].size)
-
-    # Act
-    with TemporaryDirectory() as tmpdir_:
-        tmpdir = tmpdir_ + "/"
-        vtk_exporter = VTKExporter_1d(
-            data,
-            settings,
-            path=tmpdir,
-            exclude_particle_reservoir=exclude_particle_reservoir,
-        )
-        vtk_exporter.run()
-        written_files_list = os.listdir(tmpdir)
-        data_from_file = {}
-        for t in settings.save_spec_and_attr_times:
-            leading_zeros_in_filename = [
-                "0" for i in range(len(str(settings.t_max)) - len(str(t)))
-            ]
-            filename = "time" + "".join(leading_zeros_in_filename) + str(t) + ".vtu"
-            data_from_file[t] = readVTK_1d(tmpdir + filename)
-
-    # Assert
-    for i, t in enumerate(settings.save_spec_and_attr_times):
-        filename_leading_zeros = "".join(
-            ["0" for i in range(len(str(settings.t_max)) - len(str(t)))]
-        )
-        filename = "time" + filename_leading_zeros + str(t) + ".vtu"
-        assert filename in written_files_list
-        assert z0 <= np.amin(data_from_file[t]["z"])
-        assert np.amax(data_from_file[t]["z"]) <= settings.z_max
-        assert data_from_file[t]["z"].size == number_of_exported_particles[i]
-        data_from_file[t].pop("z")
-        assert data.keys() == data_from_file[t].keys()
-        assert (
-            data["radius"][i][exported_particles_indexes[t]].mean()
-            == data_from_file[t]["radius"].mean()
-        )