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

tests/test_003_pastastore.py

@@ -168,6 +168,7 @@ def test_create_models(pstore):
         ["oseries1", "oseries2"], store=True, progressbar=False
     )
     _ = pstore.conn.models
+    assert pstore.n_models == 2
 
 
 @pytest.mark.dependency
@@ -252,15 +253,12 @@ def test_update_ts_settings(request, pstore):
 
     ml2 = pstore.get_models(ml.name, update_ts_settings=True)
 
-    try:
-        assert ml2.oseries.settings["tmax"] == o.index[-1]
-        assert ml2.stressmodels["recharge"].prec.settings["tmax"] == tmax
-        assert ml2.stressmodels["recharge"].evap.settings["tmax"] == tmax
-        assert ml2.stressmodels["prec"].stress[0].settings["tmax"] == p2.index[-1]
-    except AssertionError:
-        pstore.del_models("ml_oseries2")
-        pstore.set_check_model_series_values(True)
-        raise
+    assert ml2.oseries.settings["tmax"] == o.index[-1]
+    assert ml2.stressmodels["recharge"].prec.settings["tmax"] == tmax
+    assert ml2.stressmodels["recharge"].evap.settings["tmax"] == tmax
+    assert ml2.stressmodels["prec"].stress[0].settings["tmax"] == p2.index[-1]
+    pstore.del_models("ml_oseries2")
+    pstore.set_check_model_series_values(True)
 
 
 # @pytest.mark.dependency()
@@ -296,6 +294,16 @@ def test_to_from_zip(pstore):
         os.remove(zipname)
 
 
+def test_load_pastastore_from_config_file(pstore):
+    if pstore.type == "pas" or pstore.type == "arcticdb":
+        path = (
+            pstore.conn.path if pstore.type == "pas" else pstore.conn.uri.split("//")[1]
+        )
+        fname = os.path.join(path, f"{pstore.conn.name}.pastastore")
+        pstore2 = pst.PastaStore.from_pastastore_config_file(fname)
+        assert not pstore2.empty
+
+
 def test_example_pastastore():
     from pastastore.datasets import example_pastastore
 
@@ -319,3 +327,12 @@ def test_meta_with_name(pstore):
     pstore.add_stress(s, "what_i_want", kind="special", metadata=smeta)
     assert "what_i_want" in pstore.stresses.index, "This is not right."
     pstore.del_stress("what_i_want")
+
+
+@pytest.mark.dependency
+def test_models_metadata(request, pstore):
+    # depends(request, [f"test_create_models[{pstore.type}]"])
+    pstore.create_models_bulk(["oseries1", "oseries2"], store=True, progressbar=False)
+    df = pstore.models.metadata
+    assert df.index.size == 2
+    assert (df["n_stressmodels"] == 1).all()

tests/test_006_benchmark.py

@@ -176,6 +176,3 @@ def test_benchmark_read_model_arcticdb(benchmark):
     conn = pst.ArcticDBConnector("test", uri)
     _ = benchmark(read_model, conn=conn)
     pst.util.delete_arcticdb_connector(conn=conn)
-    import shutil
-
-    shutil.rmtree("./arctic_db/")

pastastore-1.8.0.dist-info/RECORD

@@ -1,28 +0,0 @@
-docs/conf.py,sha256=XcZUTmn9fGDhhu8k3mpaLu435SpIRNpABADCCTJJuag,6291
-pastastore/__init__.py,sha256=cWwG9-YeiI4aOU0CDBGKbQgmKmmkcPd64YwPq2rRGt0,416
-pastastore/base.py,sha256=hOvkgACew4fpLWumLfHA6PrxLDH1EvqdWjEvFbmxJR0,46436
-pastastore/connectors.py,sha256=QlaFcVEM_ZtOgLd3M2yZfgogcw0zNM08pVbnyRS3Mr8,48454
-pastastore/datasets.py,sha256=FHVfmKqb8beEs9NONsWrCoJY37BmlvFLSEQ1VAFmE8A,6415
-pastastore/plotting.py,sha256=t6gEeHVGzrwvM6q1l8V3OkklpU75O2Y4h6nKEHRWdjo,46416
-pastastore/store.py,sha256=HZtof9gIFSW2nQpc5FHMPb1RTjgo-QzIJxueywd4jDA,65840
-pastastore/styling.py,sha256=4xAY0FmhKrvmAGIuoMM7Uucww_X4KAxTpEoHlsxMldc,2280
-pastastore/util.py,sha256=KCUFV4GkocWaRpG57CdxzfkTXyTEpPjnxsKehYPVN7U,28249
-pastastore/version.py,sha256=vuqYDMX5ua14OQP04vqXH_0A_Ra2XErriiNPxgZEc5w,1205
-pastastore/yaml_interface.py,sha256=MddELxWe8_aqJRMUydOCbjoU1-ZodzxFKYnAaqJ5SqA,29947
-pastastore/extensions/__init__.py,sha256=lCN9xfX1qefUzUbE2FQ12c6NjLbf5HoNo-D8cGb5CTw,461
-pastastore/extensions/accessor.py,sha256=kftQM6dqMDoySbyTKcvmkjC5gJRp465KA18G4NVXUO0,367
-pastastore/extensions/hpd.py,sha256=fcXWb3BlWlogCbg7a2Gmha8P_eCh6zSGTyRlYp3mjXA,27466
-tests/conftest.py,sha256=u097z7LGAnviuzXPzvER9oPjsZWqdij1CJLnW_sPY8E,5258
-tests/test_001_import.py,sha256=g8AaJzWZ088A4B30_w-MrDfAVeeg8m78l--j7Onsklc,208
-tests/test_002_connectors.py,sha256=k9etSRuSFVOrSEtZyxqsCF9GwIg0T7VdDJ2SjSe6i_s,7742
-tests/test_003_pastastore.py,sha256=2nC0pU478iRbYKnVVSjh5F6PA_7SvFROwD6SABL2YSE,9370
-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=yuExF35qqxhw04uYMH3OIOlGr71c4AJSJDMjGD8GefY,4983
-tests/test_007_hpdextension.py,sha256=1QNUahq3hzqxjKbzsjofi9Yuyqe_oDGL0vWp6iouYe4,3004
-tests/test_008_stressmodels.py,sha256=733fyCvuzjKcaLjvSMt5dTTLp-T4alzNJAToSxTIUug,4003
-pastastore-1.8.0.dist-info/LICENSE,sha256=DtHftfUEm99KzgwLr3rQUTg8H3kAS0Z-p5WWJgLf_OY,1082
-pastastore-1.8.0.dist-info/METADATA,sha256=nWFELMkHg9yfp76Ib1xlnPdrp7RZh-9KyhW3CO3rOVY,8032
-pastastore-1.8.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-pastastore-1.8.0.dist-info/top_level.txt,sha256=1bgyMk1p23f04RK83Jju2_YAQBwyoQD_fInxoPB4YRw,22
-pastastore-1.8.0.dist-info/RECORD,,