pvlib-python/tests/iotools/test_ecmwf.py at f34309ed87331d8fe2b2006a03102b0da70405a1 · kandersolar/pvlib-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
"""
tests for pvlib/iotools/ecmwf.py
"""

import pandas as pd
import pytest
import pvlib
import os
from tests.conftest import RERUNS, RERUNS_DELAY, requires_ecmwf_credentials


@pytest.fixture
def params():
    api_key = os.environ["ECMWF_API_KEY"]

    return {
        'latitude': 40.01, 'longitude': -80.01,
        'start': '2020-06-01', 'end': '2020-06-01',
        'variables': ['ghi', 'temp_air'],
        'api_key': api_key,
    }


@pytest.fixture
def expected():
    index = pd.date_range("2020-06-01 00:00", "2020-06-01 23:59", freq="h",
                          tz="UTC")
    index.name = 'valid_time'
    temp_air = [16.6, 15.2, 13.5, 11.2, 10.8, 9.1, 7.3, 6.8, 7.6, 7.4, 8.5,
                8.1, 9.8, 11.5, 14.1, 17.4, 18.3, 20., 20.7, 20.9, 21.5,
                21.6, 21., 20.7]
    ghi = [153., 18.4, 0., 0., 0., 0., 0., 0., 0., 0., 0., 60., 229.5,
           427.8, 620.1, 785.5, 910.1, 984.2, 1005.9, 962.4, 844.1, 685.2,
           526.9, 331.4]
    df = pd.DataFrame({'temp_air': temp_air, 'ghi': ghi}, index=index)
    return df


@requires_ecmwf_credentials
@pytest.mark.remote_data
@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
def test_get_era5(params, expected):
    df, meta = pvlib.iotools.get_era5(**params)
    pd.testing.assert_frame_equal(df, expected, check_freq=False, atol=0.1)
    assert meta['longitude'] == -80.0
    assert meta['latitude'] == 40.0
    assert isinstance(meta['jobID'], str)


@requires_ecmwf_credentials
@pytest.mark.remote_data
@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
def test_get_era5_map_variables(params, expected):
    df, meta = pvlib.iotools.get_era5(**params, map_variables=False)
    expected = expected.rename(columns={'temp_air': 't2m', 'ghi': 'ssrd'})
    expected['t2m'] -= 273.15  # apply unit conversions manually
    expected['ssrd'] /= 3600
    pd.testing.assert_frame_equal(df, expected, check_freq=False, atol=0.1)
    assert meta['longitude'] == -80.0
    assert meta['latitude'] == 40.0
    assert isinstance(meta['jobID'], str)


@requires_ecmwf_credentials
@pytest.mark.remote_data
@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
def test_get_era5_error(params):
    params['variables'] = ['nonexistent']
    match = 'Request failed. Please check the ECMWF website'
    with pytest.raises(Exception, match=match):
        df, meta = pvlib.iotools.get_era5(**params)


@requires_ecmwf_credentials
@pytest.mark.remote_data
@pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)
def test_get_era5_timeout(params):
    match = 'Request timed out. Try increasing'
    with pytest.raises(Exception, match=match):
        df, meta = pvlib.iotools.get_era5(**params, timeout=1)