webviz-subsurface 0.2.39__py3-none-any.whl → 0.2.41__py3-none-any.whl

tests/unit_tests/plugin_tests/test_simulation_time_series/test_utils/test_vector_statistics.py

@@ -1,171 +0,0 @@
-import datetime
-
-import pandas as pd
-import pytest
-from pandas._testing import assert_frame_equal
-
-from webviz_subsurface._utils.dataframe_utils import make_date_column_datetime_object
-from webviz_subsurface.plugins._simulation_time_series._views._subplot_view._types import (
-    StatisticsOptions,
-)
-
-# pylint: disable=line-too-long
-from webviz_subsurface.plugins._simulation_time_series._views._subplot_view._utils.vector_statistics import (
-    create_vectors_statistics_df,
-)
-
-# *******************************************************************
-#####################################################################
-#
-# CONFIGURE TESTDATA
-#
-#####################################################################
-# *******************************************************************
-
-# fmt: off
-INPUT_YEAR_2020_DF = pd.DataFrame(
-    columns = ["DATE", "REAL",  "A", "B"],
-    data = [
-        [datetime.datetime(2020,1,1),  1,      10.0,       50.0    ],
-        [datetime.datetime(2020,2,1),  1,      100.0,      500.0   ],
-        [datetime.datetime(2020,3,1),  1,      1000.0,     5000.0  ],
-        [datetime.datetime(2020,4,1),  1,      10000.0,    50000.0 ],
-        [datetime.datetime(2020,5,1),  1,      100000.0,   500000.0],
-        [datetime.datetime(2020,1,1),  2,      20.0,       60.0    ],
-        [datetime.datetime(2020,2,1),  2,      200.0,      600.0   ],
-        [datetime.datetime(2020,3,1),  2,      2000.0,     6000.0  ],
-        [datetime.datetime(2020,4,1),  2,      20000.0,    60000.0 ],
-        [datetime.datetime(2020,5,1),  2,      200000.0,   600000.0],
-        [datetime.datetime(2020,1,1),  4,      30.0,       70.0    ],
-        [datetime.datetime(2020,2,1),  4,      300.0,      700.0   ],
-        [datetime.datetime(2020,3,1),  4,      3000.0,     7000.0  ],
-        [datetime.datetime(2020,4,1),  4,      30000.0,    70000.0 ],
-        [datetime.datetime(2020,5,1),  4,      300000.0,   700000.0],
-        [datetime.datetime(2020,1,1),  5,      40.0,       80.0    ],
-        [datetime.datetime(2020,2,1),  5,      400.0,      800.0   ],
-        [datetime.datetime(2020,3,1),  5,      4000.0,     8000.0  ],
-        [datetime.datetime(2020,4,1),  5,      40000.0,    80000.0 ],
-        [datetime.datetime(2020,5,1),  5,      400000.0,   800000.0]
-    ]
-)
-make_date_column_datetime_object(INPUT_YEAR_2020_DF)
-
-# pylint: disable=line-too-long
-# Columns are:
-#           ["DATE", "A"                             "B"                          ]
-#           [        MEAN, MIN, MAX, P10, P90, P50,  MEAN, MIN, MAX, P10, P90, P50]
-# NOTE: P10 is 90 percentil and P90 is 10 percentile for oil standard
-EXPECTED_STATISTICS_YEAR_2020_DF = pd.DataFrame(
-    columns = pd.MultiIndex.from_tuples([
-        ("DATE", ""),
-        ("A", StatisticsOptions.MEAN),
-        ("A", StatisticsOptions.MIN),
-        ("A", StatisticsOptions.MAX),
-        ("A", StatisticsOptions.P10),
-        ("A", StatisticsOptions.P90),
-        ("A", StatisticsOptions.P50),
-        ("B", StatisticsOptions.MEAN),
-        ("B", StatisticsOptions.MIN),
-        ("B", StatisticsOptions.MAX),
-        ("B", StatisticsOptions.P10),
-        ("B", StatisticsOptions.P90),
-        ("B", StatisticsOptions.P50),
-    ]),
-    data = [
-        [datetime.datetime(2020,1,1), 25.0,     10.0,     40.0,     37.0,     13.0,     25.0,     65.0,     50.0,     80.0,     77.0,     53.0,     65.0    ],
-        [datetime.datetime(2020,2,1), 250.0,    100.0,    400.0,    370.0,    130.0,    250.0,    650.0,    500.0,    800.0,    770.0,    530.0,    650.0   ],
-        [datetime.datetime(2020,3,1), 2500.0,   1000.0,   4000.0,   3700.0,   1300.0,   2500.0,   6500.0,   5000.0,   8000.0,   7700.0,   5300.0,   6500.0  ],
-        [datetime.datetime(2020,4,1), 25000.0,  10000.0,  40000.0,  37000.0,  13000.0,  25000.0,  65000.0,  50000.0,  80000.0,  77000.0,  53000.0,  65000.0 ],
-        [datetime.datetime(2020,5,1), 250000.0, 100000.0, 400000.0, 370000.0, 130000.0, 250000.0, 650000.0, 500000.0, 800000.0, 770000.0, 530000.0, 650000.0],
-    ]
-)
-make_date_column_datetime_object(EXPECTED_STATISTICS_YEAR_2020_DF)
-# fmt: on
-
-# Dates AFTER year 2262!
-INPUT_YEAR_2263_DATES = pd.Series(
-    [
-        datetime.datetime(2263, 1, 1),
-        datetime.datetime(2263, 2, 1),
-        datetime.datetime(2263, 3, 1),
-        datetime.datetime(2263, 4, 1),
-        datetime.datetime(2263, 5, 1),
-        datetime.datetime(2263, 1, 1),
-        datetime.datetime(2263, 2, 1),
-        datetime.datetime(2263, 3, 1),
-        datetime.datetime(2263, 4, 1),
-        datetime.datetime(2263, 5, 1),
-        datetime.datetime(2263, 1, 1),
-        datetime.datetime(2263, 2, 1),
-        datetime.datetime(2263, 3, 1),
-        datetime.datetime(2263, 4, 1),
-        datetime.datetime(2263, 5, 1),
-        datetime.datetime(2263, 1, 1),
-        datetime.datetime(2263, 2, 1),
-        datetime.datetime(2263, 3, 1),
-        datetime.datetime(2263, 4, 1),
-        datetime.datetime(2263, 5, 1),
-    ]
-)
-EXPECTED_YEAR_2263_DATES = pd.Series(
-    [
-        datetime.datetime(2263, 1, 1),
-        datetime.datetime(2263, 2, 1),
-        datetime.datetime(2263, 3, 1),
-        datetime.datetime(2263, 4, 1),
-        datetime.datetime(2263, 5, 1),
-    ]
-)
-# NOTE: datetime.datetime after year 2262 is not converted to pd.Timestamp, thus
-# no need to make date column datetime object
-INPUT_YEAR_2263_DF = INPUT_YEAR_2020_DF.copy()
-INPUT_YEAR_2263_DF["DATE"] = INPUT_YEAR_2263_DATES
-EXPECTED_STATISTICS_YEAR_2263_DF = EXPECTED_STATISTICS_YEAR_2020_DF.copy()
-EXPECTED_STATISTICS_YEAR_2263_DF["DATE"] = EXPECTED_YEAR_2263_DATES
-
-# *******************************************************************
-#####################################################################
-#
-# UNIT TESTS
-#
-#####################################################################
-# *******************************************************************
-
-TEST_VALID_CASES = [
-    pytest.param(INPUT_YEAR_2020_DF, EXPECTED_STATISTICS_YEAR_2020_DF),
-    pytest.param(INPUT_YEAR_2263_DF, EXPECTED_STATISTICS_YEAR_2263_DF),
-]
-
-
-@pytest.mark.parametrize("input_df, expected_statistics_df", TEST_VALID_CASES)
-def test_create_vectors_statistics_df(
-    input_df: pd.DataFrame, expected_statistics_df: pd.DataFrame
-) -> None:
-    statistics_df = create_vectors_statistics_df(input_df)
-
-    assert_frame_equal(statistics_df, expected_statistics_df)
-
-
-def test_create_vectors_statstics_df_invalid_input() -> None:
-    """Test assert check assert_date_column_is_datetime_object() in
-    webviz_subsurface._utils.dataframe_utils.py
-    """
-    # fmt: off
-    input_df = pd.DataFrame(
-        columns=["DATE", "REAL", "A", "B"],
-        data=[
-            [pd.Timestamp(2020, 1, 1), 1, 10.0,     50.0    ],
-            [pd.Timestamp(2020, 2, 1), 1, 100.0,    500.0   ],
-            [pd.Timestamp(2020, 3, 1), 1, 1000.0,   5000.0  ],
-            [pd.Timestamp(2020, 4, 1), 1, 10000.0,  50000.0 ],
-            [pd.Timestamp(2020, 5, 1), 1, 100000.0, 500000.0],
-        ],
-    )
-    # fmt: on
-
-    with pytest.raises(ValueError) as err:
-        create_vectors_statistics_df(input_df)
-    assert (
-        str(err.value)
-        == '"DATE"-column in dataframe is not on datetime.datetime format!'
-    )

