pvlib 0.11.2__py3-none-any.whl → 0.12.1a1__py3-none-any.whl

pvlib/tests/iotools/test_solcast.py

@@ -1,324 +0,0 @@
-from unittest.mock import patch
-import pandas as pd
-import pvlib
-import pytest
-
-
-@pytest.mark.parametrize("endpoint,params,api_key,json_response", [
-    (
-        "live/radiation_and_weather",
-        dict(
-            latitude=-33.856784,
-            longitude=151.215297,
-            output_parameters='dni,ghi'
-        ),
-        "1234",
-        {'estimated_actuals':
-            [{'dni': 836, 'ghi': 561,
-              'period_end': '2023-09-18T05:00:00.0000000Z', 'period': 'PT30M'},
-             {'dni': 866, 'ghi': 643,
-              'period_end': '2023-09-18T04:30:00.0000000Z', 'period': 'PT30M'},
-             {'dni': 890, 'ghi': 713,
-              'period_end': '2023-09-18T04:00:00.0000000Z', 'period': 'PT30M'},
-             {'dni': 909, 'ghi': 768,
-              'period_end': '2023-09-18T03:30:00.0000000Z', 'period': 'PT30M'}]
-         }
-    ),
-])
-def test__get_solcast(requests_mock, endpoint, params, api_key, json_response):
-    mock_url = f"https://api.solcast.com.au/data/{endpoint}?" \
-               f"latitude={params['latitude']}&" \
-               f"longitude={params['longitude']}&" \
-               f"output_parameters={params['output_parameters']}"
-
-    requests_mock.get(mock_url, json=json_response)
-
-    # with variables remapping
-    pd.testing.assert_frame_equal(
-        pvlib.iotools.solcast._get_solcast(
-            endpoint, params, api_key, True
-        ),
-        pvlib.iotools.solcast._solcast2pvlib(
-            pd.DataFrame.from_dict(
-                json_response[list(json_response.keys())[0]])
-        )
-    )
-
-    # no remapping of variables
-    pd.testing.assert_frame_equal(
-        pvlib.iotools.solcast._get_solcast(
-            endpoint, params, api_key, False
-        ),
-        pd.DataFrame.from_dict(
-            json_response[list(json_response.keys())[0]])
-    )
-
-
-@pytest.mark.parametrize("map_variables", [True, False])
-@pytest.mark.parametrize("endpoint,function,params,json_response", [
-    (
-        "live/radiation_and_weather",
-        pvlib.iotools.get_solcast_live,
-        dict(
-            api_key="1234",
-            latitude=-33.856784,
-            longitude=151.215297,
-            output_parameters='dni,ghi'
-        ),
-        {'estimated_actuals':
-            [{'dni': 836, 'ghi': 561,
-              'period_end': '2023-09-18T05:00:00.0000000Z', 'period': 'PT30M'},
-             {'dni': 866, 'ghi': 643,
-              'period_end': '2023-09-18T04:30:00.0000000Z', 'period': 'PT30M'},
-             {'dni': 890, 'ghi': 713,
-              'period_end': '2023-09-18T04:00:00.0000000Z', 'period': 'PT30M'},
-             {'dni': 909, 'ghi': 768,
-              'period_end': '2023-09-18T03:30:00.0000000Z', 'period': 'PT30M'}]
-         }
-    ),
-])
-def test_get_solcast_live(
-    requests_mock, endpoint, function, params, json_response, map_variables
-):
-    mock_url = (
-        f"https://api.solcast.com.au/data/{endpoint}?"
-        f"&latitude={params['latitude']}&"
-        f"longitude={params['longitude']}&"
-        f"output_parameters={params['output_parameters']}&format=json"
-    )
-
-    requests_mock.get(mock_url, json=json_response)
-
-    if map_variables:
-        pd.testing.assert_frame_equal(
-            function(**params, map_variables=map_variables)[0],
-            pvlib.iotools.solcast._solcast2pvlib(
-                pd.DataFrame.from_dict(
-                    json_response[list(json_response.keys())[0]]
-                )
-            ),
-        )
-    else:
-        pd.testing.assert_frame_equal(
-            function(**params, map_variables=map_variables)[0],
-            pd.DataFrame.from_dict(
-                json_response[list(json_response.keys())[0]]
-            ),
-        )
-
-
-@pytest.mark.parametrize("map_variables", [True, False])
-@pytest.mark.parametrize("endpoint,function,params,json_response", [
-    (
-        "tmy/radiation_and_weather",
-        pvlib.iotools.get_solcast_tmy,
-        dict(
-            api_key="1234",
-            latitude=-33.856784,
-            longitude=51.215297
-        ),
-        {'estimated_actuals': [
-            {'dni': 151, 'ghi': 609,
-             'period_end': '2059-01-01T01:00:00.0000000Z', 'period': 'PT60M'},
-            {'dni': 0, 'ghi': 404,
-             'period_end': '2059-01-01T02:00:00.0000000Z', 'period': 'PT60M'},
-            {'dni': 0, 'ghi': 304,
-             'period_end': '2059-01-01T03:00:00.0000000Z', 'period': 'PT60M'},
-            {'dni': 0, 'ghi': 174,
-             'period_end': '2059-01-01T04:00:00.0000000Z', 'period': 'PT60M'},
-            {'dni': 0, 'ghi': 111,
-             'period_end': '2059-01-01T05:00:00.0000000Z', 'period': 'PT60M'}]
-         }
-    ),
-])
-def test_get_solcast_tmy(
-    requests_mock, endpoint, function, params, json_response, map_variables
-):
-
-    mock_url = f"https://api.solcast.com.au/data/{endpoint}?" \
-               f"&latitude={params['latitude']}&" \
-               f"longitude={params['longitude']}&format=json"
-
-    requests_mock.get(mock_url, json=json_response)
-
-    if map_variables:
-        pd.testing.assert_frame_equal(
-            function(**params, map_variables=map_variables)[0],
-            pvlib.iotools.solcast._solcast2pvlib(
-                pd.DataFrame.from_dict(
-                    json_response[list(json_response.keys())[0]]
-                )
-            ),
-        )
-    else:
-        pd.testing.assert_frame_equal(
-            function(**params, map_variables=map_variables)[0],
-            pd.DataFrame.from_dict(
-                json_response[list(json_response.keys())[0]]
-            ),
-        )
-
-
-@pytest.mark.parametrize("in_df,out_df", [
-    (
-        pd.DataFrame(
-            [[942, 843, 1017.4, 30, 7.8, 316, 1010, -2, 4.6, 16.4,
-              '2023-09-20T02:00:00.0000000Z', 'PT30M', 90],
-             [936, 832, 1017.9, 30, 7.9, 316, 996, -14, 4.5, 16.3,
-              '2023-09-20T01:30:00.0000000Z', 'PT30M', 0]],
-            columns=[
-                'dni', 'ghi', 'surface_pressure', 'air_temp', 'wind_speed_10m',
-                'wind_direction_10m', 'gti', 'azimuth', 'dewpoint_temp',
-                'precipitable_water', 'period_end', 'period', 'zenith'],
-            index=pd.RangeIndex(start=0, stop=2, step=1)
-        ),
-        pd.DataFrame(
-            [[9.4200e+02, 8.4300e+02, 1.0174e+05, 3.0000e+01, 7.8000e+00,
-              3.1600e+02, 1.0100e+03, 2.0000e+00, 4.6000e+00, 1.6400e+00, 90],
-             [9.3600e+02, 8.3200e+02, 1.0179e+05, 3.0000e+01, 7.9000e+00,
-              3.1600e+02, 9.9600e+02, 1.4000e+01, 4.5000e+00, 1.6300e+00, 0]],
-            columns=[
-                'dni', 'ghi', 'pressure', 'temp_air', 'wind_speed',
-                'wind_direction', 'poa_global', 'solar_azimuth',
-                'temp_dew', 'precipitable_water', 'solar_zenith'],
-            index=pd.DatetimeIndex(
-                ['2023-09-20 01:45:00+00:00', '2023-09-20 01:15:00+00:00'],
-                dtype='datetime64[ns, UTC]', name='period_mid', freq=None)
-        )
-    )
-])
-def test_solcast2pvlib(in_df, out_df):
-    df = pvlib.iotools.solcast._solcast2pvlib(in_df)
-    pd.testing.assert_frame_equal(df.astype(float), out_df.astype(float))
-
-
-@pytest.mark.parametrize("map_variables", [True, False])
-@pytest.mark.parametrize("endpoint,function,params,json_response", [
-    (
-        "historic/radiation_and_weather",
-        pvlib.iotools.get_solcast_historic,
-        dict(
-            api_key="1234",
-            latitude=-33.856784,
-            longitude=51.215297,
-            start="2023-01-01T08:00",
-            duration="P1D",
-            period="PT1H",
-            output_parameters='dni'
-        ), {'estimated_actuals': [
-            {'dni': 822, 'period_end': '2023-01-01T09:00:00.0000000Z',
-             'period': 'PT60M'},
-            {'dni': 918, 'period_end': '2023-01-01T10:00:00.0000000Z',
-             'period': 'PT60M'},
-            {'dni': 772, 'period_end': '2023-01-01T11:00:00.0000000Z',
-             'period': 'PT60M'},
-            {'dni': 574, 'period_end': '2023-01-01T12:00:00.0000000Z',
-             'period': 'PT60M'},
-            {'dni': 494, 'period_end': '2023-01-01T13:00:00.0000000Z',
-             'period': 'PT60M'}
-        ]}
-    ),
-])
-def test_get_solcast_historic(
-    requests_mock, endpoint, function, params, json_response, map_variables
-):
-    mock_url = f"https://api.solcast.com.au/data/{endpoint}?" \
-               f"&latitude={params['latitude']}&" \
-               f"longitude={params['longitude']}&format=json"
-
-    requests_mock.get(mock_url, json=json_response)
-
-    if map_variables:
-        pd.testing.assert_frame_equal(
-            function(**params, map_variables=map_variables)[0],
-            pvlib.iotools.solcast._solcast2pvlib(
-                pd.DataFrame.from_dict(
-                    json_response[list(json_response.keys())[0]]
-                )
-            ),
-        )
-    else:
-        pd.testing.assert_frame_equal(
-            function(**params, map_variables=map_variables)[0],
-            pd.DataFrame.from_dict(
-                json_response[list(json_response.keys())[0]]
-            ),
-        )
-
-
-@pytest.mark.parametrize("map_variables", [True, False])
-@pytest.mark.parametrize("endpoint,function,params,json_response", [
-    (
-        "forecast/radiation_and_weather",
-        pvlib.iotools.get_solcast_forecast,
-        dict(
-            api_key="1234",
-            latitude=-33.856784,
-            longitude=51.215297,
-            hours="5",
-            period="PT1H",
-            output_parameters='dni'
-        ), {
-            'forecast': [
-                {'dni': 0, 'period_end': '2023-12-13T01:00:00.0000000Z',
-                 'period': 'PT1H'},
-                {'dni': 1, 'period_end': '2023-12-13T02:00:00.0000000Z',
-                 'period': 'PT1H'},
-                {'dni': 2, 'period_end': '2023-12-13T03:00:00.0000000Z',
-                 'period': 'PT1H'},
-                {'dni': 3, 'period_end': '2023-12-13T04:00:00.0000000Z',
-                 'period': 'PT1H'},
-                {'dni': 4, 'period_end': '2023-12-13T05:00:00.0000000Z',
-                 'period': 'PT1H'},
-                {'dni': 5, 'period_end': '2023-12-13T06:00:00.0000000Z',
-                 'period': 'PT1H'}
-            ]}
-    ),
-])
-def test_get_solcast_forecast(
-    requests_mock, endpoint, function, params, json_response, map_variables
-):
-    mock_url = f"https://api.solcast.com.au/data/{endpoint}?" \
-               f"&latitude={params['latitude']}&" \
-               f"longitude={params['longitude']}&format=json"
-
-    requests_mock.get(mock_url, json=json_response)
-
-    if map_variables:
-        pd.testing.assert_frame_equal(
-            function(**params, map_variables=map_variables)[0],
-            pvlib.iotools.solcast._solcast2pvlib(
-                pd.DataFrame.from_dict(
-                    json_response[list(json_response.keys())[0]]
-                )
-            ),
-        )
-    else:
-        pd.testing.assert_frame_equal(
-            function(**params, map_variables=map_variables)[0],
-            pd.DataFrame.from_dict(
-                json_response[list(json_response.keys())[0]]
-            ),
-        )
-
-
-@pytest.mark.parametrize(
-    "function",
-    [
-        pvlib.iotools.get_solcast_forecast,
-        pvlib.iotools.get_solcast_live,
-        pvlib.iotools.get_solcast_tmy,
-        pvlib.iotools.get_solcast_historic,
-    ],
-)
-@patch("requests.api.request")
-def test_raises_exception(mock_response, function):
-    dummy_args = {
-        "latitude": 0,
-        "longitude": 0,
-        "api_key": "",
-    }
-    with patch.object(mock_response, "status_code", return_value=404):
-        with pytest.raises(Exception):
-            function(**dummy_args)
-            mock_response.json.assert_called_once()

