def test_import_should_fail(): """Import INIT and UNRST Reek but ask for wrong name or date""" g = Grid() g.from_file(GFILE1, fformat="egrid") x = GridProperties() names = ["PORO", "NOSUCHNAME"] with pytest.raises(ValueError) as e_info: logger.warning(e_info) x.from_file(IFILE1, fformat="init", names=names, grid=g) rx = GridProperties() names = ["PRESSURE"] dates = [19991201, 19991212] # last date does not exist rx.from_file( RFILE1, fformat="unrst", names=names, dates=dates, grid=g, strict=(True, False) ) with pytest.raises(ValueError) as e_info: rx.from_file( RFILE1, fformat="unrst", names=names, dates=dates, grid=g, strict=(True, True), )
def test_import_restart(): """Import Restart""" g = Grid() g.from_file(GFILE1, fformat="egrid") x = GridProperties() names = ["PRESSURE", "SWAT"] dates = [19991201, 20010101] x.from_file(RFILE1, fformat="unrst", names=names, dates=dates, grid=g) # get the object pr = x.get_prop_by_name("PRESSURE_19991201") swat = x.get_prop_by_name("SWAT_19991201") logger.info(x.names) logger.info(swat.values3d.mean()) logger.info(pr.values3d.mean()) txt = "Average PRESSURE_19991201" assert pr.values.mean() == pytest.approx(334.52327, abs=0.0001), txt txt = "Average SWAT_19991201" assert swat.values.mean() == pytest.approx(0.87, abs=0.01), txt pr = x.get_prop_by_name("PRESSURE_20010101") logger.info(pr.values3d.mean()) txt = "Average PRESSURE_20010101" assert pr.values.mean() == pytest.approx(304.897, abs=0.01), txt
def test_import_should_warn(): """Import INIT and UNRST Reek but ask for wrong name or date""" g = Grid() g.from_file(GFILE1, fformat="egrid") rx = GridProperties() names = ['PRESSURE'] dates = [19991201, 19991212] # last date does not exist rx.from_file(RFILE1, fformat='unrst', names=names, dates=dates, grid=g)
def test_import_should_pass(): """Import INIT and UNRST but ask for wrong name or date , using strict=False""" g = Grid() g.from_file(GFILE1, fformat="egrid") rx = GridProperties() names = ["PRESSURE", "DUMMY"] # dummy should exist dates = [19991201, 19991212] # last date does not exist rx.from_file( RFILE1, fformat="unrst", names=names, dates=dates, grid=g, strict=(False, False) ) assert "PRESSURE_19991201" in rx assert "PRESSURE_19991212" not in rx assert "DUMMY_19991201" not in rx
def test_get_dataframe(): """Get a Pandas dataframe from the gridproperties""" g = Grid(GFILE1, fformat="egrid") x = GridProperties() names = ["SOIL", "SWAT", "PRESSURE"] dates = [19991201] x.from_file(RFILE1, fformat="unrst", names=names, dates=dates, grid=g) df = x.dataframe(activeonly=True, ijk=True, xyz=False) print(df.head()) assert df["SWAT_19991201"].mean() == pytest.approx(0.87802, abs=0.001) assert df["PRESSURE_19991201"].mean() == pytest.approx(334.523, abs=0.005)
def test_import_init(): """Import INIT Reek""" g = Grid() g.from_file(GFILE1, fformat="egrid") x = GridProperties() names = ["PORO", "PORV"] x.from_file(IFILE1, fformat="init", names=names, grid=g) # get the object poro = x.get_prop_by_name("PORO") logger.info("PORO avg {}".format(poro.values.mean())) porv = x.get_prop_by_name("PORV") logger.info("PORV avg {}".format(porv.values.mean())) assert poro.values.mean() == pytest.approx(0.1677402, abs=0.00001)
def test_import_soil(): """SOIL need to be computed in code from SWAT and SGAS""" g = Grid() g.from_file(GFILE1, fformat="egrid") x = GridProperties() names = ["SOIL", "SWAT", "PRESSURE"] dates = [19991201] x.from_file(RFILE1, fformat="unrst", names=names, dates=dates, grid=g) logger.info(x.names) # get the object instance soil = x.get_prop_by_name("SOIL_19991201") logger.info(soil.values3d.mean()) logger.debug(x.names) txt = "Average SOIL_19850101" assert soil.values.mean() == pytest.approx(0.121977, abs=0.001), txt
def test_import_restart_gull(): """Import Restart Reek""" g = Grid() g.from_file(GFILE1, fformat="egrid") x = GridProperties() names = ["PRESSURE", "SWAT"] dates = [19991201] x.from_file(RFILE1, fformat="unrst", names=names, dates=dates, grid=g) # get the object pr = x.get_prop_by_name("PRESSURE_19991201") swat = x.get_prop_by_name("SWAT_19991201") logger.info(x.names) logger.info(swat.values3d.mean()) logger.info(pr.values3d.mean())