tests/unit_tests/plugin_tests/test_tornado_data.py

@@ -1,130 +0,0 @@
-# pylint: disable=protected-access
-import pandas as pd
-import pytest
-
-from webviz_subsurface._components.tornado._tornado_data import (
-    SensitivityType,
-    TornadoData,
-)
-
-
-def test_tornado_data_init():
-    # fmt: off
-    input_data = [
-        ["REAL",  "SENSNAME",   "SENSCASE",  "SENSTYPE", "VALUE" ],
-        [     0,         "A",    "p10_p90", "mc"       ,     10.0],
-        [     1,         "A",    "p10_p90", "mc"       ,     20.0],
-        [     2,         "B",       "deep", "scalar"   ,     5.0],
-        [     3,         "B",       "deep", "scalar"   ,     6.0],
-        [     4,         "C",    "shallow", "scalar"   ,     25.0],
-        [     5,         "C",    "shallow", "scalar"   ,     26.0],
-        [     6,         "D", "simulation", "mc"       ,     9.0],
-        [     7,         "D", "simulation", "mc"       ,     11.0],
-    ]
-    # fmt: on
-
-    input_df = pd.DataFrame(input_data[1:], columns=input_data[0])
-
-    with pytest.raises(ValueError) as exc:
-        TornadoData(dframe=input_df)
-    assert exc.value.args[0] == "Reference SENSNAME rms_seed not in input data"
-
-    tornado_data = TornadoData(dframe=input_df, reference="A")
-
-    assert tornado_data._calculate_ref_average(input_df) == 15.0
-    avg_list = tornado_data._calculate_sensitivity_averages(input_df)
-    assert avg_list[0] == {
-        "sensname": "A",
-        "senscase": "P90",
-        "values": 11.0,
-        "values_ref": -26.666666666666668,
-        "reals": [0],
-        "senstype": "mc",
-    }
-    low_high_list = tornado_data._calculate_tornado_low_high_list(avg_list)
-    assert low_high_list[0] == {
-        "low": -26.666666666666668,
-        "low_base": 0,
-        "low_label": "P90",
-        "low_tooltip": -26.666666666666668,
-        "true_low": 11.0,
-        "low_reals": [0],
-        "sensname": "A",
-        "high": 26.666666666666668,
-        "high_base": 0,
-        "high_label": "P10",
-        "high_tooltip": 26.666666666666668,
-        "true_high": 19.0,
-        "high_reals": [1],
-        "senstype": "mc",
-    }
-
-    tornado_data = TornadoData(
-        dframe=input_df, reference="B", scale="Absolute", cutbyref=True
-    )
-    assert tornado_data._calculate_tornado_table(input_df).to_dict("records") == [
-        {
-            "low": 0.0,
-            "low_base": 5.5,
-            "low_label": "P90",
-            "low_tooltip": 5.5,
-            "true_low": 11.0,
-            "low_reals": [],
-            "sensname": "A",
-            "high": 8.0,
-            "high_base": 5.5,
-            "high_label": "P10",
-            "high_tooltip": 13.5,
-            "true_high": 19.0,
-            "high_reals": [0, 1],
-            "senstype": SensitivityType.MONTE_CARLO,
-        },
-        {
-            "low": 0.0,
-            "low_base": 0.0,
-            "low_label": None,
-            "low_tooltip": 0.0,
-            "true_low": 5.5,
-            "low_reals": [],
-            "sensname": "B",
-            "high": 0.0,
-            "high_base": 0.0,
-            "high_label": "deep",
-            "high_tooltip": 0.0,
-            "true_high": 5.5,
-            "high_reals": [2, 3],
-            "senstype": SensitivityType.SCALAR,
-        },
-        {
-            "low": 0.0,
-            "low_base": 0.0,
-            "low_label": None,
-            "low_tooltip": 0.0,
-            "true_low": 5.5,
-            "low_reals": [],
-            "sensname": "C",
-            "high": 20.0,
-            "high_base": 0.0,
-            "high_label": "shallow",
-            "high_tooltip": 20.0,
-            "true_high": 25.5,
-            "high_reals": [4, 5],
-            "senstype": SensitivityType.SCALAR,
-        },
-        {
-            "low": 0.0,
-            "low_base": 3.6999999999999993,
-            "low_label": "P90",
-            "low_tooltip": 3.6999999999999993,
-            "true_low": 9.2,
-            "low_reals": [],
-            "sensname": "D",
-            "high": 1.6000000000000014,
-            "high_base": 3.6999999999999993,
-            "high_label": "P10",
-            "high_tooltip": 5.300000000000001,
-            "true_high": 10.8,
-            "high_reals": [6, 7],
-            "senstype": SensitivityType.MONTE_CARLO,
-        },
-    ]