pvlib/tests/iotools/test_solrad.py

@@ -1,152 +0,0 @@
-import pandas as pd
-import numpy as np
-from numpy import nan
-
-import pytest
-
-from pvlib.iotools import solrad
-from ..conftest import DATA_DIR, assert_frame_equal, RERUNS, RERUNS_DELAY
-
-
-testfile = DATA_DIR / 'abq19056.dat'
-testfile_mad = DATA_DIR / 'msn19056.dat'
-https_testfile = ('https://gml.noaa.gov/aftp/data/radiation/solrad/msn/'
-                  '2019/msn19056.dat')
-
-columns = [
-    'year', 'julian_day', 'month', 'day', 'hour', 'minute', 'decimal_time',
-    'solar_zenith', 'ghi', 'ghi_flag', 'dni', 'dni_flag', 'dhi', 'dhi_flag',
-    'uvb', 'uvb_flag', 'uvb_temp', 'uvb_temp_flag', 'std_dw_psp', 'std_direct',
-    'std_diffuse', 'std_uvb']
-index = pd.DatetimeIndex(['2019-02-25 00:00:00',
-                          '2019-02-25 00:01:00',
-                          '2019-02-25 00:02:00',
-                          '2019-02-25 00:03:00'],
-                         freq=None).tz_localize('UTC')
-values = np.array([
-    [2.019e+03, 5.600e+01, 2.000e+00, 2.500e+01, 0.000e+00, 0.000e+00,
-        0.000e+00, 7.930e+01, 1.045e+02, 0.000e+00, 6.050e+01, 0.000e+00,
-        9.780e+01, 0.000e+00, 5.900e+00, 0.000e+00, 4.360e+01, 0.000e+00,
-        3.820e-01, 2.280e+00, 4.310e-01, 6.000e-02],
-    [2.019e+03, 5.600e+01, 2.000e+00, 2.500e+01, 0.000e+00, 1.000e+00,
-        1.700e-02, 7.949e+01, 1.026e+02, 0.000e+00, 5.970e+01, 0.000e+00,
-        9.620e+01, 0.000e+00, 5.700e+00, 0.000e+00, 4.360e+01, 0.000e+00,
-        7.640e-01, 1.800e+00, 4.310e-01, 6.000e-02],
-    [2.019e+03, 5.600e+01, 2.000e+00, 2.500e+01, 0.000e+00, 2.000e+00,
-        3.300e-02, 7.968e+01, 1.021e+02, 0.000e+00, 6.580e+01, 0.000e+00,
-        9.480e+01, 0.000e+00, 5.500e+00, 0.000e+00, 4.360e+01, 0.000e+00,
-        3.820e-01, 4.079e+00, 3.230e-01, 6.000e-02],
-    [2.019e+03, 5.600e+01, 2.000e+00, 2.500e+01, 0.000e+00, 3.000e+00,
-        5.000e-02, 7.987e+01, 1.026e+02, 0.000e+00, 7.630e+01, 0.000e+00,
-        nan, 0.000e+00, 5.300e+00, 0.000e+00, 4.360e+01, 0.000e+00,
-        5.090e-01, 1.920e+00, 2.150e-01, 5.000e-02]])
-dtypes = [
-    'int64', 'int64', 'int64', 'int64', 'int64', 'int64', 'float64',
-    'float64', 'float64', 'int64', 'float64', 'int64', 'float64', 'int64',
-    'float64', 'int64', 'float64', 'int64', 'float64', 'float64',
-    'float64', 'float64']
-
-columns_mad = [
-    'year', 'julian_day', 'month', 'day', 'hour', 'minute', 'decimal_time',
-    'solar_zenith', 'ghi', 'ghi_flag', 'dni', 'dni_flag', 'dhi', 'dhi_flag',
-    'uvb', 'uvb_flag', 'uvb_temp', 'uvb_temp_flag', 'dpir', 'dpir_flag',
-    'dpirc', 'dpirc_flag', 'dpird', 'dpird_flag', 'std_dw_psp',
-    'std_direct', 'std_diffuse', 'std_uvb', 'std_dpir', 'std_dpirc',
-    'std_dpird']
-values_mad = np.array([
-    [ 2.019e+03,  5.600e+01,  2.000e+00,  2.500e+01,  0.000e+00,
-      0.000e+00,  0.000e+00,  9.428e+01, -2.300e+00,  0.000e+00,
-      0.000e+00,  0.000e+00,  4.000e-01,  0.000e+00,        nan,
-      1.000e+00,        nan,  1.000e+00,  1.872e+02,  0.000e+00,
-      2.656e+02,  0.000e+00,  2.653e+02,  0.000e+00,  0.000e+00,
-      0.000e+00,  0.000e+00,        nan,  2.000e-03,  2.600e+01,
-      2.700e+01],
-    [ 2.019e+03,  5.600e+01,  2.000e+00,  2.500e+01,  0.000e+00,
-      1.000e+00,  1.700e-02,  9.446e+01, -2.300e+00,  0.000e+00,
-      0.000e+00,  0.000e+00,  1.000e-01,  0.000e+00,        nan,
-      1.000e+00,        nan,  1.000e+00,  1.882e+02,  0.000e+00,
-      2.656e+02,  0.000e+00,  2.653e+02,  0.000e+00,  1.330e-01,
-      1.280e-01,  2.230e-01,        nan,  1.000e-03,  2.600e+01,
-      7.200e+01],
-    [ 2.019e+03,  5.600e+01,  2.000e+00,  2.500e+01,  0.000e+00,
-      2.000e+00,  3.300e-02,  9.464e+01, -2.700e+00,  0.000e+00,
-     -2.000e-01,  0.000e+00,  0.000e+00,  0.000e+00,        nan,
-      1.000e+00,        nan,  1.000e+00,  1.876e+02,  0.000e+00,
-      2.656e+02,  0.000e+00,  2.653e+02,  0.000e+00,  0.000e+00,
-      2.570e-01,  0.000e+00,        nan,  1.000e-03,  2.400e+01,
-      4.200e+01],
-    [ 2.019e+03,  5.600e+01,  2.000e+00,  2.500e+01,  0.000e+00,
-      3.000e+00,  5.000e-02,  9.482e+01, -2.500e+00,  0.000e+00,
-      4.000e-01,  0.000e+00,  0.000e+00,  0.000e+00,        nan,
-      1.000e+00,        nan,  1.000e+00,  1.873e+02,  0.000e+00,
-      2.656e+02,  0.000e+00,  2.653e+02,  0.000e+00,  2.660e-01,
-      3.850e-01,  0.000e+00,        nan,  1.000e-03,  2.600e+01,
-      4.800e+01]])
-dtypes_mad = [
-    'int64', 'int64', 'int64', 'int64', 'int64', 'int64', 'float64', 'float64',
-    'float64', 'int64', 'float64', 'int64', 'float64', 'int64', 'float64',
-    'int64', 'float64', 'int64', 'float64', 'int64', 'float64', 'int64',
-    'float64', 'int64', 'float64', 'float64', 'float64', 'float64', 'float64',
-    'float64', 'float64']
-meta = {'station_name': 'Albuquerque', 'latitude': 35.03796,
-        'longitude': -106.62211, 'altitude': 1617, 'TZ': -7}
-meta_mad = {'station_name': 'Madison', 'latitude': 43.07250,
-            'longitude': -89.41133, 'altitude': 271, 'TZ': -6}
-
-
-@pytest.mark.parametrize('testfile,index,columns,values,dtypes,meta', [
-    (testfile, index, columns, values, dtypes, meta),
-    (testfile_mad, index, columns_mad, values_mad, dtypes_mad, meta_mad)
-])
-def test_read_solrad(testfile, index, columns, values, dtypes, meta):
-    expected = pd.DataFrame(values, columns=columns, index=index)
-    for (col, _dtype) in zip(expected.columns, dtypes):
-        expected[col] = expected[col].astype(_dtype)
-    out, m = solrad.read_solrad(testfile)
-    assert_frame_equal(out, expected)
-    assert m == meta
-
-
-@pytest.mark.remote_data
-@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
-def test_read_solrad_https():
-    # Test reading of https files.
-    # If this test begins failing, SOLRAD's data structure or data
-    # archive may have changed.
-    local_data, _ = solrad.read_solrad(testfile_mad)
-    remote_data, _ = solrad.read_solrad(https_testfile)
-    # local file only contains four rows to save space
-    assert_frame_equal(local_data, remote_data.iloc[:4])
-
-
-@pytest.mark.remote_data
-@pytest.mark.parametrize('testfile, station', [
-    (testfile, 'abq'),
-    (testfile_mad, 'msn'),
-])
-def test_get_solrad(testfile, station):
-    df, meta = solrad.get_solrad(station, "2019-02-25", "2019-02-25")
-
-    assert meta['station'] == station
-    assert isinstance(meta['filenames'], list)
-
-    assert len(df) == 1440
-    assert df.index[0] == pd.to_datetime('2019-02-25 00:00+00:00')
-    assert df.index[-1] == pd.to_datetime('2019-02-25 23:59+00:00')
-
-    expected, _ = solrad.read_solrad(testfile)
-    actual = df.reindex(expected.index)
-    # ABQ test file has an unexplained NaN in row 4; just verify first 3 rows
-    assert_frame_equal(actual.iloc[:3], expected.iloc[:3])
-
-
-@pytest.mark.remote_data
-def test_get_solrad_missing_day():
-    # data availability begins for ABQ on 2002-02-01 (DOY 32), so requesting
-    # data before that will raise a warning
-    message = 'The following file was not found: abq/2002/abq02031.dat'
-    with pytest.warns(UserWarning, match=message):
-        df, meta = solrad.get_solrad('abq', '2002-01-31', '2002-02-01')
-
-    # but the data for 2022-02-01 is still returned
-    assert not df.empty

