pastastore 1.7.2__py3-none-any.whl → 1.8.0__py3-none-any.whl

tests/test_003_pastastore.py

@@ -210,7 +210,8 @@ def test_solve_models_parallel(request, pstore):
 def test_apply(request, pstore):
     depends(request, [f"test_solve_models_and_get_stats[{pstore.type}]"])
 
-    def func(ml):
+    def func(ml_name):
+        ml = pstore.conn.get_models(ml_name)
         return ml.parameters.loc["recharge_A", "optimal"]
 
     result = pstore.apply("models", func)

tests/test_007_hpdextension.py

@@ -25,7 +25,7 @@ def test_hpd_download_precipitation_from_knmi():
     activate_hydropandas_extension()
     pstore = pst.PastaStore()
     pstore.hpd.download_knmi_precipitation(
-        stns=[260], tmin="2022-01-01", tmax="2022-01-31"
+        stns=[260], meteo_var="RH", tmin="2022-01-01", tmax="2022-01-31"
     )
     assert pstore.n_stresses == 1
 
@@ -50,10 +50,10 @@ def test_update_oseries():
     activate_hydropandas_extension()
 
     pstore = pst.PastaStore.from_zip("tests/data/test_hpd_update.zip")
-    pstore.hpd.update_bro_gmw(tmax="2024-01-31")
+    pstore.hpd.update_bro_gmw(tmax="2022-02-28")
     tmintmax = pstore.get_tmin_tmax("oseries")
-    assert tmintmax.loc["GMW000000036319_1", "tmax"] >= Timestamp("2024-01-30")
-    assert tmintmax.loc["GMW000000036327_1", "tmax"] >= Timestamp("2024-01-20")
+    assert tmintmax.loc["GMW000000036319_1", "tmax"] >= Timestamp("2022-02-27")
+    assert tmintmax.loc["GMW000000036327_1", "tmax"] >= Timestamp("2022-02-27")
 
 
 @pytest.mark.xfail(reason="KNMI is being flaky, so allow this test to xfail/xpass.")
@@ -64,9 +64,9 @@ def test_update_stresses():
     activate_hydropandas_extension()
 
     pstore = pst.PastaStore.from_zip("tests/data/test_hpd_update.zip")
-    pstore.hpd.update_knmi_meteo(tmax="2024-01-31", normalize_datetime_index=False)
+    pstore.hpd.update_knmi_meteo(tmax="2022-02-28", normalize_datetime_index=True)
     tmintmax = pstore.get_tmin_tmax("stresses")
-    assert (tmintmax["tmax"] >= Timestamp("2024-01-31")).all()
+    assert (tmintmax["tmax"] >= Timestamp("2024-02-27")).all()
 
 
 @pytest.mark.xfail(reason="KNMI is being flaky, so allow this test to xfail/xpass.")
@@ -78,8 +78,10 @@ def test_nearest_stresses():
 
     pstore = pst.PastaStore.from_zip("tests/data/test_hpd_update.zip")
     pstore.hpd.download_nearest_knmi_precipitation(
-        "GMW000000036319_1", tmin="2024-01-01"
+        "GMW000000036319_1", tmin="2024-01-01", tmax="2024-01-31"
     )
     assert "RD_GROOT-AMMERS" in pstore.stresses_names
-    pstore.hpd.download_nearest_knmi_evaporation("GMW000000036319_1", tmin="2024-01-01")
+    pstore.hpd.download_nearest_knmi_evaporation(
+        "GMW000000036319_1", tmin="2024-01-01", tmax="2024-01-31"
+    )
     assert "EV24_CABAUW-MAST" in pstore.stresses_names