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()