Merge pull request #3 from atsangarides/wind

Wind

Author: atsangarides

README.md

@@ -16,7 +16,7 @@ from pyngeso import NgEso
 resource = "historic-day-ahead-demand-forecast"
 date_col = "TARGETDATE"
 start_date = "2018-01-01"
-end_date = "2018-01-02"
+end_date = "2018-01-01"
 
 client = NgEso(resource)
 # returns content of response
@@ -29,3 +29,5 @@ r: bytes = client.query(date_col=date_col, start_date=start_date, end_date=end_d
 * `day-ahead-demand-forecast`
 * `historic-2day-ahead-demand-forecast`
 * `2day-ahead-demand-forecast`
+* `historic-day-ahead-wind-forecast`
+* `day-ahead-wind-forecast`

pyngeso/__init__.py

@@ -1,3 +1,3 @@
 from .pyngeso import NgEso
 
-__version__ = '0.2.0'
+__version__ = '0.2.1'

pyngeso/resources.py

@@ -16,5 +16,13 @@
     "2day-ahead-demand-forecast": {
         "id": "cda26f27-4bb6-4632-9fb5-2d029ca605e1",
         "url": "https://data.nationalgrideso.com/demand/2-day-ahead-demand-forecast/r/2_day_ahead_demand_forecast"
+    },
+    "historic-day-ahead-wind-forecast": {
+        "id": "7524ec65-f782-4258-aaf8-5b926c17b966",
+        "url": "https://data.nationalgrideso.com/demand/day-ahead-wind-forecast/r/historic_day_ahead_wind_forecasts"
+    },
+    "day-ahead-wind-forecast": {
+        "id": "b2f03146-f05d-4824-a663-3a4f36090c71",
+        "url": "https://data.nationalgrideso.com/demand/day-ahead-wind-forecast/r/day_ahead_wind_forecast"
     }
 }

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyngeso"
-version = "0.2.0"
+version = "0.2.1"
 description = "Simple python wrapper for the National Grid ESO Portal"
 authors = ["atsangarides <[email protected]>"]
 license = "MIT"

tests/test_pyngeso.py

@@ -53,3 +53,20 @@ def test_2day_ahead_historic_forecast():
     assert len(records) > 0
     unique_target_dates = set([record.get("TARGETDATE") for record in records])
     assert len(unique_target_dates) == 1
+
+
+@pytest.mark.vcr
+def test_historic_day_ahead_wind_forecast():
+    date_col = "Date"
+    start_date = "2018-04-17"
+    end_date = "2018-04-17"
+    client = NgEso("historic-day-ahead-wind-forecast")
+    r = client.query(date_col=date_col, start_date=start_date, end_date=end_date)
+
+    assert isinstance(r, bytes)
+    r_dict = json.loads(r)
+    records = r_dict.get("result").get("records")
+    assert isinstance(records, list)
+    assert len(records) > 0
+    unique_target_dates = set([record.get(date_col) for record in records])
+    assert len(unique_target_dates) == 1