webviz-subsurface 0.2.33__py3-none-any.whl → 0.2.35__py3-none-any.whl

tests/integration_tests/plugin_tests/test_reservoir_simulation_timeseries_onebyone.py

@@ -1,16 +0,0 @@
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import ReservoirSimulationTimeSeriesOneByOne
-
-
-def test_reservoir_simulation_timeseries_onebyone(
-    dash_duo, app, shared_settings
-) -> None:
-    plugin = ReservoirSimulationTimeSeriesOneByOne(
-        app,
-        shared_settings["SENS_SETTINGS"],
-        ensembles=shared_settings["SENS_ENSEMBLES"],
-        initial_vector="FOPT",
-    )
-    app.layout = plugin.layout
-    dash_duo.start_server(app)
-    assert not dash_duo.get_logs()

tests/integration_tests/plugin_tests/test_reservoir_simulation_timeseries_regional.py

@@ -1,23 +0,0 @@
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import ReservoirSimulationTimeSeriesRegional
-
-
-def test_reservoir_simulation_timeseries_regional(
-    dash_duo, app, shared_settings, testdata_folder
-) -> None:
-    plugin = ReservoirSimulationTimeSeriesRegional(
-        app,
-        shared_settings["HM_SETTINGS"],
-        ensembles=shared_settings["HM_ENSEMBLES"],
-        fipfile=testdata_folder
-        / "01_drogon_ahm"
-        / "realization-0"
-        / "iter-0"
-        / "share"
-        / "results"
-        / "tables"
-        / "fip.yml",
-    )
-    app.layout = plugin.layout
-    dash_duo.start_server(app)
-    assert not dash_duo.get_logs()

tests/integration_tests/plugin_tests/test_rft_plotter.py

@@ -1,43 +0,0 @@
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import RftPlotter
-from webviz_config.testing import WebvizComposite
-
-
-def test_rft_plotter(
-    _webviz_duo: WebvizComposite, shared_settings, testdata_folder
-) -> None:
-    plugin = RftPlotter(
-        shared_settings["HM_SETTINGS"],
-        ensembles=shared_settings["HM_ENSEMBLES"],
-        formations=testdata_folder
-        / "01_drogon_ahm"
-        / "realization-0"
-        / "iter-0"
-        / "share"
-        / "results"
-        / "tables"
-        / "formations_res_only.csv",
-        faultlines=testdata_folder
-        / "01_drogon_ahm"
-        / "realization-0"
-        / "iter-0"
-        / "share"
-        / "results"
-        / "polygons"
-        / "toptherys--gl_faultlines_extract_postprocess.csv",
-    )
-
-    _webviz_duo.start_server(plugin)
-
-    _webviz_duo.toggle_webviz_settings_drawer()
-    _webviz_duo.toggle_webviz_settings_group(
-        plugin.view("map-view").settings_group_unique_id("map-settings")
-    )
-    # Using str literals directly, not IDs from the plugin as intended because
-    # the run test did not accept the imports
-
-    my_component_id = _webviz_duo.view_settings_group_unique_component_id(
-        "map-view", "map-settings", "map-ensemble"
-    )
-    _webviz_duo.wait_for_contains_text(my_component_id, "iter-0")
-    assert not _webviz_duo.get_logs()

tests/integration_tests/plugin_tests/test_segy_viewer.py

@@ -1,22 +0,0 @@
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import SegyViewer
-
-
-def test_segy_viewer(dash_duo, app, shared_settings, testdata_folder) -> None:
-    plugin = SegyViewer(
-        app,
-        shared_settings["HM_SETTINGS"],
-        segyfiles=[
-            testdata_folder
-            / "01_drogon_ahm"
-            / "realization-0"
-            / "iter-0"
-            / "share"
-            / "results"
-            / "seismic"
-            / "seismic--amplitude_depth--20180701_20180101.segy"
-        ],
-    )
-    app.layout = plugin.layout
-    dash_duo.start_server(app)
-    assert not dash_duo.get_logs()

tests/integration_tests/plugin_tests/test_simulation_timeseries_onebyone.py

