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

@@ -1,322 +0,0 @@
-import datetime
-
-import pandas as pd
-import pytest
-from pandas._testing import assert_frame_equal
-
-from webviz_subsurface._providers import Frequency
-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.from_timeseries_cumulatives import (
-    calculate_from_resampled_cumulative_vectors_df,
-    create_per_day_vector_name,
-    create_per_interval_vector_name,
-    datetime_to_intervalstr,
-    get_cumulative_vector_name,
-    is_per_interval_or_per_day_vector,
-)
-
-# *******************************************************************
-#####################################################################
-#
-# CONFIGURE TESTDATA
-#
-#####################################################################
-# *******************************************************************
-
-# fmt: off
-# Monthly frequency - rate per day implies divide on days in month
-INPUT_WEEKLY_DF = pd.DataFrame(
-    columns=["DATE", "REAL", "A", "B"],
-    data=[
-        [datetime.datetime(2021, 1, 1),  1, 50.0,   250.0 ],
-        [datetime.datetime(2021, 1, 8),  1, 100.0,  500.0 ],
-        [datetime.datetime(2021, 1, 15), 1, 150.0,  750.0 ],
-        [datetime.datetime(2021, 1, 1),  2, 300.0,  350.0 ],
-        [datetime.datetime(2021, 1, 8),  2, 400.0,  700.0 ],
-        [datetime.datetime(2021, 1, 15), 2, 500.0,  1050.0],
-        [datetime.datetime(2021, 1, 1),  4, 1000.0, 450.0 ],
-        [datetime.datetime(2021, 1, 8),  4, 1200.0, 900.0 ],
-        [datetime.datetime(2021, 1, 15), 4, 1400.0, 1350.0],
-    ],
-)
-EXPECTED_PER_INTVL_WEEKLY_DF = pd.DataFrame(
-    columns=["DATE", "REAL", "PER_INTVL_A", "PER_INTVL_B"],
-    data=[
-        [datetime.datetime(2021, 1, 1),  1, 50.0,  250.0],
-        [datetime.datetime(2021, 1, 8),  1, 50.0,  250.0],
-        [datetime.datetime(2021, 1, 15), 1, 0.0,   0.0  ],
-        [datetime.datetime(2021, 1, 1),  2, 100.0, 350.0],
-        [datetime.datetime(2021, 1, 8),  2, 100.0, 350.0],
-        [datetime.datetime(2021, 1, 15), 2, 0.0,   0.0  ],
-        [datetime.datetime(2021, 1, 1),  4, 200.0, 450.0],
-        [datetime.datetime(2021, 1, 8),  4, 200.0, 450.0],
-        [datetime.datetime(2021, 1, 15), 4, 0.0,   0.0  ],
-    ],
-)
-EXPECTED_PER_DAY_WEEKLY_DF = pd.DataFrame(
-    columns=["DATE", "REAL", "PER_DAY_A", "PER_DAY_B"],
-    data=[
-        [datetime.datetime(2021, 1, 1),  1, 50.0/7.0,  250.0/7.0],
-        [datetime.datetime(2021, 1, 8),  1, 50.0/7.0,  250.0/7.0],
-        [datetime.datetime(2021, 1, 15), 1, 0.0,       0.0      ],
-        [datetime.datetime(2021, 1, 1),  2, 100.0/7.0, 350.0/7.0],
-        [datetime.datetime(2021, 1, 8),  2, 100.0/7.0, 350.0/7.0],
-        [datetime.datetime(2021, 1, 15), 2, 0.0,       0.0      ],
-        [datetime.datetime(2021, 1, 1),  4, 200.0/7.0, 450.0/7.0],
-        [datetime.datetime(2021, 1, 8),  4, 200.0/7.0, 450.0/7.0],
-        [datetime.datetime(2021, 1, 15), 4, 0.0,       0.0      ],
-    ],
-)
-# Convert date columns to datetime.datetime
-make_date_column_datetime_object(INPUT_WEEKLY_DF)
-make_date_column_datetime_object(EXPECTED_PER_INTVL_WEEKLY_DF)
-make_date_column_datetime_object(EXPECTED_PER_DAY_WEEKLY_DF)
-
-# Monthly frequency - rate per day implies divide on days in month
-INPUT_MONTHLY_DF = pd.DataFrame(
-    columns=["DATE", "REAL", "A", "B"],
-    data=[
-        [datetime.datetime(2021, 1, 1), 1, 50.0,   250.0 ],
-        [datetime.datetime(2021, 2, 1), 1, 100.0,  500.0 ],
-        [datetime.datetime(2021, 3, 1), 1, 150.0,  750.0 ],
-        [datetime.datetime(2021, 1, 1), 2, 300.0,  350.0 ],
-        [datetime.datetime(2021, 2, 1), 2, 400.0,  700.0 ],
-        [datetime.datetime(2021, 3, 1), 2, 500.0,  1050.0],
-        [datetime.datetime(2021, 1, 1), 4, 1000.0, 450.0 ],
-        [datetime.datetime(2021, 2, 1), 4, 1200.0, 900.0 ],
-        [datetime.datetime(2021, 3, 1), 4, 1400.0, 1350.0],
-    ],
-)
-EXPECTED_PER_INTVL_MONTHLY_DF = pd.DataFrame(
-    columns=["DATE", "REAL", "PER_INTVL_A", "PER_INTVL_B"],
-    data=[
-        [datetime.datetime(2021, 1, 1), 1, 50.0,  250.0],
-        [datetime.datetime(2021, 2, 1), 1, 50.0,  250.0],
-        [datetime.datetime(2021, 3, 1), 1, 0.0,   0.0  ],
-        [datetime.datetime(2021, 1, 1), 2, 100.0, 350.0],
-        [datetime.datetime(2021, 2, 1), 2, 100.0, 350.0],
-        [datetime.datetime(2021, 3, 1), 2, 0.0,   0.0  ],
-        [datetime.datetime(2021, 1, 1), 4, 200.0, 450.0],
-        [datetime.datetime(2021, 2, 1), 4, 200.0, 450.0],
-        [datetime.datetime(2021, 3, 1), 4, 0.0,   0.0  ],
-    ],
-)
-EXPECTED_PER_DAY_MONTHLY_DF = pd.DataFrame(
-    columns=["DATE", "REAL", "PER_DAY_A", "PER_DAY_B"],
-    data=[
-        [datetime.datetime(2021, 1, 1), 1, 50.0/31.0,  250.0/31.0],
-        [datetime.datetime(2021, 2, 1), 1, 50.0/28.0,  250.0/28.0],
-        [datetime.datetime(2021, 3, 1), 1, 0.0,        0.0       ],
-        [datetime.datetime(2021, 1, 1), 2, 100.0/31.0, 350.0/31.0],
-        [datetime.datetime(2021, 2, 1), 2, 100.0/28.0, 350.0/28.0],
-        [datetime.datetime(2021, 3, 1), 2, 0.0,        0.0       ],
-        [datetime.datetime(2021, 1, 1), 4, 200.0/31.0, 450.0/31.0],
-        [datetime.datetime(2021, 2, 1), 4, 200.0/28.0, 450.0/28.0],
-        [datetime.datetime(2021, 3, 1), 4, 0.0,        0.0       ],
-    ],
-)
-# Convert date columns to datetime.datetime
-make_date_column_datetime_object(INPUT_MONTHLY_DF)
-make_date_column_datetime_object(EXPECTED_PER_INTVL_MONTHLY_DF)
-make_date_column_datetime_object(EXPECTED_PER_DAY_MONTHLY_DF)
-
-# Yearly frequency - rate per day implies divide on days in year
-INPUT_YEARLY_DF = pd.DataFrame(
-    columns=["DATE", "REAL", "A", "B"],
-    data=[
-        [datetime.datetime(2021, 1, 1), 1, 50.0,   250.0 ],
-        [datetime.datetime(2022, 1, 1), 1, 100.0,  500.0 ],
-        [datetime.datetime(2023, 1, 1), 1, 150.0,  750.0 ],
-        [datetime.datetime(2021, 1, 1), 2, 300.0,  350.0 ],
-        [datetime.datetime(2022, 1, 1), 2, 400.0,  700.0 ],
-        [datetime.datetime(2023, 1, 1), 2, 500.0,  1050.0],
-        [datetime.datetime(2021, 1, 1), 4, 1000.0, 450.0 ],
-        [datetime.datetime(2022, 1, 1), 4, 1200.0, 900.0 ],
-        [datetime.datetime(2023, 1, 1), 4, 1400.0, 1350.0],
-    ],
-)
-EXPECTED_PER_INTVL_YEARLY_DF = pd.DataFrame(
-    columns=["DATE", "REAL", "PER_INTVL_A", "PER_INTVL_B"],
-    data=[
-        [datetime.datetime(2021, 1, 1), 1, 50.0,  250.0],
-        [datetime.datetime(2022, 1, 1), 1, 50.0,  250.0],
-        [datetime.datetime(2023, 1, 1), 1, 0.0,   0.0  ],
-        [datetime.datetime(2021, 1, 1), 2, 100.0, 350.0],
-        [datetime.datetime(2022, 1, 1), 2, 100.0, 350.0],
-        [datetime.datetime(2023, 1, 1), 2, 0.0,   0.0  ],
-        [datetime.datetime(2021, 1, 1), 4, 200.0, 450.0],
-        [datetime.datetime(2022, 1, 1), 4, 200.0, 450.0],
-        [datetime.datetime(2023, 1, 1), 4, 0.0,   0.0  ],
-    ],
-)
-EXPECTED_PER_DAY_YEARLY_DF = pd.DataFrame(
-    columns=["DATE", "REAL", "PER_DAY_A", "PER_DAY_B"],
-    data=[
-        [datetime.datetime(2021, 1, 1), 1, 50.0/365.0,  250.0/365.0],
-        [datetime.datetime(2022, 1, 1), 1, 50.0/365.0,  250.0/365.0],
-        [datetime.datetime(2023, 1, 1), 1, 0.0,         0.0        ],
-        [datetime.datetime(2021, 1, 1), 2, 100.0/365.0, 350.0/365.0],
-        [datetime.datetime(2022, 1, 1), 2, 100.0/365.0, 350.0/365.0],
-        [datetime.datetime(2023, 1, 1), 2, 0.0,         0.0        ],
-        [datetime.datetime(2021, 1, 1), 4, 200.0/365.0, 450.0/365.0],
-        [datetime.datetime(2022, 1, 1), 4, 200.0/365.0, 450.0/365.0],
-        [datetime.datetime(2023, 1, 1), 4, 0.0,         0.0        ],
-    ],
-)
-# Convert date columns to datetime.datetime
-make_date_column_datetime_object(INPUT_YEARLY_DF)
-make_date_column_datetime_object(EXPECTED_PER_INTVL_YEARLY_DF)
-make_date_column_datetime_object(EXPECTED_PER_DAY_YEARLY_DF)
-
-
-# Monthly frequency after year 2262 - rate per day implies divide on days in month
-AFTER_2262_MONTHLY_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!
-INPUT_MONTHLY_AFTER_2262_DF = INPUT_MONTHLY_DF.copy()
-INPUT_MONTHLY_AFTER_2262_DF["DATE"] = AFTER_2262_MONTHLY_DATES
-EXPECTED_PER_INTVL_MONTHLY_AFTER_2262_DF = EXPECTED_PER_INTVL_MONTHLY_DF.copy()
-EXPECTED_PER_INTVL_MONTHLY_AFTER_2262_DF["DATE"] = AFTER_2262_MONTHLY_DATES
-EXPECTED_PER_DAY_MONTHLY_AFTER_2262_DF = EXPECTED_PER_DAY_MONTHLY_DF.copy()
-EXPECTED_PER_DAY_MONTHLY_AFTER_2262_DF["DATE"] = AFTER_2262_MONTHLY_DATES
-
-# fmt: on
-
-
-# *******************************************************************
-#####################################################################
-#
-# UNIT TESTS
-#
-#####################################################################
-# *******************************************************************
-
-TEST_CASES = [
-    pytest.param(
-        INPUT_WEEKLY_DF, EXPECTED_PER_INTVL_WEEKLY_DF, EXPECTED_PER_DAY_WEEKLY_DF
-    ),
-    pytest.param(
-        INPUT_MONTHLY_DF, EXPECTED_PER_INTVL_MONTHLY_DF, EXPECTED_PER_DAY_MONTHLY_DF
-    ),
-    pytest.param(
-        INPUT_YEARLY_DF, EXPECTED_PER_INTVL_YEARLY_DF, EXPECTED_PER_DAY_YEARLY_DF
-    ),
-    pytest.param(
-        INPUT_MONTHLY_AFTER_2262_DF,
-        EXPECTED_PER_INTVL_MONTHLY_AFTER_2262_DF,
-        EXPECTED_PER_DAY_MONTHLY_AFTER_2262_DF,
-    ),
-]
-
-
-@pytest.mark.parametrize(
-    "input_df, expected_per_intvl_df, expected_per_day_df", TEST_CASES
-)
-def test_calculate_from_resampled_cumulative_vectors_df(
-    input_df: pd.DataFrame,
-    expected_per_intvl_df: pd.DataFrame,
-    expected_per_day_df: pd.DataFrame,
-) -> None:
-    # PER_INTVL_ due to as_rate_per_day = False
-    calculated_per_intvl_df = calculate_from_resampled_cumulative_vectors_df(
-        input_df, False
-    )
-
-    # PER_DAY_ due to as_rate_per_day = True
-    calculated_per_day_df = calculate_from_resampled_cumulative_vectors_df(
-        input_df, True
-    )
-
-    assert_frame_equal(expected_per_intvl_df, calculated_per_intvl_df)
-    assert_frame_equal(expected_per_day_df, calculated_per_day_df)
-
-
-def test_calculate_from_resampled_cumulative_vectors_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, 50.0,   250.0 ],
-            [pd.Timestamp(2020, 2, 1), 1, 100.0,  500.0 ],
-            [pd.Timestamp(2020, 3, 1), 1, 150.0,  750.0 ],
-            [pd.Timestamp(2020, 4, 1), 1, 200.0,  1000.0],
-            [pd.Timestamp(2020, 5, 1), 1, 250.0,  1250.0],
-        ],
-    )
-    # fmt: on
-
-    with pytest.raises(ValueError) as err:
-        calculate_from_resampled_cumulative_vectors_df(input_df, True)
-    assert (
-        str(err.value)
-        == '"DATE"-column in dataframe is not on datetime.datetime format!'
-    )
-
-
-def test_is_per_interval_or_per_day_vector() -> None:
-    assert is_per_interval_or_per_day_vector("PER_DAY_Vector")
-    assert is_per_interval_or_per_day_vector("PER_INTVL_Vector")
-    assert not is_per_interval_or_per_day_vector("per_day_Vector")
-    assert not is_per_interval_or_per_day_vector("per_intvl_Vector")
-    assert not is_per_interval_or_per_day_vector("vector")
-
-
-def test_get_cumulative_vector_name() -> None:
-    assert get_cumulative_vector_name("PER_DAY_FOPT") == "FOPT"
-    assert get_cumulative_vector_name("PER_INTVL_FOPT") == "FOPT"
-
-    assert get_cumulative_vector_name("PER_DAY_FOPR") == "FOPR"
-    assert get_cumulative_vector_name("PER_INTVL_FOPR") == "FOPR"
-
-    # Expect ValueError when verifying vector not starting with "PER_DAY_" or "PER_INTVL_"
-    try:
-        get_cumulative_vector_name("Test_vector")
-        pytest.fail('Expected retrieving of cumulative vector name for "Test_vector"')
-    except ValueError as err:
-        assert (
-            f"{err}"
-            == 'Expected "Test_vector" to be a vector calculated from cumulative!'
-        )
-
-
-def test_create_per_day_vector_name() -> None:
-    assert create_per_day_vector_name("Vector") == "PER_DAY_Vector"
-    assert create_per_day_vector_name("FOPT") == "PER_DAY_FOPT"
-    assert create_per_day_vector_name("FOPS") == "PER_DAY_FOPS"
-
-
-def test_create_per_interval_vector_name() -> None:
-    assert create_per_interval_vector_name("Vector") == "PER_INTVL_Vector"
-    assert create_per_interval_vector_name("FOPT") == "PER_INTVL_FOPT"
-    assert create_per_interval_vector_name("FOPS") == "PER_INTVL_FOPS"
-
-
-def test_datetime_to_intervalstr() -> None:
-    # Verify early return (ignore mypy)
-    assert datetime_to_intervalstr(None, Frequency.WEEKLY) is None  # type: ignore
-
-    test_date = datetime.datetime(2021, 11, 12, 13, 37)
-    assert datetime_to_intervalstr(test_date, Frequency.DAILY) == "2021-11-12"
-    assert datetime_to_intervalstr(test_date, Frequency.WEEKLY) == "2021-W45"
-    assert datetime_to_intervalstr(test_date, Frequency.MONTHLY) == "2021-11"
-    assert datetime_to_intervalstr(test_date, Frequency.QUARTERLY) == "2021-Q4"
-    assert datetime_to_intervalstr(test_date, Frequency.YEARLY) == "2021"
-
-    # Verify invalid frequency - i.e. isoformat!
-    assert datetime_to_intervalstr(test_date, None) == "2021-11-12T13:37:00"  # type: ignore

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

