unhardcode test value for number of aero_modes in the aero_dist

open-atmos · May 17, 2023 · b072960 · b072960
1 parent 79d9aea
commit b072960
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_scenario.py b/tests/test_scenario.py
@@ -215,4 +215,4 @@ def test_time_varying_aero_multimode(key):
  )
  for i in range(scenario.emissions_n_times):
  dist = scenario.emissions(aero_data, i)
- assert dist.n_mode == 2
+ assert dist.n_mode == len(list(scenario_ctor_arg[key][2]["dist"][i])[0])