@@ -1,23 +0,0 @@
-import warnings
-
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import SimulationTimeSeriesOneByOne
-from webviz_config.testing import WebvizComposite
-
-
-def test_simulation_timeseries_onebyone(
-    _webviz_duo: WebvizComposite, shared_settings: dict
-) -> None:
-    plugin = SimulationTimeSeriesOneByOne(
-        webviz_settings=shared_settings["SENS_SETTINGS"],
-        ensembles=shared_settings["SENS_ENSEMBLES"],
-        initial_vector="FOPT",
-    )
-    _webviz_duo.start_server(plugin)
-    logs = []
-    for log in _webviz_duo.get_logs() or []:
-        if "dash_renderer" in log.get("message"):
-            warnings.warn(log.get("message"))
-        else:
-            logs.append(log)
-    assert not logs

tests/integration_tests/plugin_tests/test_structural_uncertainty.py

@@ -1,244 +0,0 @@
-import json
-
-from dash import Input, Output, State, html
-from webviz_config import WebvizSettings
-
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import StructuralUncertainty
-from webviz_config.themes import default_theme
-
-# pylint: enable=no-name-in-module
-
-
-def stringify_object_id(uuid) -> str:
-    """Object ids must be sorted and converted to
-    css strings to be recognized as dom elements"""
-    sorted_uuid_obj = json.loads(
-        json.dumps(
-            uuid,
-            sort_keys=True,
-            separators=(",", ":"),
-        )
-    )
-    string = ["{"]
-    for idx, (key, value) in enumerate(sorted_uuid_obj.items()):
-        string.append(f'\\"{key}\\"\\:\\"{value}\\"\\')
-        if idx == len(sorted_uuid_obj) - 1:
-            string.append("}")
-        else:
-            string.append(",")
-    return ("").join(string)
-
-
-# pylint: disable=too-many-locals
-def test_default_configuration(dash_duo, app, testdata_folder) -> None:
-    webviz_settings = WebvizSettings(
-        shared_settings={
-            "scratch_ensembles": {
-                "iter-0": str(testdata_folder / "01_drogon_ahm/realization-*/iter-0")
-            }
-        },
-        theme=default_theme,
-    )
-    plugin = StructuralUncertainty(
-        app,
-        webviz_settings,
-        ensembles=["iter-0"],
-        surface_attributes=["ds_extract_postprocess"],
-        surface_name_filter=[
-            "topvolon",
-            "toptherys",
-            "topvolantis",
-            "basevolantis",
-        ],
-        wellsuffix=".rmswell",
-        wellfolder=testdata_folder / "observed_data" / "wells",
-    )
-
-    app.layout = plugin.layout
-    dash_duo.start_server(app)
-
-    intersection_data_id = plugin.uuid("intersection-data")
-    dialog_id = plugin.uuid("dialog")
-    # Check some initialization
-    # Check dropdowns
-    for element, return_val in zip(
-        ["well", "surface_attribute"], ["55_33-1", "ds_extract_postprocess"]
-    ):
-        uuid = stringify_object_id(
-            uuid={"element": element, "id": intersection_data_id}
-        )
-        assert dash_duo.wait_for_element(f"#\\{uuid} .Select-value").text == return_val
-
-    # Check Selects
-    for element, return_val in zip(
-        ["surface_names"],
-        [["topvolon", "toptherys", "topvolantis", "basevolantis"]],
-    ):
-        uuid = stringify_object_id(
-            uuid={"element": element, "id": intersection_data_id}
-        )
-        assert (
-            dash_duo.wait_for_element(f"#\\{uuid} select").text.splitlines()
-            == return_val
-        )
-
-    # Check Calculation checkbox
-    uuid = stringify_object_id(
-        uuid={"element": "calculation", "id": intersection_data_id}
-    )
-    calculation_element = dash_duo.driver.find_elements_by_css_selector(
-        f"#\\{uuid} > label > input"
-    )
-    assert len(calculation_element) == len(
-        ["Min", "Max", "Mean", "Realizations", "Uncertainty envelope"]
-    )
-    for checkbox, selected in zip(
-        calculation_element,
-        ["true", "true", "true", None, None],
-    ):
-        assert checkbox.get_attribute("selected") == selected
-
-    # Check realizations
-    real_filter_btn_uuid = stringify_object_id(
-        {
-            "id": dialog_id,
-            "dialog_id": "realization-filter",
-            "element": "button-open",
-        }
-    )
-    real_uuid = stringify_object_id(
-        uuid={"element": "realizations", "id": intersection_data_id}
-    )
-
-    ### Open realization filter and check realizations
-    dash_duo.wait_for_element_by_id(real_filter_btn_uuid).click()
-    real_selector = dash_duo.wait_for_element_by_id(real_uuid)
-    assert real_selector.text.splitlines() == ["0", "1"]
-
-    assert not dash_duo.get_logs(), "browser console should contain no error"
-
-
-def test_full_configuration(dash_duo, app, testdata_folder) -> None:
-    webviz_settings = WebvizSettings(
-        shared_settings={
-            "scratch_ensembles": {
-                "iter-0": str(testdata_folder / "01_drogon_ahm/realization-*/iter-0"),
-            }
-        },
-        theme=default_theme,
-    )
-    plugin = StructuralUncertainty(
-        app,
-        webviz_settings,
-        ensembles=["iter-0"],
-        surface_attributes=["ds_extract_postprocess"],
-        surface_name_filter=["topvolon", "toptherys", "topvolantis", "basevolantis"],
-        wellfolder=testdata_folder / "observed_data" / "wells",
-        wellsuffix=".rmswell",
-        zonelog="Zone",
-        initial_settings={
-            "intersection_data": {
-                "surface_names": ["topvolon", "toptherys", "topvolantis"],
-                "surface_attribute": "ds_extract_postprocess",
-                "ensembles": [
-                    "iter-0",
-                ],
-                "calculation": ["Mean", "Min", "Max"],
-                # - Uncertainty envelope
-                "well": "55_33-1",
-                "realizations": [0, 1],
-                "colors": {
-                    "topvolon": {"iter-0": "#2C82C9"},
-                    "toptherys": {
-                        "iter-0": "#512E34",
-                    },
-                    "topvolantis": {
-                        "iter-0": "#EEE657",
-                    },
-                },
-            },
-            "intersection_layout": {
-                "yaxis": {
-                    "range": [1700, 1550],
-                    "title": "True vertical depth [m]",
-                },
-                "xaxis": {"title": "Lateral distance [m]"},
-            },
-        },
-    )
-
-    app.layout = plugin.layout
-
-    # Injecting a div that will be updated when the plot data stores are
-    # changed. Since the plot data are stored in LocalStorage and Selenium
-    # has no functionality to wait for LocalStorage to equal some value we
-    # instead populate this injected div with some data before we check the content
-    # of Localstorage.
-    @app.callback(
-        Output(plugin.uuid("layout"), "children"),
-        Input(plugin.uuid("intersection-graph-layout"), "data"),
-        State(plugin.uuid("layout"), "children"),
-    )
-    def _add_or_update_div(data, children):
-        plot_is_updated = html.Div(
-            id=plugin.uuid("plot_is_updated"), children=data.get("title")
-        )
-        if len(children) == 6:
-            children[5] = plot_is_updated
-        else:
-            children.append(plot_is_updated)
-
-        return children
-
-    dash_duo.start_server(app)
-
-    intersection_data_id = plugin.uuid("intersection-data")
-
-    # Check some initialization
-    # Check dropdowns
-    for element, return_val in zip(
-        ["well", "surface_attribute"], ["55_33-1", "ds_extract_postprocess"]
-    ):
-        uuid = stringify_object_id(
-            uuid={"element": element, "id": intersection_data_id}
-        )
-        assert dash_duo.wait_for_text_to_equal(f"#\\{uuid} .Select-value", return_val)
-
-    # Wait for the callbacks to execute
-    dash_duo.wait_for_text_to_equal(
-        f'#{plugin.uuid("plot_is_updated")}',
-        "Intersection along well: 55_33-1",
-        timeout=30,
-    )
-
-    # Check that graph data is stored
-    graph_data = dash_duo.get_session_storage(plugin.uuid("intersection-graph-data"))
-    assert len(graph_data) == 14
-    graph_layout = dash_duo.get_session_storage(
-        plugin.uuid("intersection-graph-layout")
-    )
-    assert isinstance(graph_layout, dict)
-    assert graph_layout.get("title") == "Intersection along well: 55_33-1"
-
-    ### Change well and check graph
-    well_uuid = stringify_object_id(
-        uuid={"element": "well", "id": intersection_data_id}
-    )
-
-    apply_btn = dash_duo.wait_for_element_by_id(
-        plugin.uuid("apply-intersection-data-selections")
-    )
-    well_dropdown = dash_duo.wait_for_element_by_id(well_uuid)
-    dash_duo.select_dcc_dropdown(well_dropdown, value="55_33-2")
-    apply_btn.click()
-
-    # dash_duo.wait_for_text_to_equal(
-    #     f'#{plugin.uuid("plot_is_updated")}',
-    #     "Intersection along well: 55_33-1",
-    #     timeout=100,
-    # )
-    graph_layout = dash_duo.get_session_storage(
-        plugin.uuid("intersection-graph-layout")
-    )
-    # assert graph_layout.get("title") == "Intersection along well: 55_33-2"