@@ -1,201 +0,0 @@
-from datetime import datetime
-from typing import List, Optional, Sequence
-
-import pandas as pd
-import pytest
-from pandas._testing import assert_frame_equal
-
-from webviz_subsurface._providers import Frequency
-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.history_vectors import (
-    create_history_vectors_df,
-)
-
-from ....mocks.ensemble_summary_provider_dummy import EnsembleSummaryProviderDummy
-
-# *******************************************************************
-#####################################################################
-#
-# CONFIGURE TESTDATA
-#
-#####################################################################
-# *******************************************************************
-
-
-class EnsembleSummaryProviderMock(EnsembleSummaryProviderDummy):
-    """Mock implementation of EnsembleSummaryProvider for testing of creating history
-    vector dataframe
-
-    Implements necessary methods for obtaining wanted test data
-    """
-
-    def __init__(self, input_df: pd.DataFrame) -> None:
-        super().__init__()
-
-        # Configure dataframe
-        self._df = input_df
-        self._vector_names: List[str] = list(
-            set(self._df.columns) ^ set(["DATE", "REAL"])
-        )
-        self._realizations: List[int] = list(self._df["REAL"].unique())
-
-    def supports_resampling(self) -> bool:
-        return False
-
-    def vector_names(self) -> List[str]:
-        return self._vector_names
-
-    def realizations(self) -> List[int]:
-        return self._realizations
-
-    def get_vectors_df(
-        self,
-        vector_names: Sequence[str],
-        __resampling_frequency: Optional[Frequency],
-        realizations: Optional[Sequence[int]] = None,
-    ) -> pd.DataFrame:
-        if realizations:
-            if not set(realizations).issubset(set(self._realizations)):
-                raise ValueError(
-                    "Requested realizations are not subset of provider realizations!"
-                )
-            return (
-                self._df[["DATE", "REAL"] + list(vector_names)]
-                .loc[self._df["REAL"] == realizations[0]]
-                .reset_index(drop=True)
-            )
-        raise ValueError("Expected valid realizations argument for mock!")
-
-
-# fmt: off
-# Vector WA and WB, with corresponding historical vectors WAH and WBH, respectively
-# NOTE:
-#  - Method utilize guess of historical vector
-#       - Assume vectors starting with "F", "G" or "W". Thereby "WA" and "WB" in test
-#       - String split on ":" and append "H" on first part, i.e. WA -> WAH,
-#         WA:OP_1 -> WAH:OP_1
-
-INPUT_DF = pd.DataFrame(
-    columns = ["DATE", "REAL",  "WA", "WAH", "WB", "WBH"],
-    data = [
-        [datetime(2000,1,1),  0,  11.0,  13.0,  15.0,   17.0],
-        [datetime(2000,2,1),  0,  21.0,  23.0,  25.0,   27.0],
-        [datetime(2000,3,1),  0,  31.0,  33.0,  35.0,   37.0],
-        [datetime(2000,4,1),  0,  41.0,  43.0,  45.0,   47.0],
-        [datetime(2000,5,1),  0,  51.0,  53.0,  55.0,   57.0],
-        [datetime(2000,1,1),  3,  110.0, 115.0, 135.0,  139.0],
-        [datetime(2000,2,1),  3,  310.0, 215.0, 235.0,  239.0],
-        [datetime(2000,3,1),  3,  410.0, 315.0, 335.0,  339.0],
-        [datetime(2000,4,1),  3,  510.0, 415.0, 435.0,  439.0],
-        [datetime(2000,5,1),  3,  610.0, 515.0, 535.0,  539.0],
-    ]
-)
-# Data of real = 0 for "WAH"
-EXPECTED_WA_HISTORY_DF = pd.DataFrame(
-    columns = ["DATE", "REAL", "WA"],
-    data = [
-        [datetime(2000,1,1),  0,  13.0],
-        [datetime(2000,2,1),  0,  23.0],
-        [datetime(2000,3,1),  0,  33.0],
-        [datetime(2000,4,1),  0,  43.0],
-        [datetime(2000,5,1),  0,  53.0],
-    ]
-)
-# Data of real = 0 for "WBH" and "WAH"
-EXPECTED_WB_WA_HISTORY_DF = pd.DataFrame(
-    columns = ["DATE", "REAL", "WB", "WA"],
-    data = [
-        [datetime(2000,1,1),  0,  17.0,  13.0],
-        [datetime(2000,2,1),  0,  27.0,  23.0],
-        [datetime(2000,3,1),  0,  37.0,  33.0],
-        [datetime(2000,4,1),  0,  47.0,  43.0],
-        [datetime(2000,5,1),  0,  57.0,  53.0],
-    ]
-)
-make_date_column_datetime_object(INPUT_DF)
-make_date_column_datetime_object(EXPECTED_WA_HISTORY_DF)
-make_date_column_datetime_object(EXPECTED_WB_WA_HISTORY_DF)
-
-
-# Dates AFTER year 2262!
-# NOTE: datetime.datetime after year 2262 is not converted to pd.Timestamp, thus
-# no need to make date column datetime object
-INPUT_YEAR_2265_DF = pd.DataFrame(
-    columns = ["DATE", "REAL",  "WA", "WAH", "WB", "WBH"],
-    data = [
-        [datetime(2265,1,1),  4,  11.0,  13.0,  15.0,   17.0],
-        [datetime(2265,2,1),  4,  21.0,  23.0,  25.0,   27.0],
-        [datetime(2265,3,1),  4,  31.0,  33.0,  35.0,   37.0],
-        [datetime(2265,4,1),  4,  41.0,  43.0,  45.0,   47.0],
-        [datetime(2265,5,1),  4,  51.0,  53.0,  55.0,   57.0],
-        [datetime(2265,1,1),  2,  110.0, 115.0, 135.0,  139.0],
-        [datetime(2265,2,1),  2,  310.0, 215.0, 235.0,  239.0],
-        [datetime(2265,3,1),  2,  410.0, 315.0, 335.0,  339.0],
-        [datetime(2265,4,1),  2,  510.0, 415.0, 435.0,  439.0],
-        [datetime(2265,5,1),  2,  610.0, 515.0, 535.0,  539.0],
-    ]
-)
-# Data of real = 2 for "WAH"
-EXPECTED_YEAR_2265_WA_HISTORY_DF = pd.DataFrame(
-    columns = ["DATE", "REAL", "WA"],
-    data = [
-        [datetime(2265,1,1),  2,  115.0],
-        [datetime(2265,2,1),  2,  215.0],
-        [datetime(2265,3,1),  2,  315.0],
-        [datetime(2265,4,1),  2,  415.0],
-        [datetime(2265,5,1),  2,  515.0],
-    ]
-)
-# Data of real = 2 for "WBH" and "WAH"
-EXPECTED_YEAR_2265_WB_WA_HISTORY_DF = pd.DataFrame(
-    columns = ["DATE", "REAL", "WB", "WA"],
-    data = [
-        [datetime(2265,1,1),  2,  139.0,  115.0],
-        [datetime(2265,2,1),  2,  239.0,  215.0],
-        [datetime(2265,3,1),  2,  339.0,  315.0],
-        [datetime(2265,4,1),  2,  439.0,  415.0],
-        [datetime(2265,5,1),  2,  539.0,  515.0],
-    ]
-)
-
-# fmt: on
-
-
-# *******************************************************************
-#####################################################################
-#
-# UNIT TESTS
-#
-#####################################################################
-# *******************************************************************
-
-
-TEST_CASES = [
-    pytest.param(
-        EnsembleSummaryProviderMock(INPUT_DF),
-        EXPECTED_WA_HISTORY_DF,
-        EXPECTED_WB_WA_HISTORY_DF,
-    ),
-    pytest.param(
-        EnsembleSummaryProviderMock(INPUT_YEAR_2265_DF),
-        EXPECTED_YEAR_2265_WA_HISTORY_DF,
-        EXPECTED_YEAR_2265_WB_WA_HISTORY_DF,
-    ),
-]
-
-
-@pytest.mark.parametrize(
-    "provider, expected_wa_history_df, expected_wb_wa_history_df", TEST_CASES
-)
-def test_create_history_vectors_df(
-    provider: pd.DataFrame,
-    expected_wa_history_df: pd.DataFrame,
-    expected_wb_wa_history_df: pd.DataFrame,
-) -> None:
-    create_wa_history_df = create_history_vectors_df(provider, ["WA"], None)
-    create_wb_wa_history_df = create_history_vectors_df(provider, ["WB", "WA"], None)
-
-    assert_frame_equal(create_wa_history_df, expected_wa_history_df)
-    assert_frame_equal(create_wb_wa_history_df, expected_wb_wa_history_df)

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

