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_delta_ensemble_utils.py

@@ -1,222 +0,0 @@
-from typing import List
-
-import pytest
-
-from webviz_subsurface._utils.ensemble_summary_provider_set import (
-    EnsembleSummaryProviderSet,
-)
-
-# pylint: disable=line-too-long
-from webviz_subsurface.plugins._simulation_time_series._views._subplot_view._utils.delta_ensemble_utils import (
-    DeltaEnsemble,
-    create_delta_ensemble_name,
-    create_delta_ensemble_name_dict,
-    create_delta_ensemble_names,
-    create_delta_ensemble_provider_pair,
-    is_delta_ensemble_providers_in_provider_set,
-)
-
-from ....mocks.ensemble_summary_provider_dummy import EnsembleSummaryProviderDummy
-
-# *******************************************************************
-#####################################################################
-#
-# CONFIGURE TESTDATA
-#
-#####################################################################
-# *******************************************************************
-
-
-class EnsembleSummaryProviderMock(EnsembleSummaryProviderDummy):
-    """Ensemble summary provider mock for testing
-
-    Note empty list returned in override methods, only to allow constructing
-    EnsembleSummaryProviderSet objects!
-    """
-
-    def __init__(self, name: str) -> None:
-        super().__init__()
-        self._name = name
-
-    def get_name(self) -> str:
-        return self._name
-
-    ########################################
-    #
-    # Override methods
-    #
-    ########################################
-    def vector_names(self) -> List[str]:
-        """Return empty list only to allow constructing EnsembleSummaryProviderSet object"""
-        return []
-
-    def realizations(self) -> List[int]:
-        """Return empty list only to allow constructing EnsembleSummaryProviderSet object"""
-        return []
-
-
-# *******************************************************************
-#####################################################################
-#
-# UNIT TESTS
-#
-#####################################################################
-# *******************************************************************
-
-
-def test_create_delta_ensemble_name() -> None:
-    assert (
-        create_delta_ensemble_name(
-            DeltaEnsemble(ensemble_a="first name", ensemble_b="second name")
-        )
-        == "(first name)-(second name)"
-    )
-    assert (
-        create_delta_ensemble_name(
-            DeltaEnsemble(ensemble_a="first-Name", ensemble_b="second-Name")
-        )
-        == "(first-Name)-(second-Name)"
-    )
-    assert (
-        create_delta_ensemble_name(
-            DeltaEnsemble(ensemble_a="ens-0", ensemble_b="ens-3")
-        )
-        == "(ens-0)-(ens-3)"
-    )
-
-
-def test_create_delta_ensemble_names() -> None:
-    first_delta_ensemble = DeltaEnsemble(
-        ensemble_a="first name", ensemble_b="second name"
-    )
-    second_delta_ensemble = DeltaEnsemble(
-        ensemble_a="first-Name", ensemble_b="second-Name"
-    )
-    third_delta_ensemble = DeltaEnsemble(ensemble_a="ens-0", ensemble_b="ens-3")
-
-    assert create_delta_ensemble_names(
-        [first_delta_ensemble, second_delta_ensemble, third_delta_ensemble]
-    ) == ["(first name)-(second name)", "(first-Name)-(second-Name)", "(ens-0)-(ens-3)"]
-
-
-def test_create_delta_ensemble_name_dict() -> None:
-    first_delta_ensemble = DeltaEnsemble(
-        ensemble_a="first name", ensemble_b="second name"
-    )
-    second_delta_ensemble = DeltaEnsemble(
-        ensemble_a="first-Name", ensemble_b="second-Name"
-    )
-    third_delta_ensemble = DeltaEnsemble(ensemble_a="ens-0", ensemble_b="ens-3")
-
-    expected_delta_ensemble_name_dict = {
-        "(first name)-(second name)": first_delta_ensemble,
-        "(first-Name)-(second-Name)": second_delta_ensemble,
-        "(ens-0)-(ens-3)": third_delta_ensemble,
-    }
-
-    assert (
-        create_delta_ensemble_name_dict(
-            [first_delta_ensemble, second_delta_ensemble, third_delta_ensemble]
-        )
-        == expected_delta_ensemble_name_dict
-    )
-
-
-def test_is_delta_ensemble_providers_in_provider_set() -> None:
-    provider_set = EnsembleSummaryProviderSet(
-        {
-            "First provider": EnsembleSummaryProviderMock("First mock"),
-            "Second provider": EnsembleSummaryProviderMock("Second mock"),
-        }
-    )
-
-    valid_delta_ensemble = DeltaEnsemble(
-        ensemble_a="First provider", ensemble_b="Second provider"
-    )
-    invalid_delta_ensemble = DeltaEnsemble(
-        ensemble_a="First provider", ensemble_b="Third provider"
-    )
-
-    assert is_delta_ensemble_providers_in_provider_set(
-        valid_delta_ensemble, provider_set
-    )
-    assert not is_delta_ensemble_providers_in_provider_set(
-        invalid_delta_ensemble, provider_set
-    )
-
-
-def test_create_delta_ensemble_provider_pair() -> None:
-    provider_set = EnsembleSummaryProviderSet(
-        {
-            "First provider": EnsembleSummaryProviderMock("First mock"),
-            "Second provider": EnsembleSummaryProviderMock("Second mock"),
-            "Third provider": EnsembleSummaryProviderMock(("Third mock")),
-        }
-    )
-
-    first_delta_ensemble = DeltaEnsemble(
-        ensemble_a="First provider", ensemble_b="Third provider"
-    )
-    second_delta_ensemble = DeltaEnsemble(
-        ensemble_a="Third provider", ensemble_b="Second provider"
-    )
-
-    first_provider_pair = create_delta_ensemble_provider_pair(
-        first_delta_ensemble, provider_set
-    )
-    second_provider_pair = create_delta_ensemble_provider_pair(
-        second_delta_ensemble, provider_set
-    )
-
-    assert len(first_provider_pair) == 2
-    ensemble_a = first_provider_pair[0]
-    ensemble_b = first_provider_pair[1]
-    if not isinstance(ensemble_a, EnsembleSummaryProviderMock):
-        pytest.fail(
-            f'Expected "{ensemble_a}" in second provider pair to be of type '
-            "EnsembleSummaryProviderMock"
-        )
-    if not isinstance(ensemble_b, EnsembleSummaryProviderMock):
-        pytest.fail(
-            f'Expected "{ensemble_b}" in second provider pair to be of type '
-            "EnsembleSummaryProviderMock"
-        )
-    assert ensemble_a.get_name() == "First mock"
-    assert ensemble_b.get_name() == "Third mock"
-
-    assert len(second_provider_pair) == 2
-    ensemble_a = second_provider_pair[0]
-    ensemble_b = second_provider_pair[1]
-    if not isinstance(ensemble_a, EnsembleSummaryProviderMock):
-        pytest.fail(
-            f'Expected "{ensemble_a}" in second provider pair to be of type '
-            "EnsembleSummaryProviderMock"
-        )
-    if not isinstance(ensemble_b, EnsembleSummaryProviderMock):
-        pytest.fail(
-            f'Expected "{ensemble_b}" in second provider pair to be of type '
-            "EnsembleSummaryProviderMock"
-        )
-    assert ensemble_a.get_name() == "Third mock"
-    assert ensemble_b.get_name() == "Second mock"
-
-
-def test_create_delta_ensemble_provider_pair_invalid_ensemble() -> None:
-    provider_set = EnsembleSummaryProviderSet(
-        {
-            "First provider": EnsembleSummaryProviderMock("First mock"),
-            "Second provider": EnsembleSummaryProviderMock("Second mock"),
-        }
-    )
-
-    # Expect ValueError when retrieveing ensemble_b is invalid
-    with pytest.raises(ValueError):
-        invalid_provider = "Invalid provider"
-        invalid_provider_pair = DeltaEnsemble(
-            ensemble_a="First provider", ensemble_b=invalid_provider
-        )
-        create_delta_ensemble_provider_pair(invalid_provider_pair, provider_set)
-        pytest.fail(
-            "Expected ValueError when creating delta ensemble provider pair, as ensemble b = "
-            f'"{invalid_provider}" is not existing in provider set!'
-        )

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

