diff --git a/india_forecast_app/models/all_models.yaml b/india_forecast_app/models/all_models.yaml index ed0c2c4..52e67fe 100644 --- a/india_forecast_app/models/all_models.yaml +++ b/india_forecast_app/models/all_models.yaml @@ -13,6 +13,12 @@ models: version: 546baded3d4216736d8ee8d6798d47235bd72b08 client: ruvnl asset_type: wind + - name: windnet_india_mo_v2 + type: pvnet + id: openclimatefix/windnet_india + version: 165267d34500cf0c881ed70d9318421f4e0d10f1 + client: ruvnl + asset_type: wind # RU client solar - name: pvnet_india type: pvnet diff --git a/tests/models/test_pydantic_models.py b/tests/models/test_pydantic_models.py index edb83f1..df530f1 100644 --- a/tests/models/test_pydantic_models.py +++ b/tests/models/test_pydantic_models.py @@ -5,7 +5,7 @@ def test_get_all_models(): """Test for getting all models""" models = get_all_models() - assert len(models.models) == 5 + assert len(models.models) == 6 def test_get_all_models_client(): diff --git a/tests/test_app.py b/tests/test_app.py index 737eeb5..cfa69f4 100644 --- a/tests/test_app.py +++ b/tests/test_app.py @@ -161,9 +161,9 @@ def test_app( assert result.exit_code == 0 if write_to_db: - assert db_session.query(ForecastSQL).count() == init_n_forecasts + 3 * 2 - assert db_session.query(ForecastValueSQL).count() == init_n_forecast_values + (3 * 2 * 192) - assert db_session.query(MLModelSQL).count() == 3 * 2 + assert db_session.query(ForecastSQL).count() == init_n_forecasts + 4 * 2 + assert db_session.query(ForecastValueSQL).count() == init_n_forecast_values + (4 * 2 * 192) + assert db_session.query(MLModelSQL).count() == 4 * 2 else: assert db_session.query(ForecastSQL).count() == init_n_forecasts assert db_session.query(ForecastValueSQL).count() == init_n_forecast_values @@ -183,8 +183,8 @@ def test_app_no_pv_data( result = run_click_script(app, args) assert result.exit_code == 0 - assert db_session.query(ForecastSQL).count() == init_n_forecasts + 2 * 3 - assert db_session.query(ForecastValueSQL).count() == init_n_forecast_values + (2 * 3 * 192) + assert db_session.query(ForecastSQL).count() == init_n_forecasts + 2 * 4 + assert db_session.query(ForecastValueSQL).count() == init_n_forecast_values + (2 * 4 * 192) @pytest.mark.requires_hf_token