def test_check_irradiance_limits_qcrad(irradiance_qcrad):
"""Test different input combinations to check_irradiance_limits_qcrad.
Notes
-----
Copyright (c) 2019 SolarArbiter. See the file
LICENSES/SOLARFORECASTARBITER_LICENSE at the top level directory
of this distribution and at `<https://github.com/pvlib/
pvanalytics/blob/master/LICENSES/SOLARFORECASTARBITER_LICENSE>`_
for more information.
"""
expected = irradiance_qcrad
ghi_out_expected = expected['ghi_limit_flag']
ghi_out, dhi_out, dni_out = irradiance.check_irradiance_limits_qcrad(
expected['solar_zenith'], expected['dni_extra'], ghi=expected['ghi'])
assert_series_equal(ghi_out, ghi_out_expected, check_names=False)
assert dhi_out is None
assert dni_out is None
dhi_out_expected = expected['dhi_limit_flag']
ghi_out, dhi_out, dni_out = irradiance.check_irradiance_limits_qcrad(
expected['solar_zenith'],
expected['dni_extra'],
ghi=expected['ghi'],
dhi=expected['dhi'])
assert_series_equal(dhi_out, dhi_out_expected, check_names=False)
dni_out_expected = expected['dni_limit_flag']
ghi_out, dhi_out, dni_out = irradiance.check_irradiance_limits_qcrad(
expected['solar_zenith'], expected['dni_extra'], dni=expected['dni'])
assert_series_equal(dni_out, dni_out_expected, check_names=False)
solar_position = pvlib.solarposition.get_solarposition(data.index, latitude,
longitude)
# %%
# Generate the estimated extraterrestrial radiation for the time series,
# referred to as dni_extra. This is done using the
# :py:func:`pvlib.irradiance.get_extra_radiation` function.
dni_extra = pvlib.irradiance.get_extra_radiation(data.index)
# %%
# Use :py:func:`pvanalytics.quality.irradiance.check_irradiance_limits_qcrad`
# to generate the QCRAD irradiance limit mask
qcrad_limit_mask = check_irradiance_limits_qcrad(
solar_zenith=solar_position['zenith'],
dni_extra=dni_extra,
ghi=data['irradiance_ghi__7981'],
dhi=data['irradiance_dhi__7983'],
dni=data['irradiance_dni__7982'])
# %%
# Plot the 'irradiance_ghi__7981' data stream with its associated QCRAD limit
# mask.
data['irradiance_ghi__7981'].plot()
data.loc[qcrad_limit_mask[0], 'irradiance_ghi__7981'].plot(ls='', marker='.')
plt.legend(labels=["RMIS GHI", "Within QCRAD Limits"], loc="upper left")
plt.xlabel("Date")
plt.ylabel("GHI (W/m^2)")
plt.tight_layout()
plt.show()
# %%