pastastore 1.10.2__py3-none-any.whl → 1.11.0__py3-none-any.whl

tests/test_005_maps_plots.py

@@ -63,6 +63,18 @@ def test_map_models(request, pstore):
     plt.close(ax.figure)
 
 
+@pytest.mark.dependency
+def test_map_signatures(request, pstore):
+    ax = pstore.maps.signature("mean_annual_maximum")
+    plt.close(ax.figure)
+
+
+@pytest.mark.dependency
+def test_map_modelparam(request, pstore):
+    ax = pstore.maps.modelparam("recharge_A")
+    plt.close(ax.figure)
+
+
 @pytest.mark.dependency
 def test_map_model(request, pstore):
     depends(request, [f"test_map_models[{pstore.type}]"])

tests/test_006_benchmark.py

@@ -128,7 +128,7 @@ def test_benchmark_write_model_arcticdb(benchmark):
 
 
 def write_model_nocheckts(conn, ml):
-    conn.set_check_model_series_values(False)
+    conn.validator.set_check_model_series_values(False)
     conn.add_model(ml, overwrite=True)
 
 

tests/test_007_hpdextension.py

@@ -1,10 +1,31 @@
 # ruff: noqa: D100 D103
 import pytest
 from pandas import Timestamp
+from pandas.testing import assert_series_equal
+from pastas.timeseries_utils import timestep_weighted_resample
 
 import pastastore as pst
 
 
+def create_test_zip():
+    from pastastore.extensions import activate_hydropandas_extension
+
+    activate_hydropandas_extension()
+    pstore = pst.PastaStore()
+    pstore.hpd.download_bro_gmw(
+        extent=(117_850, 118_180, 439_550, 439_900),
+        tmin="2022-01-01",
+        tmax="2022-01-02",
+    )
+    pstore.hpd.download_knmi_precipitation(
+        stns=[260], meteo_var="RH", tmin="2022-01-01", tmax="2022-01-31"
+    )
+    pstore.hpd.download_knmi_evaporation(
+        stns=[260], tmin="2022-01-01", tmax="2022-01-31"
+    )
+    pstore.to_zip("tests/data/test_hpd_update.zip", overwrite=True)
+
+
 @pytest.mark.pastas150
 def test_hpd_download_from_bro():
     from pastastore.extensions import activate_hydropandas_extension
@@ -12,12 +33,14 @@ def test_hpd_download_from_bro():
     activate_hydropandas_extension()
     pstore = pst.PastaStore()
     pstore.hpd.download_bro_gmw(
-        extent=(117850, 118180, 439550, 439900), tmin="2022-01-01", tmax="2022-01-02"
+        extent=(117_850, 118_180, 439_550, 439_900),
+        tmin="2022-01-01",
+        tmax="2022-01-02",
     )
     assert pstore.n_oseries == 3
 
 
-@pytest.mark.xfail(reason="KNMI is being flaky, so allow this test to xfail/xpass.")
+# @pytest.mark.xfail(reason="KNMI is being flaky, so allow this test to xfail/xpass.")
 @pytest.mark.pastas150
 def test_hpd_download_precipitation_from_knmi():
     from pastastore.extensions import activate_hydropandas_extension
@@ -30,7 +53,7 @@ def test_hpd_download_precipitation_from_knmi():
     assert pstore.n_stresses == 1
 
 
-@pytest.mark.xfail(reason="KNMI is being flaky, so allow this test to xfail/xpass.")
+# @pytest.mark.xfail(reason="KNMI is being flaky, so allow this test to xfail/xpass.")
 @pytest.mark.pastas150
 def test_hpd_download_evaporation_from_knmi():
     from pastastore.extensions import activate_hydropandas_extension
@@ -56,9 +79,11 @@ def test_update_oseries():
     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.")
+# @pytest.mark.xfail(reason="KNMI is being flaky, so allow this test to xfail/xpass.")
 @pytest.mark.pastas150
 def test_update_stresses():
+    import hydropandas as hpd
+
     from pastastore.extensions import activate_hydropandas_extension
 
     activate_hydropandas_extension()
@@ -66,10 +91,23 @@ def test_update_stresses():
     pstore = pst.PastaStore.from_zip("tests/data/test_hpd_update.zip")
     pstore.hpd.update_knmi_meteo(tmax="2022-02-28", normalize_datetime_index=True)
     tmintmax = pstore.get_tmin_tmax("stresses")
-    assert (tmintmax["tmax"] >= Timestamp("2024-02-27")).all()
+    assert (tmintmax["tmax"] >= Timestamp("2022-02-27")).all()
+
+    # check if result is equal to hydropandas result after resampling
+    oc = hpd.read_knmi(
+        stns=[260], meteo_vars=["RH", "EV24"], starts="2022-01-01", ends="2022-02-28"
+    )
+    for i in range(2):
+        o = oc.obs.iloc[i].squeeze("columns") * 1e3
+        resampled_result = (timestep_weighted_resample(o, o.index.normalize())).dropna()
+        assert_series_equal(
+            pstore.get_stress(pstore.stresses_names[i]).squeeze(),
+            resampled_result,
+            check_names=False,
+        )
 
 
-@pytest.mark.xfail(reason="KNMI is being flaky, so allow this test to xfail/xpass.")
+# @pytest.mark.xfail(reason="KNMI is being flaky, so allow this test to xfail/xpass.")
 @pytest.mark.pastas150
 def test_nearest_stresses():
     from pastastore.extensions import activate_hydropandas_extension