tests/integration_tests/plugin_tests/test_surface_viewer_fmu.py

@@ -1,21 +0,0 @@
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import SurfaceViewerFMU
-
-
-def test_surface_viewer_fmu(dash_duo, app, shared_settings, testdata_folder) -> None:
-    plugin = SurfaceViewerFMU(
-        app,
-        shared_settings["HM_SETTINGS"],
-        ensembles=shared_settings["HM_ENSEMBLES"],
-        wellsuffix=".rmswell",
-        wellfolder=testdata_folder
-        / "01_drogon_ahm"
-        / "realization-0"
-        / "iter-0"
-        / "share"
-        / "results"
-        / "wells",
-    )
-    app.layout = plugin.layout
-    dash_duo.start_server(app)
-    assert not dash_duo.get_logs()

tests/integration_tests/plugin_tests/test_surface_with_grid_crossection.py

@@ -1,45 +0,0 @@
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import SurfaceWithGridCrossSection
-
-
-def test_surface_with_grid_crosssection(
-    dash_duo, app, shared_settings, testdata_folder
-) -> None:
-    plugin = SurfaceWithGridCrossSection(
-        app,
-        shared_settings["HM_SETTINGS"],
-        gridfile=(
-            testdata_folder
-            / "01_drogon_ahm"
-            / "realization-0"
-            / "iter-0"
-            / "share"
-            / "results"
-            / "grids"
-            / "geogrid.roff"
-        ),
-        gridparameterfiles=[
-            testdata_folder
-            / "01_drogon_ahm"
-            / "realization-0"
-            / "iter-0"
-            / "share"
-            / "results"
-            / "grids"
-            / "geogrid--phit.roff"
-        ],
-        surfacefiles=[
-            testdata_folder
-            / "01_drogon_ahm"
-            / "realization-0"
-            / "iter-0"
-            / "share"
-            / "results"
-            / "maps"
-            / "topvolon--ds_extract_geogrid.gri"
-        ],
-        surfacenames=["Top Volon"],
-    )
-    app.layout = plugin.layout
-    dash_duo.start_server(app)
-    assert not dash_duo.get_logs()

