def test_fast():
"""Test the fast option"""
# First without fast-mode:
gasoil = GasOil(h=0.1)
gasoil.add_corey_gas()
gasoil.add_corey_oil()
# This crosspoint computation is only present for fast=False:
assert "-- krg = krog @ sg=0.5" in gasoil.SGOF()
# Provoke non-strict-monotone krow:
gasoil.table.loc[0:2, "krog"] = [1.00, 0.81, 0.81]
# (this is valid in non-imbibition, but pyscal will correct it for all
# curves)
assert "0.1000000 0.0100000 0.8100000 0.0000000" in gasoil.SGOF()
assert "0.2000000 0.0400000 0.8099999 0.0000000" in gasoil.SGOF()
# monotonocity correction: ^^^^^^
# Now redo with fast option:
gasoil = GasOil(h=0.1, fast=True)
gasoil.add_corey_gas()
gasoil.add_corey_oil()
# This crosspoint computation is only present for fast=False:
assert "-- krg = krog" not in gasoil.SGOF()
# Provoke non-strict-monotone krow, in fast-mode
# this slips through:
gasoil.table.loc[0:2, "krog"] = [1.00, 0.81, 0.81]
assert "0.1000000 0.0100000 0.8100000 0.0000000" in gasoil.SGOF()
assert "0.2000000 0.0400000 0.8100000 0.0000000" in gasoil.SGOF()
# not corrected: ^^^^^^
gasoil.table.loc[0:2, "krg"] = [0.00, 0.01, 0.01]
assert "0.1000000 0.0100000" in gasoil.SGFN()
assert "0.2000000 0.0100000" in gasoil.SGFN()
def test_sgof_generation() -> None:
"""
Testing if the FlowNet code and pyscal generate
the same SGOF table - test tolerance set to 4 decimals
"""
parameter_dict = {}
parameter_dict["swirr"] = 0.01
parameter_dict["swl"] = 0.05
parameter_dict["sgcr"] = 0.055
parameter_dict["sorg"] = 0.15
parameter_dict["krgend"] = 0.95
parameter_dict["kroend"] = 0.95
parameter_dict["ng"] = 2.25
parameter_dict["nog"] = 2.25
gasoil = GasOil(
swirr=parameter_dict["swirr"],
swl=parameter_dict["swl"],
sgcr=parameter_dict["sgcr"],
sorg=parameter_dict["sorg"],
h=H_CONSTANT,
)
gasoil.add_corey_oil(nog=parameter_dict["nog"], kroend=parameter_dict["kroend"])
gasoil.add_corey_gas(ng=parameter_dict["ng"], krgend=parameter_dict["krgend"])
pyscal_sgof_string = gasoil.SGOF(header=False, dataincommentrow=False).splitlines()[
3:-1
]
numpy_sgof_string = sgof_from_parameters(parameter_dict).splitlines()
for i, line in enumerate(pyscal_sgof_string):
assert [round(float(elem), 4) for elem in line.split()] == [
round(float(elem), 4) for elem in numpy_sgof_string[i].split()
]
def test_gasoil_tag(tag):
"""Test tagging of GasOil objects,
that we are not able to produce something that
can crash Eclipse"""
gasoil = GasOil(h=0.5, tag=tag)
gasoil.add_corey_gas()
gasoil.add_corey_oil()
sat_table_str_ok(gasoil.SGOF())
sat_table_str_ok(gasoil.SGFN())
def test_roundoff():
"""Test robustness to monotonicity issues arising from
representation errors
https://docs.python.org/3/tutorial/floatingpoint.html#representation-error
The dataframe injected in this function has occured in the wild, and
caused fatal errors in Eclipse100. The error lies in
pd.dataframe.to_csv(float_format=".7f") which does truncation of floating points
instead of rounding (intentional). Since we have a strict dependency on
monotonicity properties for Eclipse to work, the data must be rounded
before being sent to to_csv(). This is being done in the .SGOF() and SWOF() as
it is a representation issue, not a numerical issues in the objects themselves.
"""
gasoil = GasOil()
# Inject a custom dataframe that has occured in the wild,
# and given monotonicity issues in GasOil.SGOF().
gasoil.table = pd.DataFrame(
columns=["sg", "krg", "krog", "pc"],
data=[
[0.02, 0, 0.19524045000000001, 0],
[0.040000000000000001, 0, 0.19524044999999998, 0],
[0.059999999999999998, 0, 0.19524045000000004, 0],
[0.080000000000000002, 0, 0.19524045000000001, 0],
[0.10000000000000001, 0, 0.19524045000000001, 0],
[0.16, 0, 0.19524045000000001, 0],
[0.17999999999999999, 0, 0.19524045000000001, 0],
[0.19999999999999998, 0, 0.19524044999999998, 0],
[0.22, 0, 0.19524045000000001, 0],
[1, 1, 0, 0],
],
)
gasoil.table["sgn"] = gasoil.table["sg"]
gasoil.table["son"] = 1 - gasoil.table["sg"]
# If this value (as string) occurs, then we are victim of floating point truncation
# in float_format=".7f":
assert "0.1952404" not in gasoil.SGOF()
assert "0.1952405" in gasoil.SGOF()
check_table(gasoil.table) # This function allows this monotonicity hiccup.
def test_selfcheck(columnname, errorvalues):
"""Test the selfcheck feature of a GasOil object"""
gasoil = GasOil(h=1)
gasoil.add_corey_gas()
gasoil.add_corey_oil()
assert gasoil.selfcheck()
# Punch the internal table directly to trigger error:
gasoil.table[columnname] = errorvalues
assert not gasoil.selfcheck()
assert gasoil.SGOF() == ""
if not columnname == "KROG":
assert gasoil.SGFN() == ""
def test_gasoil_corey1(ng, nog):
"""Test the Corey formulation for gasoil"""
gasoil = GasOil()
try:
gasoil.add_corey_oil(nog=nog)
gasoil.add_corey_gas(ng=ng)
except AssertionError:
# This happens for "invalid" input
return
assert "krog" in gasoil.table
assert "krg" in gasoil.table
assert isinstance(gasoil.krgcomment, str)
check_table(gasoil.table)
sgofstr = gasoil.SGOF()
assert len(sgofstr) > 100
assert sat_table_str_ok(sgofstr)
gasoil.resetsorg()
check_table(gasoil.table)
sgofstr = gasoil.SGOF()
assert len(sgofstr) > 100
assert sat_table_str_ok(sgofstr)
def test_gasoil_let1(l, e, t, krgend, krgmax):
go = GasOil()
try:
go.add_LET_oil(l, e, t, krgend)
go.add_LET_gas(l, e, t, krgend, krgmax)
except AssertionError:
# This happens for negative values f.ex.
return
assert "krog" in go.table
assert "krg" in go.table
assert isinstance(go.krgcomment, str)
check_table(go.table)
sgofstr = go.SGOF()
assert len(sgofstr) > 100
def testgascurves():
"""test of gas-oil curves"""
sgof = GasOil(tag="Testcurve", h=0.02, swirr=0.18, swl=0.31, sorg=0.09, sgcr=0.04)
sgof.add_corey_gas(ng=1.5, krgend=0.7)
sgof.add_corey_oil(nog=2, kroend=0.4)
sgof.add_LET_gas(l=2, e=1, t=1.4, krgend=0.9)
sgof.add_LET_oil(l=2, e=3, t=1.4, kroend=0.7)
print(sgof.table)
_, mpl_ax = plt.subplots()
sgof.plotkrgkrog(mpl_ax)
# mpl_ax.set_yscale('log')
print(sgof.SGOF())
plt.show()
def test_gasoil_corey1(ng, nog):
go = GasOil()
try:
go.add_corey_oil(nog=nog)
go.add_corey_gas(ng=ng)
except AssertionError:
# This happens for "invalid" input
return
assert "krog" in go.table
assert "krg" in go.table
assert isinstance(go.krgcomment, str)
check_table(go.table)
sgofstr = go.SGOF()
assert len(sgofstr) > 100
def test_gasoil_let1(l, e, t, krgend, krgmax):
"""Test the LET formulation, take 1"""
gasoil = GasOil()
try:
gasoil.add_LET_oil(l, e, t, krgend)
gasoil.add_LET_gas(l, e, t, krgend, krgmax)
except AssertionError:
# This happens for negative values f.ex.
return
assert "krog" in gasoil.table
assert "krg" in gasoil.table
assert isinstance(gasoil.krgcomment, str)
check_table(gasoil.table)
sgofstr = gasoil.SGOF()
assert len(sgofstr) > 100
assert sat_table_str_ok(sgofstr)
def main():
"""Entry point for interactive tests, will run
some tests where the intention is that the user will
look at what is displayed, and potentially see if
something looks really bad"""
print("-- **********************************")
print("-- Manual check of output")
swof = WaterOil(tag="Good sand, SATNUM 1", h=0.1, swl=0.1)
swof.add_corey_water()
swof.add_LET_water()
swof.add_corey_oil()
swof.add_simple_J()
print(swof.SWOF())
sgof = GasOil(tag="Good sand, SATNUM 1", h=0.1)
sgof.add_corey_gas()
sgof.add_corey_oil()
print(sgof.SGOF())
print("")
print("-- ******************************************")
print("-- Manual visual check of interpolation in LET-space")
print("-- Check:")
print("-- * green curves are between red and blue blue line")
print("-- (close plot window to continue)")
for _ in range(0, 5):
test_interpolate_wo()
test_interpolate_go()
test_interpolate_gw()
print("")
print("-- ******************************************")
print("-- Manual visual check of interpolation in LET-space")
print("-- Check:")
print("-- * Red curves are between dotted and solid blue line")
print("-- * Green curves are between solid blue and dashed")
print("-- (close plot window to continue)")
interpolateplottest()
print("")
print("-- ***********************************************")
print("-- Span of LET curves when LET parameters are varied")
print("-- within the bounds of the parameters of the red curves")
print(
"-- Blue dim curves are allowed to go outside the red boundary curves")
print("-- (close plot window to continue)")
letspan()
def main():
print("-- **********************************")
print("-- Manual check of output")
swof = WaterOil(tag="Good sand, SATNUM 1", h=0.1, swl=0.1)
swof.add_corey_water()
swof.add_LET_water()
swof.add_corey_oil()
swof.add_simple_J()
print(swof.SWOF())
sgof = GasOil(tag="Good sand, SATNUM 1", h=0.1)
sgof.add_corey_gas()
sgof.add_corey_oil()
print(sgof.SGOF())
print("")
print("-- ***************************************")
print("-- Test of one Corey curve set")
print("-- Check that all the defined endpoints are correct")
print("-- (close plot window to continue)")
testplot()
print("")
print("-- ******************************************")
print("-- Manual visual check of interpolation in LET-space")
print("-- Check:")
print("-- * Red curves are between dotted and solid blue line")
print("-- * Green curves are between solid blue and dashed")
print("-- (close plot window to continue)")
interpolateplottest()
print("")
print("-- ***********************************************")
print("-- Span of LET curves when LET parameters are varied")
print("-- within the bounds of the parameters of the red curves")
print("-- Blue dim curves are allowed to go outside the red boundary curves")
print("-- (close plot window to continue)")
letspan()
def test_comments():
"""Test that the outputters include endpoints in comments"""
gasoil = GasOil(h=0.3)
gasoil.add_corey_gas()
gasoil.add_corey_oil()
sgfn = gasoil.SGFN()
assert "--" in sgfn
assert "pyscal: " in sgfn # part of version string
assert "swirr=0" in sgfn
assert "sgcr=0" in sgfn
assert "swl=0" in sgfn
assert "sorg=0" in sgfn
assert "ng=2" in sgfn
assert "krgend=1" in sgfn
assert "Corey" in sgfn
assert "krg = krog @ sg=0.5" in sgfn
assert "Zero capillary pressure" in sgfn
assert "SG" in sgfn
assert "KRG" in sgfn
assert "PC" in sgfn
sgof = gasoil.SGOF()
assert "--" in sgof
assert "pyscal: " in sgof # part of version string
assert "swirr=0" in sgof
assert "sgcr=0" in sgof
assert "swl=0" in sgof
assert "sorg=0" in sgof
assert "ng=2" in sgof
assert "nog=2" in sgof
assert "krgend=1" in sgof
assert "Corey" in sgof
assert "krg = krog @ sg=0.5" in sgof
assert "Zero capillary pressure" in sgof
assert "SG" in sgof
assert "KRG" in sgof
assert "KROG" in sgof
assert "PC" in sgof
