Add dataset option to get_era5

Author: AdamRJensen

pvlib/iotools/era5.py

@@ -58,6 +58,7 @@ def _m_to_cm(m):
 
 
 def get_era5(latitude, longitude, start, end, variables, api_key,
+             dataset="reanalysis-era5-single-levels-timeseries",
              map_variables=True, timeout=60,
              url='https://cds.climate.copernicus.eu/api/retrieve/v1/'):
     """
@@ -84,6 +85,10 @@ def get_era5(latitude, longitude, start, end, variables, api_key,
         See [1]_ for additional options.
     api_key : str
         ECMWF CDS API key.
+    dataset : str, default "reanalysis-era5-single-levels-timeseries"
+        The dataset to query.  May be either
+        "reanalysis-era5-single-levels-timeseries" or
+        "reanalysis-era5-land-timeseries".
     map_variables : bool, default True
         When true, renames columns of the DataFrame to pvlib variable names
         where applicable. Also converts units of some variables. See variable
@@ -137,7 +142,7 @@ def _to_utc_dt_notz(dt):
             "data_format": "csv"
         }
     }
-    slug = "processes/reanalysis-era5-single-levels-timeseries/execution"
+    slug = f"processes/{dataset}/execution"
     response = requests.post(url + slug, json=params, headers=headers,
                              timeout=timeout)
     submission_response = response.json()

tests/iotools/test_era5.py

@@ -48,6 +48,17 @@ def test_get_era5(params, expected):
     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_land(params, expected):
+    params['dataset'] = "reanalysis-era5-land-timeseries"
+    df, meta = pvlib.iotools.get_era5(**params)
+    assert meta['longitude'] == -80.0
+    assert meta['latitude'] == 40.0
+    assert pd.testing.assert_index_equal(df.index, expected.index)
+
+
 @requires_ecmwf_credentials
 @pytest.mark.remote_data
 @pytest.mark.flaky(reruns=RERUNS, reruns_delay=RERUNS_DELAY)