pvlib/tests/iotools/test_srml.py

@@ -1,124 +0,0 @@
-from numpy import isnan
-import pandas as pd
-import pytest
-
-from pvlib.iotools import srml
-from ..conftest import (DATA_DIR, RERUNS, RERUNS_DELAY, assert_index_equal,
-                        assert_frame_equal)
-
-srml_testfile = DATA_DIR / 'SRML-day-EUPO1801.txt'
-
-
-def test_read_srml():
-    srml.read_srml(srml_testfile)
-
-
-@pytest.mark.remote_data
-@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
-def test_read_srml_remote():
-    srml.read_srml(
-        'http://solardata.uoregon.edu/download/Archive/EUPO1801.txt'
-    )
-
-
-def test_read_srml_columns_exist():
-    data = srml.read_srml(srml_testfile)
-    assert 'ghi_0' in data.columns
-    assert 'ghi_0_flag' in data.columns
-    assert 'dni_1' in data.columns
-    assert 'dni_1_flag' in data.columns
-    assert '7008' in data.columns
-    assert '7008_flag' in data.columns
-
-
-def test_read_srml_map_variables_false():
-    data = srml.read_srml(srml_testfile, map_variables=False)
-    assert '1000' in data.columns
-    assert '1000_flag' in data.columns
-    assert '2010' in data.columns
-    assert '2010_flag' in data.columns
-    assert '7008' in data.columns
-    assert '7008_flag' in data.columns
-
-
-def test_read_srml_nans_exist():
-    data = srml.read_srml(srml_testfile)
-    assert isnan(data['dni_0'].iloc[1119])
-    assert data['dni_0_flag'].iloc[1119] == 99
-
-
-@pytest.mark.parametrize('url,year,month', [
-    ('http://solardata.uoregon.edu/download/Archive/EUPO1801.txt',
-     2018, 1),
-    ('http://solardata.uoregon.edu/download/Archive/EUPO1612.txt',
-     2016, 12),
-])
-@pytest.mark.remote_data
-@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
-def test_read_srml_dt_index(url, year, month):
-    data = srml.read_srml(url)
-    start = pd.Timestamp(f'{year:04d}{month:02d}01 00:00')
-    start = start.tz_localize('Etc/GMT+8')
-    end = pd.Timestamp(f'{year:04d}{month:02d}31 23:59')
-    end = end.tz_localize('Etc/GMT+8')
-    assert data.index[0] == start
-    assert data.index[-1] == end
-    assert (data.index[59::60].minute == 59).all()
-    assert str(year) not in data.columns
-
-
-@pytest.mark.parametrize('column,expected', [
-    ('1001', 'ghi_1'),
-    ('7324', '7324'),
-    ('2001', '2001'),
-    ('2017', 'dni_7')
-])
-def test__map_columns(column, expected):
-    assert srml._map_columns(column) == expected
-
-
-@pytest.mark.remote_data
-@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
-def test_get_srml():
-    url = 'http://solardata.uoregon.edu/download/Archive/EUPO1801.txt'
-    file_data = srml.read_srml(url)
-    requested, _ = srml.get_srml(station='EU', start='2018-01-01',
-                                 end='2018-01-31')
-    assert_frame_equal(file_data, requested)
-
-
-@pytest.mark.remote_data
-@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
-def test_get_srml_hourly():
-    data, meta = data, meta = srml.get_srml(station='CD', start='1986-04-01',
-                                            end='1986-05-31', filetype='PH')
-    expected_index = pd.date_range(start='1986-04-01', end='1986-05-31 23:59',
-                                   freq='1h', tz='Etc/GMT+8')
-    assert_index_equal(data.index, expected_index)
-
-
-@pytest.mark.remote_data
-@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
-def test_get_srml_minute():
-    data_read = srml.read_srml(srml_testfile)
-    data_get, meta = srml.get_srml(station='EU', start='2018-01-01',
-                                   end='2018-01-31', filetype='PO')
-    expected_index = pd.date_range(start='2018-01-01', end='2018-01-31 23:59',
-                                   freq='1min', tz='Etc/GMT+8')
-    assert_index_equal(data_get.index, expected_index)
-    assert all(c in data_get.columns for c in data_read.columns)
-    # Check that all indices in example file are present in remote file
-    assert data_read.index.isin(data_get.index).all()
-    assert meta['station'] == 'EU'
-    assert meta['filetype'] == 'PO'
-    assert meta['filenames'] == ['EUPO1801.txt']
-
-
-@pytest.mark.remote_data
-@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
-def test_get_srml_nonexisting_month_warning():
-    with pytest.warns(UserWarning, match='file was not found: EUPO0912.txt'):
-        # Request data for a period where not all files exist
-        # Eugene (EU) station started reporting 1-minute data in January 2010
-        data, meta = data, meta = srml.get_srml(
-            station='EU', start='2009-12-01', end='2010-01-31', filetype='PO')

