def test_soiling_hsu_defaults(rainfall_input, expected_output_1):
"""
Test Soiling HSU function with default deposition velocity and default rain
accumulation period.
"""
result = soiling_hsu(rainfall=rainfall_input,
cleaning_threshold=0.5,
tilt=0.0,
pm2_5=1.0,
pm10=2.0)
assert np.allclose(result.values, expected_output_1)
def test_soiling_hsu_no_cleaning(rainfall_input, expected_output):
"""Test Soiling HSU function"""
rainfall = rainfall_input
pm2_5 = 1.0
pm10 = 2.0
depo_veloc = {'2_5': 1.0, '10': 1.0}
tilt = 0.
expected_no_cleaning = expected_output
result = soiling_hsu(rainfall=rainfall, cleaning_threshold=10., tilt=tilt,
pm2_5=pm2_5, pm10=pm10, depo_veloc=depo_veloc,
rain_accum_period=pd.Timedelta('1h'))
assert_series_equal(result, expected_no_cleaning)
def test_soiling_hsu(rainfall_input, expected_output_2):
"""Test Soiling HSU function"""
rainfall = rainfall_input
pm2_5 = 1.0
pm10 = 2.0
depo_veloc = {'2_5': 1.0, '10': 1.0}
tilt = 0.
expected = expected_output_2
# three cleaning events at 4:00-6:00, 8:00-11:00, and 17:00-20:00
result = soiling_hsu(rainfall=rainfall, cleaning_threshold=0.5, tilt=tilt,
pm2_5=pm2_5, pm10=pm10, depo_veloc=depo_veloc,
rain_accum_period=pd.Timedelta('3h'))
assert_series_equal(result, expected)