@@ -1,56 +0,0 @@
-from webviz_subsurface._providers import VectorMetadata
-
-# pylint: disable=line-too-long
-from webviz_subsurface.plugins._simulation_time_series._views._subplot_view._utils.trace_line_shape import (
-    get_simulation_line_shape,
-)
-
-
-def test_get_simulation_line_shape() -> None:
-    total_vector_metadata = VectorMetadata(
-        unit="M3",
-        is_total=True,
-        is_rate=False,
-        is_historical=False,
-        keyword="Test",
-        wgname=None,
-        get_num=None,
-    )
-
-    rate_vector_metadata = VectorMetadata(
-        unit="M3/Day",
-        is_total=False,
-        is_rate=True,
-        is_historical=False,
-        keyword="Test rate",
-        wgname=None,
-        get_num=None,
-    )
-
-    fallthrough_vector_metadata = VectorMetadata(
-        unit="M3/M3",
-        is_total=False,
-        is_rate=False,
-        is_historical=False,
-        keyword="Test fallthrough",
-        wgname=None,
-        get_num=None,
-    )
-
-    assert get_simulation_line_shape("Fallback", "PER_INTVL_vector", None) == "hv"
-    assert get_simulation_line_shape("Fallback", "PER_DAY_vector", None) == "hv"
-    assert get_simulation_line_shape("Fallback", "test_vector", None) == "Fallback"
-    assert (
-        get_simulation_line_shape("Fallback", "test_vector", total_vector_metadata)
-        == "linear"
-    )
-    assert (
-        get_simulation_line_shape("Fallback", "test_vector", rate_vector_metadata)
-        == "vh"
-    )
-    assert (
-        get_simulation_line_shape(
-            "Fallback", "test_vector", fallthrough_vector_metadata
-        )
-        == "Fallback"
-    )