def test_pbvd():
"""Test PBVD tables"""
deckstr = """
PBVD
10 100 /
30 400 /
50 100 /"""
pbvd_df = equil.df(deckstr)
assert "KEYWORD" in pbvd_df
assert "EQUIL" not in pbvd_df["KEYWORD"].values
assert max(pbvd_df["EQLNUM"]) == 3
assert set(pbvd_df["Z"].values) == {10, 30, 50}
assert set(pbvd_df["PB"].values) == {100, 400}
inc = equil.df2ecl(pbvd_df)
df_from_inc = equil.df(inc)
pd.testing.assert_frame_equal(pbvd_df, df_from_inc)
assert equil.df(deckstr, keywords="EQUIL").empty
# Check that we can use the underlying function directly:
pbvd_df2 = equil.pbvd_fromdeck(deckstr)
pd.testing.assert_frame_equal(pbvd_df.drop("KEYWORD", axis="columns"),
pbvd_df2)
# Check that we don't need the KEYWORD column for the underlying function:
assert equil.df2ecl_pbvd(pbvd_df) == equil.df2ecl_pbvd(
pbvd_df.drop("KEYWORD", axis="columns"))
# If EQLNUM column is dropped it is not possible to guess the
# correct include file, so the code must fail:
with pytest.raises(KeyError):
equil.df2ecl_pbvd(pbvd_df.drop("EQLNUM", axis="columns"))
def test_pbvd():
"""Test PBVD tables"""
deckstr = """
PBVD
10 100 /
30 400 /
50 100 /"""
pbvd_df = equil.df(deckstr)
assert "KEYWORD" in pbvd_df
assert "EQUIL" not in pbvd_df["KEYWORD"].values
assert max(pbvd_df["EQLNUM"]) == 3
assert set(pbvd_df["Z"].values) == {10, 30, 50}
assert set(pbvd_df["PB"].values) == {100, 400}
inc = equil.df2ecl(pbvd_df)
df_from_inc = equil.df(inc)
pd.testing.assert_frame_equal(pbvd_df, df_from_inc)
assert equil.df(deckstr, keywords="EQUIL").empty
# Check that we can use the underlying function directly:
pbvd_df2 = equil.pbvd_fromdeck(deckstr)
pd.testing.assert_frame_equal(pbvd_df.drop("KEYWORD", axis="columns"),
pbvd_df2)