Skip to content

Commit 200af63

Browse files
committed
fix test assert statement
1 parent 0f519c5 commit 200af63

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/spectrum/test_irradiance.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,7 @@ def test_average_photon_energy_zero_irr():
131131
assert_allclose(out_1, expected_1, atol=1e-3)
132132
assert_allclose(out_2, expected_2, atol=1e-3)
133133

134+
134135
def test_average_photon_energy_nan_irr():
135136
# test for handling NaN input
136137

@@ -144,11 +145,11 @@ def test_average_photon_energy_nan_irr():
144145
spectra_series_singlenan.loc[315.0] = np.nan
145146
spectra_series_allnan = spectra_series_nan*np.nan
146147

147-
out = [
148+
output = [
148149
spectrum.average_photon_energy(spectra_df_singlenan),
149150
spectrum.average_photon_energy(spectra_df_allnan),
150151
spectrum.average_photon_energy(spectra_series_singlenan),
151152
spectrum.average_photon_energy(spectra_series_allnan)
152153
]
153154

154-
assert all(np.isnan(out))
155+
assert (all(np.isnan(out)) for out in output)

0 commit comments

Comments
 (0)