@@ -78,10 +78,10 @@ def test_average_photon_energy_series():
78
78
# test that the APE is calculated correctly with single spectrum
79
79
# series input
80
80
81
- si = spectrum .get_reference_spectra ()
82
- si = si ['global' ]
81
+ spectra = spectrum .get_reference_spectra ()
82
+ spectra = spectra ['global' ]
83
83
84
- ape = spectrum .average_photon_energy (si )
84
+ ape = spectrum .average_photon_energy (spectra )
85
85
86
86
expected = 1.45017
87
87
@@ -92,50 +92,50 @@ def test_average_photon_energy_dataframe():
92
92
# test that the APE is calculated correctly with multiple spectra
93
93
# dataframe input
94
94
95
- si = spectrum .get_reference_spectra ().T
96
- ape = spectrum .average_photon_energy (si )
95
+ spectra = spectrum .get_reference_spectra ().T
96
+ ape = spectrum .average_photon_energy (spectra )
97
97
expected = [1.36848 , 1.45017 , 1.40885 ]
98
98
assert_allclose (ape , expected , rtol = 1e-4 )
99
99
100
100
101
101
def test_average_photon_energy_invalid_type ():
102
102
# test that spectrum argument is either a pandas Series or dataframe
103
- si = 5
103
+ spectra = 5
104
104
with pytest .raises (TypeError , match = 'must be either a pandas Series or'
105
105
' DataFrame' ):
106
- spectrum .average_photon_energy (si )
106
+ spectrum .average_photon_energy (spectra )
107
107
108
108
109
109
def test_average_photon_energy_neg_irr_series ():
110
110
# test for handling of negative spectral irradiance values with a
111
111
# pandas Series input
112
112
113
- si = spectrum .get_reference_spectra ()['global' ]* - 1
113
+ spectra = spectrum .get_reference_spectra ()['global' ]* - 1
114
114
115
115
with pytest .raises (ValueError , match = 'must be positive' ):
116
- spectrum .average_photon_energy (si )
116
+ spectrum .average_photon_energy (spectra )
117
117
118
118
119
119
def test_average_photon_energy_neg_irr_dataframe ():
120
120
# test for handling of negative spectral irradiance values with a
121
121
# pandas DataFrame input
122
122
123
- si = spectrum .get_reference_spectra ().T * - 1
123
+ spectra = spectrum .get_reference_spectra ().T * - 1
124
124
125
125
with pytest .raises (ValueError , match = 'must be positive' ):
126
- spectrum .average_photon_energy (si )
126
+ spectrum .average_photon_energy (spectra )
127
127
128
128
129
129
def test_average_photon_energy_zero_irr ():
130
130
# test for handling of zero spectral irradiance values with
131
131
# pandas DataFrame and pandas Series input
132
132
133
- si_df_zero = spectrum .get_reference_spectra ().T
134
- si_df_zero .iloc [1 ] = 0
135
- si_series_zero = spectrum .get_reference_spectra ()['global' ]* 0
133
+ spectra_df_zero = spectrum .get_reference_spectra ().T
134
+ spectra_df_zero .iloc [1 ] = 0
135
+ spectra_series_zero = spectrum .get_reference_spectra ()['global' ]* 0
136
136
137
- out_1 = spectrum .average_photon_energy (si_df_zero )
138
- out_2 = spectrum .average_photon_energy (si_series_zero )
137
+ out_1 = spectrum .average_photon_energy (spectra_df_zero )
138
+ out_2 = spectrum .average_photon_energy (spectra_series_zero )
139
139
140
140
expected_1 = np .array ([1.36848 , np .nan , 1.40885 ])
141
141
expected_2 = np .nan
0 commit comments