@@ -80,8 +118,8 @@ def test_nearest_stresses():
     pstore.hpd.download_nearest_knmi_precipitation(
         "GMW000000036319_1", tmin="2024-01-01", tmax="2024-01-31"
     )
-    assert "RD_GROOT-AMMERS" in pstore.stresses_names
+    assert "RD_GROOT-AMMERS_434" in pstore.stresses_names
     pstore.hpd.download_nearest_knmi_evaporation(
         "GMW000000036319_1", tmin="2024-01-01", tmax="2024-01-31"
     )
-    assert "EV24_CABAUW-MAST" in pstore.stresses_names
+    assert "EV24_CABAUW-MAST_348" in pstore.stresses_names

pastastore-1.10.2.dist-info/RECORD

@@ -1,28 +0,0 @@
-docs/conf.py,sha256=XcZUTmn9fGDhhu8k3mpaLu435SpIRNpABADCCTJJuag,6291
-pastastore/__init__.py,sha256=cWwG9-YeiI4aOU0CDBGKbQgmKmmkcPd64YwPq2rRGt0,416
-pastastore/base.py,sha256=ng64KGuKo2iMLjL0H7qG3NVNvXPwMGJBf3oLtQbt6DQ,48264
-pastastore/connectors.py,sha256=hum8KkPi28B3XBsKL_RghRCsZndf-jeFlb4baTvyDG4,50197
-pastastore/datasets.py,sha256=FHVfmKqb8beEs9NONsWrCoJY37BmlvFLSEQ1VAFmE8A,6415
-pastastore/plotting.py,sha256=ygKXdi42sPLaehze4EjU8kRE2Dk46wVxSkB9RJ2Re84,54535
-pastastore/store.py,sha256=Fs_WCEGqafOKm3whLUR54tqXL1dL3FdzwxNI1DSZ2KI,69272
-pastastore/styling.py,sha256=0IEp_r-SpcaslShAZvZV6iuEhTG_YzNq-ad8krib3U0,2304
-pastastore/util.py,sha256=31dzHaK6xdFHGDkYh49qGBq1dGel2m9r7i797S3WUpQ,28505
-pastastore/version.py,sha256=2Y6_CKnmHYQTxRyvvUEW2519LmIeszWd-0NYpCRyV6M,1206
-pastastore/yaml_interface.py,sha256=n6zjQ7ENrUvxszb6zE-jPLa-XVsoEOTJHQmRV1_fFt0,30818
-pastastore/extensions/__init__.py,sha256=lCN9xfX1qefUzUbE2FQ12c6NjLbf5HoNo-D8cGb5CTw,461
-pastastore/extensions/accessor.py,sha256=kftQM6dqMDoySbyTKcvmkjC5gJRp465KA18G4NVXUO0,367
-pastastore/extensions/hpd.py,sha256=i9Ty9GyRyrtY0uqjMpWheH33oYDQM8WMmygaDEPx-w4,28581
-pastastore-1.10.2.dist-info/licenses/LICENSE,sha256=MB_6p4kXDCUsYNjslcMByBu6i7wMNRKPC36JnhzpN4o,1087
-tests/conftest.py,sha256=TB0ZUH1m45gvQd_EZO7iudvhFw4JA-8rTJ71GT6Nf1w,5061
-tests/test_001_import.py,sha256=g8AaJzWZ088A4B30_w-MrDfAVeeg8m78l--j7Onsklc,208
-tests/test_002_connectors.py,sha256=X7IbxZZXBYd2UT8B5gu0_Jw--3Twe7KHjSBbmMwNnRk,8193
-tests/test_003_pastastore.py,sha256=nhcUJHC2KiF9KREP_2uj_T2skKooUk13T1EVtkbwQnM,10051
-tests/test_004_yaml.py,sha256=3hMNjb9s0S2rbmpyEjW6FDRAxfUZS_U1qoPl4wB-cCo,4440
-tests/test_005_maps_plots.py,sha256=L0ppGf-cudsrdxteWy3qsV4We96DW4bCBE7c6jEm6aM,1866
-tests/test_006_benchmark.py,sha256=VZG0bY7uz8DkfIZTgRCzkEDG8rguBEt_-mdGSMQLN2w,4930
-tests/test_007_hpdextension.py,sha256=1QNUahq3hzqxjKbzsjofi9Yuyqe_oDGL0vWp6iouYe4,3004
-tests/test_008_stressmodels.py,sha256=733fyCvuzjKcaLjvSMt5dTTLp-T4alzNJAToSxTIUug,4003
-pastastore-1.10.2.dist-info/METADATA,sha256=8oLyVDQ95wA9wzSxBR29fMg2s81HjlvXGenukFxx2NA,7662
-pastastore-1.10.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pastastore-1.10.2.dist-info/top_level.txt,sha256=1bgyMk1p23f04RK83Jju2_YAQBwyoQD_fInxoPB4YRw,22
-pastastore-1.10.2.dist-info/RECORD,,