tests/integration_tests/plugin_tests/test_surface_with_seismic_crossection.py

@@ -1,35 +0,0 @@
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import SurfaceWithSeismicCrossSection
-
-
-def test_surface_with_seismic_crosssection(
-    dash_duo, app, shared_settings, testdata_folder
-) -> None:
-    plugin = SurfaceWithSeismicCrossSection(
-        app,
-        shared_settings["HM_SETTINGS"],
-        segyfiles=[
-            testdata_folder
-            / "01_drogon_ahm"
-            / "realization-0"
-            / "iter-0"
-            / "share"
-            / "results"
-            / "seismic"
-            / "seismic--amplitude_depth--20180701_20180101.segy"
-        ],
-        surfacefiles=[
-            testdata_folder
-            / "01_drogon_ahm"
-            / "realization-0"
-            / "iter-0"
-            / "share"
-            / "results"
-            / "maps"
-            / "topvolon--ds_extract_geogrid.gri"
-        ],
-        surfacenames=["Top Volon"],
-    )
-    app.layout = plugin.layout
-    dash_duo.start_server(app)
-    assert not dash_duo.get_logs()

tests/integration_tests/plugin_tests/test_tornado_plotter_fmu.py

@@ -1,14 +0,0 @@
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import TornadoPlotterFMU
-
-
-def test_tornado_plotter_fmu(dash_duo, app, shared_settings) -> None:
-    plugin = TornadoPlotterFMU(
-        shared_settings["SENS_SETTINGS"],
-        ensemble=shared_settings["SENS_ENSEMBLES"][0],
-        csvfile="share/results/volumes/geogrid--vol.csv",
-        multi_value_selectors=["REGION", "ZONE"],
-    )
-    app.layout = plugin.layout
-    dash_duo.start_server(app)
-    assert not dash_duo.get_logs()

