diff --git a/snowreport/assets/report_raw_json.py b/snowreport/assets/report_raw_json.py index cd6736e..cb4edf1 100644 --- a/snowreport/assets/report_raw_json.py +++ b/snowreport/assets/report_raw_json.py @@ -55,7 +55,8 @@ def resort_raw(context, **resort_assets) -> pd.DataFrame: "low_today": pd.to_numeric(resort['weatherToday_Temperature_Low']), "low_tomorrow": pd.to_numeric(resort['weatherTomorrow_Temperature_Low']), "high_today": pd.to_numeric(resort['weatherToday_Temperature_High']), - "high_tomorrow": pd.to_numeric(resort['weatherTomorrow_Temperature_High']) + "high_tomorrow": pd.to_numeric(resort['weatherTomorrow_Temperature_High']), + "open_trails": pd.to_numeric(resort['openDownHillTrails']) }) for resort_name in resorts[1:]: @@ -70,7 +71,8 @@ def resort_raw(context, **resort_assets) -> pd.DataFrame: "low_today": pd.to_numeric(resort['weatherToday_Temperature_Low']), "low_tomorrow": pd.to_numeric(resort['weatherTomorrow_Temperature_Low']), "high_today": pd.to_numeric(resort['weatherToday_Temperature_High']), - "high_tomorrow": pd.to_numeric(resort['weatherTomorrow_Temperature_High']) + "high_tomorrow": pd.to_numeric(resort['weatherTomorrow_Temperature_High']), + "open_trails": pd.to_numeric(resort['openDownHillTrails']) }) resort_summary = resort_summary.append(add_to_summary) diff --git a/snowreport_tests/test_assets.py b/snowreport_tests/test_assets.py index 84c4b07..0a696e8 100644 --- a/snowreport_tests/test_assets.py +++ b/snowreport_tests/test_assets.py @@ -13,7 +13,8 @@ 'weatherToday_Temperature_Low': ["30"], 'weatherTomorrow_Temperature_Low': ["16"], 'weatherToday_Temperature_High': ["48"], - 'weatherTomorrow_Temperature_High': ["37"] + 'weatherTomorrow_Temperature_High': ["37"], + 'openDownHillTrails': ["1"] } resorts = { @@ -30,7 +31,8 @@ "low_today": int64, "low_tomorrow": int64, "high_today": int64, - "high_tomorrow": int64 + "high_tomorrow": int64, + "open_trails": int64 } @pytest.mark.parametrize(