Add test for test_bin_average_comp (#324)

Author: jcurtis2

tests/test_aero_dist.py

@@ -11,6 +11,7 @@
 import pytest
 
 import PyPartMC as ppmc
+from PyPartMC import si
 
 from .test_aero_data import AERO_DATA_CTOR_ARG_MINIMAL
 from .test_aero_mode import (
@@ -31,6 +32,25 @@
     AERO_MODE_CTOR_LOG_NORMAL_COAGULATION,
 ]
 
+AERO_DIST_CTOR_ARG_AVERAGE = [
+    {
+        "test_mode_A": {
+            "mass_frac": [{"SO4": [1]}],
+            "diam_type": "geometric",
+            "mode_type": "mono",
+            "num_conc": 1e12 / si.m**3,
+            "diam": 2 * si.um,
+        },
+        "test_mode_B": {
+            "mass_frac": [{"BC": [1]}],
+            "diam_type": "geometric",
+            "mode_type": "mono",
+            "num_conc": 1e12 / si.m**3,
+            "diam": 0.2 * si.um,
+        },
+    }
+]
+
 
 @pytest.fixture(name="sut_minimal")
 def sut_minimal_fixture():

tests/test_aero_state.py

@@ -12,7 +12,11 @@
 import PyPartMC as ppmc
 
 from .test_aero_data import AERO_DATA_CTOR_ARG_FULL, AERO_DATA_CTOR_ARG_MINIMAL
-from .test_aero_dist import AERO_DIST_CTOR_ARG_FULL, AERO_DIST_CTOR_ARG_MINIMAL
+from .test_aero_dist import (
+    AERO_DIST_CTOR_ARG_AVERAGE,
+    AERO_DIST_CTOR_ARG_FULL,
+    AERO_DIST_CTOR_ARG_MINIMAL,
+)
 from .test_env_state import ENV_STATE_CTOR_ARG_MINIMAL
 
 AERO_STATE_CTOR_ARG_MINIMAL = 44, "nummass_source"
@@ -42,6 +46,17 @@ def sut_full():
     return sut
 
 
+@pytest.fixture
+def sut_average():
+    aero_data = ppmc.AeroData(AERO_DATA_CTOR_ARG_FULL)
+    aero_dist = ppmc.AeroDist(aero_data, AERO_DIST_CTOR_ARG_AVERAGE)
+    sut = ppmc.AeroState(aero_data, *AERO_STATE_CTOR_ARG_MINIMAL)
+    _ = sut.dist_sample(aero_dist, 1.0, 0.0, True, True)
+    aero_data = None
+    gc.collect()
+    return sut
+
+
 class TestAeroState:
     @staticmethod
     def test_ctor():
@@ -260,15 +275,25 @@ def test_mixing_state(sut_minimal):  # pylint: disable=redefined-outer-name
         assert len(mixing_state) == 3
 
     @staticmethod
-    def test_bin_average_comp(sut_minimal):  # pylint: disable=redefined-outer-name
+    @pytest.mark.parametrize("n_bin", (1, 123))
+    def test_bin_average_comp(
+        sut_average, n_bin
+    ):  # pylint: disable=redefined-outer-name
         # arrange
-        bin_grid = ppmc.BinGrid(123, "log", 1e-9, 1e-4)
+        bin_grid = ppmc.BinGrid(n_bin, "log", 1e-9, 1e-4)
 
         # act
-        sut_minimal.bin_average_comp(bin_grid)
+        sut_average.bin_average_comp(bin_grid)
+        so4_masses = np.array(sut_average.masses(include=["SO4"]))
+        bc_masses = np.array(sut_average.masses(include=["BC"]))
 
         # assert
-        # TODO #179
+        if n_bin == 1:
+            assert np.all(
+                np.isclose(so4_masses / bc_masses, so4_masses[0] / bc_masses[0])
+            )
+        else:
+            assert np.logical_xor(so4_masses > 0, bc_masses > 0).all()
 
     @staticmethod
     def test_get_particle(sut_minimal):  # pylint: disable=redefined-outer-name