tests/integration_tests/plugin_tests/test_vfp_analysis.py

@@ -1,13 +0,0 @@
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import VfpAnalysis
-from webviz_config.testing import WebvizComposite
-
-
-def test_vfp_analysis(_webviz_duo: WebvizComposite, shared_settings: dict) -> None:
-    plugin = VfpAnalysis(
-        shared_settings["HM_SETTINGS"], vfp_file_pattern="tests/data/vfp.arrow"
-    )
-
-    _webviz_duo.start_server(plugin)
-
-    assert not _webviz_duo.get_logs()

tests/integration_tests/plugin_tests/test_volumetric_analysis.py

@@ -1,38 +0,0 @@
-import warnings
-
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import VolumetricAnalysis
-
-
-def test_volumetrics_no_sens(dash_duo, app, shared_settings) -> None:
-    plugin = VolumetricAnalysis(
-        shared_settings["HM_SETTINGS"],
-        ensembles=shared_settings["HM_ENSEMBLES"],
-        volfiles={"geogrid": "geogrid--vol.csv", "simgrid": "simgrid--vol.csv"},
-    )
-    app.layout = plugin.layout
-    dash_duo.start_server(app)
-    logs = []
-    for log in dash_duo.get_logs() or []:
-        if "dash_renderer" in log.get("message"):
-            warnings.warn(log.get("message"))
-        else:
-            logs.append(log)
-    assert not logs
-
-
-def test_volumetrics_sens(dash_duo, app, shared_settings) -> None:
-    plugin = VolumetricAnalysis(
-        shared_settings["SENS_SETTINGS"],
-        ensembles=shared_settings["SENS_ENSEMBLES"],
-        volfiles={"geogrid": "geogrid--vol.csv", "simgrid": "simgrid--vol.csv"},
-    )
-    app.layout = plugin.layout
-    dash_duo.start_server(app)
-    logs = []
-    for log in dash_duo.get_logs() or []:
-        if "dash_renderer" in log.get("message"):
-            warnings.warn(log.get("message"))
-        else:
-            logs.append(log)
-    assert not logs

tests/integration_tests/plugin_tests/test_well_log_viewer.py

@@ -1,16 +0,0 @@
-# pylint: disable=no-name-in-module
-from webviz_config.plugins import WellLogViewer
-
-
-def test_well_log_viewer(dash_duo, app, testdata_folder) -> None:
-    wellfolder = testdata_folder / "observed_data" / "wells/"
-    plugin = WellLogViewer(
-        app,
-        wellfolder=wellfolder,
-        wellsuffix=".rmswell",
-        mdlog="MDepth",
-        logtemplates=[f"{testdata_folder}/webviz_examples/all_logs_template.yml"],
-    )
-    app.layout = plugin.layout
-    dash_duo.start_server(app)
-    assert not dash_duo.get_logs()

