@@ -154,11 +154,11 @@ def test_init_site_observations(
154154 assert mock_chk_post .call_count == 2
155155
156156
157- def request_data_test (mocker , exception , test_site , test_data ):
158- get_func = mocker .patch (
159- 'solarforecastarbiter.io.reference_observations.srml.iotools.'
160- 'read_srml_month_from_solardat' )
161- get_func . return_value = test_data
157+ def test_request_data (mocker , test_site , test_data ):
158+ mocked_iotools = mocker .patch (
159+ 'solarforecastarbiter.io.reference_observations.srml.iotools' )
160+ mocked_iotools . read_srml_month_from_solardat = mocker . MagicMock (
161+ return_value = test_data )
162162 data = srml .request_data (test_site , 1 , 1 )
163163 assert_frame_equal (data , test_data )
164164
@@ -167,14 +167,13 @@ def request_data_test(mocker, exception, test_site, test_data):
167167 pd .errors .EmptyDataError ,
168168 error .URLError ,
169169])
170- def request_data_test_warnings (mocker , exception , test_site ):
170+ def test_request_data_warnings (mocker , exception , test_site ):
171+ mocked_iotools = mocker .patch (
172+ 'solarforecastarbiter.io.reference_observations.srml.iotools' )
173+ mocked_iotools .read_srml_month_from_solardat = mocker .MagicMock (
174+ side_effect = exception ('error' ))
171175 logger = mocker .patch (
172- 'solarforecastarbiter.io.reference_observations.srml.iotools.'
173- 'logger.warning' )
174- get_func = mocker .patch (
175- 'solarforecastarbiter.io.reference_observations.srml.iotools.'
176- 'read_srml_month_from_solardat' )
177- get_func .side_effect = exception ()
176+ 'solarforecastarbiter.io.reference_observations.srml.logger' )
178177 data = srml .request_data (test_site , 1 , 1 )
179- assert logger .call_count == 5
178+ assert logger .warning . call_count == 3
180179 assert data is None
0 commit comments