def test_rftrecords2df():
eclfiles = EclFiles(DATAFILE)
rftrecs = rft._rftrecords2df(eclfiles)
assert len(rftrecs[rftrecs["recordname"] == "TIME"]) == len(
rftrecs["timeindex"].unique())
assert set(rftrecs["recordtype"].unique()) == set(["REAL", "INTE", "CHAR"])
assert rftrecs["timeindex"].dtype == np.int
def test_rftrecords2df():
"""Test that RFT records can be converted to dataframes"""
eclfiles = EclFiles(DATAFILE)
rftrecs = rft._rftrecords2df(eclfiles)
assert len(rftrecs[rftrecs["recordname"] == "TIME"]) == len(
rftrecs["timeindex"].unique()
)
assert set(rftrecs["recordtype"].unique()) == set(["REAL", "INTE", "CHAR"])
assert rftrecs["timeindex"].dtype == np.int
def test_rftrecords2df():
"""Test that we can construct a dataframe for navigating in RFT
records"""
rftrecs = rft._rftrecords2df(EclFiles(DATAFILE).get_rftfile())
assert len(rftrecs[rftrecs["recordname"] == "TIME"]) == len(
rftrecs["timeindex"].unique())
assert set(rftrecs["recordtype"].unique()) == set(["REAL", "INTE", "CHAR"])
assert rftrecs["timeindex"].dtype == int
assert rftrecs["recordidx"].dtype == int
# Test that we have a consecutive index in "recordidx"
assert (rftrecs["recordidx"] == rftrecs.index).all()