Add extra wind model

openclimatefix · Nov 22, 2024 · 59fa30c · 59fa30c
1 parent fe4ff90
commit 59fa30c
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 6 deletions.
diff --git 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
@@ -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
@@ -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