webviz_subsurface/_providers/ensemble_summary_provider/dev_resampling_perf_testing.py

@@ -1,112 +0,0 @@
-import logging
-import time
-
-import numpy as np
-import pyarrow as pa
-
-from webviz_subsurface._providers.ensemble_summary_provider._resampling import (
-    sample_segmented_multi_real_table_at_date,
-)
-
-
-def _create_table(
-    num_reals: int, start_date: np.datetime64, end_date: np.datetime64, num_columns: int
-) -> pa.Table:
-    date_arr_np = np.empty(0, np.datetime64)
-    real_arr_np = np.empty(0, np.int32)
-
-    for real in range(0, num_reals):
-        dates_for_this_real = np.arange(start_date, end_date + 1)
-        dates_for_this_real = dates_for_this_real.astype("datetime64[ms]")
-        real_arr_np = np.concatenate(
-            (real_arr_np, np.full(len(dates_for_this_real), real))
-        )
-        date_arr_np = np.concatenate((date_arr_np, dates_for_this_real))
-
-    print(
-        f"real_arr_np (num unique={len(np.unique(real_arr_np))}  len={len(real_arr_np)}):"
-    )
-    print(real_arr_np)
-    print(
-        f"date_arr_np (num unique={len(np.unique(date_arr_np))}  len={len(date_arr_np)}):"
-    )
-    print(date_arr_np)
-
-    field_list = []
-    columndata_list = []
-    field_list.append(pa.field("DATE", pa.timestamp("ms")))
-    field_list.append(pa.field("REAL", pa.int64()))
-    columndata_list.append(pa.array(date_arr_np))
-    columndata_list.append(pa.array(real_arr_np))
-
-    num_rows = len(real_arr_np)
-
-    for colnum in range(0, num_columns):
-        if (colnum % 2) == 0:
-            metadata = {b"is_rate": b'{"is_rate": False}'}
-        else:
-            metadata = {b"is_rate": b'{"is_rate": True}'}
-
-        field_list.append(pa.field(f"c_{colnum}", pa.float32(), metadata=metadata))
-
-        valarr = np.linspace(colnum, colnum + num_rows, num_rows)
-        columndata_list.append(pa.array(valarr))
-
-    schema = pa.schema(field_list)
-    return pa.table(columndata_list, schema=schema)
-
-
-def main() -> None:
-    print()
-    print("## Running resampling performance tests")
-    print("## =================================================")
-
-    logging.basicConfig(
-        level=logging.WARNING,
-        format="%(asctime)s %(levelname)-3s [%(name)s]: %(message)s",
-    )
-    logging.getLogger("webviz_subsurface").setLevel(level=logging.INFO)
-    logging.getLogger("webviz_subsurface").setLevel(level=logging.DEBUG)
-
-    # table = _create_table(
-    #     num_reals=3,
-    #     start_date=np.datetime64("2020-12-30"),
-    #     end_date=np.datetime64("2021-01-05"),
-    #     num_columns=4,
-    # )
-
-    table = _create_table(
-        num_reals=100,
-        start_date=np.datetime64("2000-01-01", "M"),
-        end_date=np.datetime64("2099-12-31", "M"),
-        num_columns=10000,
-    )
-
-    print("## table shape (rows,columns):", table.shape)
-    # print(table.to_pandas())
-
-    start_tim = time.perf_counter()
-
-    res = sample_segmented_multi_real_table_at_date(
-        table, np.datetime64("2098-01-03", "ms")
-    )
-
-    # res = sample_segmented_multi_real_table_at_date(
-    #     table, np.datetime64("2098-01-01", "ms")
-    # )
-
-    elapsed_time_ms = int(1000 * (time.perf_counter() - start_tim))
-
-    # print(res)
-    # print(res.to_pandas())
-
-    print("## res shape:", res.shape)
-
-    print(f"## sample at date took: {elapsed_time_ms}ms")
-
-
-# Running:
-#   python -m webviz_subsurface._providers.ensemble_summary_provider.dev_resampling_perf_testing
-# -------------------------------------------------------------------------
-if __name__ == "__main__":
-    main()