File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff 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+
134135def 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 )
You can’t perform that action at this time.
0 commit comments