@@ -1,319 +0,0 @@
-import datetime
-
-import pandas as pd
-import pytest
-from pandas._testing import assert_frame_equal
-from webviz_subsurface_components.VectorCalculatorWrapper import (
-    ExpressionInfo,
-    VariableVectorMapInfo,
-)
-
-from webviz_subsurface._utils.dataframe_utils import make_date_column_datetime_object
-
-# pylint: disable = line-too-long
-from webviz_subsurface.plugins._simulation_time_series._views._subplot_view._utils.derived_vectors_accessor.derived_delta_ensemble_vectors_accessor_impl import (
-    DerivedDeltaEnsembleVectorsAccessorImpl,
-)
-
-from ..mocks.derived_vectors_accessor_ensemble_summary_provider_mock import (
-    EnsembleSummaryProviderMock,
-)
-
-# *******************************************************************
-#####################################################################
-#
-# CONFIGURE TESTDATA
-#
-#####################################################################
-# *******************************************************************
-
-# fmt: off
-# Ensemble A
-INPUT_A_DF = pd.DataFrame(
-    columns = ["DATE", "REAL",  "A", "B"],
-    data = [
-        [datetime.datetime(2000,1,1),  1, 10.0,   500.0  ],
-        [datetime.datetime(2000,2,1),  1, 20.0,   1000.0 ],
-        [datetime.datetime(2000,3,1),  1, 30.0,   1500.0 ],
-        [datetime.datetime(2000,1,1),  2, 60.0,   3000.0 ],
-        [datetime.datetime(2000,2,1),  2, 70.0,   4000.0 ],
-        [datetime.datetime(2000,3,1),  2, 80.0,   5000.0 ],
-        [datetime.datetime(2000,1,1),  4, 110.0,  10000.0],
-        [datetime.datetime(2000,2,1),  4, 120.0,  12000.0],
-        [datetime.datetime(2000,3,1),  4, 130.0,  14000.0],
-    ]
-)
-
-# Ensemble B
-INPUT_B_DF = pd.DataFrame(
-    columns = ["DATE", "REAL",  "A", "B"],
-    data = [
-        [datetime.datetime(2000,1,1),  1, 1.0,   50.0  ],
-        [datetime.datetime(2000,2,1),  1, 2.0,   100.0 ],
-        [datetime.datetime(2000,3,1),  1, 3.0,   150.0 ],
-        [datetime.datetime(2000,1,1),  2, 6.0,   300.0 ],
-        [datetime.datetime(2000,2,1),  2, 7.0,   400.0 ],
-        [datetime.datetime(2000,3,1),  2, 8.0,   500.0 ],
-        [datetime.datetime(2000,1,1),  4, 11.0,  1000.0],
-        [datetime.datetime(2000,2,1),  4, 12.0,  1200.0],
-        [datetime.datetime(2000,3,1),  4, 13.0,  1400.0],
-    ]
-)
-
-# Delta between A and B DF (A-B)
-EXPECTED_DELTA_DF = pd.DataFrame(
-    columns = ["DATE", "REAL",  "A", "B"],
-    data = [
-        [datetime.datetime(2000,1,1),  1, 9.0,    450.0  ],
-        [datetime.datetime(2000,2,1),  1, 18.0,   900.0  ],
-        [datetime.datetime(2000,3,1),  1, 27.0,   1350.0 ],
-        [datetime.datetime(2000,1,1),  2, 54.0,   2700.0 ],
-        [datetime.datetime(2000,2,1),  2, 63.0,   3600.0 ],
-        [datetime.datetime(2000,3,1),  2, 72.0,   4500.0 ],
-        [datetime.datetime(2000,1,1),  4, 99.0,   9000.0 ],
-        [datetime.datetime(2000,2,1),  4, 108.0,  10800.0],
-        [datetime.datetime(2000,3,1),  4, 117.0,  12600.0],
-    ]
-)
-
-# PER_INTVL_ calc for col "B" of Delta
-EXPECTED_DELTA_INVTL_DF = pd.DataFrame(
-    columns = ["DATE", "REAL",  "PER_INTVL_B"],
-    data = [
-        [datetime.datetime(2000,1,1),  1, 450.0 ],
-        [datetime.datetime(2000,2,1),  1, 450.0 ],
-        [datetime.datetime(2000,3,1),  1, 0.0   ],
-        [datetime.datetime(2000,1,1),  2, 900.0 ],
-        [datetime.datetime(2000,2,1),  2, 900.0 ],
-        [datetime.datetime(2000,3,1),  2, 0.0   ],
-        [datetime.datetime(2000,1,1),  4, 1800.0],
-        [datetime.datetime(2000,2,1),  4, 1800.0],
-        [datetime.datetime(2000,3,1),  4, 0.0   ],
-    ]
-)
-
-# Sum of col "A" and "B" in Delta
-EXPECTED_SUM_A_AND_B_DF = pd.DataFrame(
-    columns = ["DATE", "REAL",  "Sum A and B"],
-    data = [
-        [datetime.datetime(2000,1,1),  1, 459.0  ],
-        [datetime.datetime(2000,2,1),  1, 918.0  ],
-        [datetime.datetime(2000,3,1),  1, 1377.0 ],
-        [datetime.datetime(2000,1,1),  2, 2754.0 ],
-        [datetime.datetime(2000,2,1),  2, 3663.0 ],
-        [datetime.datetime(2000,3,1),  2, 4572.0 ],
-        [datetime.datetime(2000,1,1),  4, 9099.0 ],
-        [datetime.datetime(2000,2,1),  4, 10908.0],
-        [datetime.datetime(2000,3,1),  4, 12717.0],
-    ]
-)
-make_date_column_datetime_object(INPUT_A_DF)
-make_date_column_datetime_object(INPUT_B_DF)
-make_date_column_datetime_object(EXPECTED_DELTA_DF)
-make_date_column_datetime_object(EXPECTED_DELTA_INVTL_DF)
-make_date_column_datetime_object(EXPECTED_SUM_A_AND_B_DF)
-
-
-# Dates AFTER year 2262!
-AFTER_2262_DATES = pd.Series(
-    [
-        datetime.datetime(2265,1,1),
-        datetime.datetime(2265,2,1),
-        datetime.datetime(2265,3,1),
-        datetime.datetime(2265,1,1),
-        datetime.datetime(2265,2,1),
-        datetime.datetime(2265,3,1),
-        datetime.datetime(2265,1,1),
-        datetime.datetime(2265,2,1),
-        datetime.datetime(2265,3,1),
-    ]
-)
-# NOTE: datetime.datetime after year 2262 is not converted to pd.Timestamp, thus
-# no need to make date column datetime object
-INPUT_A_AFTER_2262_DF = INPUT_A_DF.copy()
-INPUT_A_AFTER_2262_DF["DATE"] = AFTER_2262_DATES
-INPUT_B_AFTER_2262_DF = INPUT_B_DF.copy()
-INPUT_B_AFTER_2262_DF["DATE"] = AFTER_2262_DATES
-EXPECTED_DELTA_AFTER_2262_DF = EXPECTED_DELTA_DF.copy()
-EXPECTED_DELTA_AFTER_2262_DF["DATE"] = AFTER_2262_DATES
-EXPECTED_DELTA_INVTL_AFTER_2262_DF = EXPECTED_DELTA_INVTL_DF.copy()
-EXPECTED_DELTA_INVTL_AFTER_2262_DF["DATE"] = AFTER_2262_DATES
-EXPECTED_SUM_A_AND_B_AFTER_2262_DF = EXPECTED_SUM_A_AND_B_DF.copy()
-EXPECTED_SUM_A_AND_B_AFTER_2262_DF["DATE"] = AFTER_2262_DATES
-
-# fmt: on
-TEST_EXPRESSION = ExpressionInfo(
-    name="Sum A and B",
-    expression="x+y",
-    id="TestId",
-    variableVectorMap=[
-        VariableVectorMapInfo(variableName="x", vectorName=["A"]),
-        VariableVectorMapInfo(variableName="y", vectorName=["B"]),
-    ],
-    isValid=True,
-    isDeletable=False,
-)
-
-TEST_ACCESSOR = DerivedDeltaEnsembleVectorsAccessorImpl(
-    name="Test accessor",
-    provider_pair=(
-        EnsembleSummaryProviderMock(INPUT_A_DF),
-        EnsembleSummaryProviderMock(INPUT_B_DF),
-    ),
-    vectors=["A", "B", "PER_INTVL_B", "Sum A and B"],
-    expressions=[TEST_EXPRESSION],
-    resampling_frequency=None,
-)
-
-TEST_AFTER_2262_ACCESSOR = DerivedDeltaEnsembleVectorsAccessorImpl(
-    name="Test 2262 accessor",
-    provider_pair=(
-        EnsembleSummaryProviderMock(INPUT_A_AFTER_2262_DF),
-        EnsembleSummaryProviderMock(INPUT_B_AFTER_2262_DF),
-    ),
-    vectors=["A", "B", "PER_INTVL_B", "Sum A and B"],
-    expressions=[TEST_EXPRESSION],
-    resampling_frequency=None,
-)
-
-TEST_EMPTY_ACCESSOR = DerivedDeltaEnsembleVectorsAccessorImpl(
-    name="Empty provider accessor",
-    provider_pair=(
-        EnsembleSummaryProviderMock(pd.DataFrame()),
-        EnsembleSummaryProviderMock(pd.DataFrame()),
-    ),
-    vectors=["A", "B", "PER_INTVL_B", "Sum A and B"],
-    expressions=None,
-    resampling_frequency=None,
-)
-
-# *******************************************************************
-#####################################################################
-#
-# UNIT TESTS
-#
-#####################################################################
-# *******************************************************************
-
-TEST_STATUS_CASES = [
-    pytest.param(TEST_ACCESSOR, True),
-    pytest.param(TEST_AFTER_2262_ACCESSOR, True),
-    pytest.param(TEST_EMPTY_ACCESSOR, False),
-]
-TEST_GET_VECTOR_CASES = [
-    pytest.param(TEST_ACCESSOR, EXPECTED_DELTA_DF),
-    pytest.param(TEST_AFTER_2262_ACCESSOR, EXPECTED_DELTA_AFTER_2262_DF),
-]
-TEST_CREATE_PER_INTVL_PER_DAY_VECTOR_CASES = [
-    pytest.param(TEST_ACCESSOR, EXPECTED_DELTA_INVTL_DF),
-    pytest.param(TEST_AFTER_2262_ACCESSOR, EXPECTED_DELTA_INVTL_AFTER_2262_DF),
-]
-TEST_CREATE_CALCULATED_VECTOR_CASES = [
-    pytest.param(TEST_ACCESSOR, EXPECTED_SUM_A_AND_B_DF),
-    pytest.param(TEST_AFTER_2262_ACCESSOR, EXPECTED_SUM_A_AND_B_AFTER_2262_DF),
-]
-
-
-@pytest.mark.parametrize("test_accessor, expected_state", TEST_STATUS_CASES)
-def test_has_provider_vectors(
-    test_accessor: DerivedDeltaEnsembleVectorsAccessorImpl, expected_state: bool
-) -> None:
-    assert test_accessor.has_provider_vectors() == expected_state
-
-
-@pytest.mark.parametrize("test_accessor, expected_state", TEST_STATUS_CASES)
-def test_has_per_interval_and_per_day_vectors(
-    test_accessor: DerivedDeltaEnsembleVectorsAccessorImpl, expected_state: bool
-) -> None:
-    assert test_accessor.has_per_interval_and_per_day_vectors() == expected_state
-
-
-@pytest.mark.parametrize("test_accessor, expected_state", TEST_STATUS_CASES)
-def test_has_vector_calculator_expressions(
-    test_accessor: DerivedDeltaEnsembleVectorsAccessorImpl, expected_state: bool
-) -> None:
-    assert test_accessor.has_vector_calculator_expressions() == expected_state
-
-
-@pytest.mark.parametrize("test_accessor, expected_df", TEST_GET_VECTOR_CASES)
-def test_get_provider_vectors(
-    test_accessor: DerivedDeltaEnsembleVectorsAccessorImpl, expected_df: pd.DataFrame
-) -> None:
-    assert_frame_equal(expected_df, test_accessor.get_provider_vectors_df())
-
-
-@pytest.mark.parametrize("test_accessor, expected_df", TEST_GET_VECTOR_CASES)
-def test_get_provider_vectors_filter_realizations(
-    test_accessor: DerivedDeltaEnsembleVectorsAccessorImpl, expected_df: pd.DataFrame
-) -> None:
-    # Filter realizations
-    expected_reals_df = (
-        expected_df.loc[expected_df["REAL"].isin([2, 4])]
-        .reset_index()
-        .drop("index", axis=1)
-    )
-
-    test_df = test_accessor.get_provider_vectors_df(realizations=[2, 4])
-
-    assert_frame_equal(expected_reals_df, test_df)
-    assert list(set(test_df["REAL"].values)) == [2, 4]
-
-
-@pytest.mark.parametrize(
-    "test_accessor, expected_df", TEST_CREATE_PER_INTVL_PER_DAY_VECTOR_CASES
-)
-def test_create_per_interval_and_per_day_vectors_df(
-    test_accessor: DerivedDeltaEnsembleVectorsAccessorImpl, expected_df: pd.DataFrame
-) -> None:
-    assert_frame_equal(
-        expected_df, test_accessor.create_per_interval_and_per_day_vectors_df()
-    )
-
-
-@pytest.mark.parametrize(
-    "test_accessor, expected_df", TEST_CREATE_PER_INTVL_PER_DAY_VECTOR_CASES
-)
-def test_create_per_interval_and_per_day_vectors_df_filter_realizations(
-    test_accessor: DerivedDeltaEnsembleVectorsAccessorImpl, expected_df: pd.DataFrame
-) -> None:
-    # Filter realizations
-    expected_reals_df = (
-        expected_df.loc[expected_df["REAL"].isin([1, 4])]
-        .reset_index()
-        .drop("index", axis=1)
-    )
-
-    test_df = test_accessor.create_per_interval_and_per_day_vectors_df(
-        realizations=[1, 4]
-    )
-
-    assert_frame_equal(expected_reals_df, test_df)
-    assert list(set(test_df["REAL"].values)) == [1, 4]
-
-
-@pytest.mark.parametrize(
-    "test_accessor, expected_df", TEST_CREATE_CALCULATED_VECTOR_CASES
-)
-def test_create_calculated_vectors_df(
-    test_accessor: DerivedDeltaEnsembleVectorsAccessorImpl, expected_df: pd.DataFrame
-) -> None:
-    assert_frame_equal(expected_df, test_accessor.create_calculated_vectors_df())
-
-
-@pytest.mark.parametrize(
-    "test_accessor, expected_df", TEST_CREATE_CALCULATED_VECTOR_CASES
-)
-def test_create_calculated_vectors_df_filter_realizations(
-    test_accessor: DerivedDeltaEnsembleVectorsAccessorImpl, expected_df: pd.DataFrame
-) -> None:
-    # Filter realizations
-    expected_reals_df = (
-        expected_df.loc[expected_df["REAL"].isin([1, 2])]
-        .reset_index()
-        .drop("index", axis=1)
-    )
-
-    test_df = test_accessor.create_calculated_vectors_df(realizations=[1, 2])
-
-    assert_frame_equal(expected_reals_df, test_df)
-    assert list(set(test_df["REAL"].values)) == [1, 2]