tests/unit_tests/plugin_tests/test_well_completions.py

@@ -1,158 +0,0 @@
-import numpy as np
-import pandas as pd
-import pytest
-from pandas._testing import assert_frame_equal
-
-from webviz_subsurface._datainput.well_completions import remove_invalid_colors
-from webviz_subsurface.plugins._well_completions._business_logic import (
-    extract_stratigraphy,
-    merge_compdat_and_connstatus,
-)
-
-
-def test_remove_invalid_colors():
-    """Tests that colors that are not 6 digit hexadecimal are removed from the lyr
-    parse zone list
-    """
-    zonelist = [
-        {"name": "ZoneA", "color": "#FFFFFF"},
-        {"name": "ZoneB", "color": "#FFF"},
-    ]
-    assert remove_invalid_colors(zonelist) == [
-        {"name": "ZoneA", "color": "#FFFFFF"},
-        {
-            "name": "ZoneB",
-        },
-    ]
-
-
-def test_extract_stratigraphy():
-    """Checks that the merging of the layer_zone_mapping and the stratigraphy is
-    correct and that the colors are added following the correct prioritization
-    rules
-    """
-    zone_names = ["ZoneA.1", "ZoneA.2", "ZoneB.1"]
-    stratigraphy = [
-        {
-            "name": "ZoneA",
-            "color": "#000000",
-            "subzones": [{"name": "ZoneA.1"}, {"name": "ZoneA.2"}, {"name": "ZoneA.3"}],
-        },
-        {
-            "name": "ZoneB",
-            "subzones": [{"name": "ZoneB.1", "subzones": [{"name": "ZoneB.1.1"}]}],
-        },
-        {
-            "name": "ZoneC",
-        },
-    ]
-    zone_color_mapping = {"ZoneA": "#111111", "ZoneA.1": "#222222"}
-    theme_colors = ["#FFFFFF"]
-
-    result = extract_stratigraphy(
-        zone_names, stratigraphy, zone_color_mapping, theme_colors
-    )
-    assert result == [
-        {
-            "name": "ZoneA",
-            "color": "#000000",  # colors given in the stratigraphy has priority
-            "subzones": [
-                {
-                    "name": "ZoneA.1",
-                    "color": "#222222",  # color from zone_color_mapping
-                },
-                {"name": "ZoneA.2", "color": "#FFFFFF"}  # color from theme_colors
-                # ZoneA.3 is not here because it is not in the layer_zone_mapping
-            ],
-        },
-        {
-            "name": "ZoneB",
-            "color": "#808080",  # Since it's not a leaf, color is set to grey
-            "subzones": [
-                {
-                    "name": "ZoneB.1",
-                    "color": "#FFFFFF"  # color from theme_colors
-                    # No subzones here because ZoneB.1.1 is not in the layer_zone_mapping
-                }
-            ],
-        },
-        # ZoneC is removed since it's not in the layer_zone_mapping
-    ]
-
-
-def test_extract_stratigraphy_errors():
-    """Checks that a ValueError is raised when a Zone is in the layer_zone_mapping, but
-    not in the stratigraphy.
-    """
-    layer_zone_mapping = {1: "ZoneA", 2: "ZoneB", 3: "ZoneD"}
-    stratigraphy = [
-        {
-            "name": "ZoneA",
-        },
-        {
-            "name": "ZoneB",
-        },
-    ]
-    with pytest.raises(ValueError):
-        extract_stratigraphy(layer_zone_mapping, stratigraphy, {}, [])
-
-
-def test_merge_compdat_and_connstatus():
-    """Tests the functionality of the merge_compdat_and_connstatus function.
-
-    The following functionality is covered:
-    * The two first rows of df_compdat is replaced with the two first from
-    df_connstatus, except for the KH (which is only available in df_compdat).
-    KH is taken from the first of the two compdat rows
-    * The A2 well is not available in df_connstatus and will be taken as is
-    from df_compdat
-    * the fourth row in df_compdat (WELL: A1, REAL:1) is ignored because A1 is
-    in df_connstatus, but not REAL 1. We don't mix compdat and connstatus data
-    for the same well
-    * The fourth row in df_compdat has KH=Nan. This will be 0 in the output
-    """
-    df_compdat = pd.DataFrame(
-        data={
-            "DATE": [
-                "2021-01-01",
-                "2021-05-01",
-                "2021-01-01",
-                "2021-01-01",
-                "2022-01-01",
-            ],
-            "REAL": [0, 0, 0, 1, 0],
-            "WELL": ["A1", "A1", "A2", "A1", "A3"],
-            "I": [1, 1, 1, 1, 1],
-            "J": [1, 1, 1, 1, 1],
-            "K1": [1, 1, 1, 1, 1],
-            "OP/SH": ["SHUT", "OPEN", "OPEN", "OPEN", "OPEN"],
-            "KH": [100, 1000, 10, 100, np.nan],
-        }
-    )
-    df_connstatus = pd.DataFrame(
-        data={
-            "DATE": ["2021-03-01", "2021-08-01", "2021-01-01"],
-            "REAL": [0, 0, 0],
-            "WELL": ["A1", "A1", "A3"],
-            "I": [1, 1, 1],
-            "J": [1, 1, 1],
-            "K1": [1, 1, 1],
-            "OP/SH": ["OPEN", "SHUT", "OPEN"],
-        }
-    )
-    df_output = pd.DataFrame(
-        data={
-            "DATE": ["2021-03-01", "2021-08-01", "2021-01-01", "2021-01-01"],
-            "REAL": [0, 0, 0, 0],
-            "WELL": ["A1", "A1", "A3", "A2"],
-            "I": [1, 1, 1, 1],
-            "J": [1, 1, 1, 1],
-            "K1": [1, 1, 1, 1],
-            "OP/SH": ["OPEN", "SHUT", "OPEN", "OPEN"],
-            "KH": [100.0, 100.0, 0.0, 10.0],
-        }
-    )
-    df_result = merge_compdat_and_connstatus(df_compdat, df_connstatus)
-    assert_frame_equal(
-        df_result, df_output, check_like=True
-    )  # Ignore order of rows and columns