pvlib/tests/iotools/test_surfrad.py

@@ -1,75 +0,0 @@
-import pandas as pd
-import pytest
-
-from pvlib.iotools import surfrad
-from ..conftest import DATA_DIR, RERUNS, RERUNS_DELAY
-
-testfile = DATA_DIR / 'surfrad-slv16001.dat'
-network_testfile = ('ftp://aftp.cmdl.noaa.gov/data/radiation/surfrad/'
-                    'Alamosa_CO/2016/slv16001.dat')
-https_testfile = ('https://gml.noaa.gov/aftp/data/radiation/surfrad/'
-                  'Alamosa_CO/2016/slv16001.dat')
-
-
-@pytest.mark.remote_data
-@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
-def test_read_surfrad_network():
-    # If this test begins failing, SURFRAD's data structure or data
-    # archive may have changed.
-    local_data, _ = surfrad.read_surfrad(testfile)
-    network_data, _ = surfrad.read_surfrad(network_testfile)
-    assert local_data.equals(network_data)
-
-
-@pytest.mark.remote_data
-@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
-def test_read_surfrad_https():
-    # Test reading of https files.
-    # If this test begins failing, SURFRAD's data structure or data
-    # archive may have changed.
-    local_data, _ = surfrad.read_surfrad(testfile)
-    network_data, _ = surfrad.read_surfrad(https_testfile)
-    assert local_data.equals(network_data)
-
-
-def test_read_surfrad_columns_no_map():
-    data, _ = surfrad.read_surfrad(testfile, map_variables=False)
-    assert 'zen' in data.columns
-    assert 'temp' in data.columns
-    assert 'par' in data.columns
-    assert 'pressure' in data.columns
-
-
-def test_read_surfrad_columns_map():
-    data, _ = surfrad.read_surfrad(testfile)
-    assert 'solar_zenith' in data.columns
-    assert 'ghi' in data.columns
-    assert 'ghi_flag' in data.columns
-    assert 'dni' in data.columns
-    assert 'dni_flag' in data.columns
-    assert 'dhi' in data.columns
-    assert 'dhi_flag' in data.columns
-    assert 'wind_direction' in data.columns
-    assert 'wind_direction_flag' in data.columns
-    assert 'wind_speed' in data.columns
-    assert 'wind_speed_flag' in data.columns
-    assert 'temp_air' in data.columns
-    assert 'temp_air_flag' in data.columns
-
-
-def test_format_index():
-    start = pd.Timestamp('20160101 00:00')
-    expected = pd.date_range(start=start, periods=1440, freq='1min', tz='UTC')
-    actual, _ = surfrad.read_surfrad(testfile)
-    assert actual.index.equals(expected)
-
-
-def test_read_surfrad_metadata():
-    expected = {'name': 'Alamosa',
-                'latitude': 37.70,
-                'longitude': 105.92,
-                'elevation': 2317,
-                'surfrad_version': 1,
-                'tz': 'UTC'}
-    _, metadata = surfrad.read_surfrad(testfile)
-    assert metadata == expected