def test_simple_j_petro(): """Simple test of the simple J petrophysical function correlation""" wateroil = WaterOil(swl=0.01) wateroil.add_simple_J_petro(a=1, b=-2) check_table(wateroil.table) assert wateroil.pccomment assert "etrophysic" in wateroil.pccomment # Zero gravity: wateroil.add_simple_J_petro(a=1, b=-2, g=0) assert wateroil.table.pc.unique() == 0.0 # Numerical test from sample numbers calculated independently in different tool: wateroil = WaterOil(swl=0.05, h=0.025) wateroil.add_simple_J_petro( a=1.45, b=-0.285, drho=143, g=9.81, perm_ref=15, poro_ref=0.27 ) float_df_checker(wateroil.table, "sw", 0.1, "pc", 22.36746) assert "Simplified" in wateroil.pccomment assert "etrophysic" in wateroil.pccomment wateroil.add_corey_oil() wateroil.add_corey_water() swof = wateroil.SWOF() assert isinstance(swof, str) assert swof assert sat_table_str_ok(swof) assert sat_table_str_ok(wateroil.SWFN())
def test_go_fromtable_problems(): """Test loading from a table where there should be problems""" df1 = pd.DataFrame(columns=["Sg", "KRG", "KROG", "PCOG"], data=[[0.1, 0, 1, 2], [0.9, 1, 0, 0]]) # Now sgcr and swl is wrong: gasoil = GasOil(h=0.1) with pytest.raises(ValueError): # Should say sg must start at zero. gasoil.add_fromtable(df1, pccolname="PCOG") df2 = pd.DataFrame(columns=["Sg", "KRG", "KROG", "PCOG"], data=[[0.0, 0, 1, 2], [0.9, 0.8, 0, 0]]) with pytest.raises(ValueError): # should say too large swl for pcog interpolation gasoil = GasOil(h=0.1) gasoil.add_fromtable(df2, pccolname="PCOG") gasoil = GasOil(h=0.1, swl=0.1) gasoil.add_fromtable(df2, pccolname="PCOG") assert np.isclose(gasoil.table["pc"].max(), 2.0) assert np.isclose(gasoil.table["pc"].min(), 0.0) assert np.isclose(gasoil.table["sg"].max(), 0.9) gasoil = GasOil(h=0.1) with pytest.raises(ValueError): # sg must start at zero gasoil.add_fromtable(df1, krgcolname="KRG", krogcolname="KROG") # This works fine, but we get warnings on swl not seemingly correct gasoil.add_fromtable(df2, krgcolname="KRG", krogcolname="KROG") # krg will be extrapolated to sg=1 float_df_checker(gasoil.table, "sg", 1.0, "krg", 0.8) float_df_checker(gasoil.table, "sg", 1.0, "krog", 0.0) gasoil = GasOil(h=0.1, swl=0.1) gasoil.add_fromtable(df2, krgcolname="KRG", krogcolname="KROG")
def check_endpoints(gasoil, krgend, krgmax, kroend, kromax): """Discrete tests that endpoints get numerically correct""" swtol = 1 / SWINTEGERS # Check endpoints for oil curve: # krog at sgcr should be kroend if gasoil.sgcr > swtol: assert float_df_checker(gasoil.table, "son", 1.0, "krog", kroend) # krog at son=0 (1-sorg-swl or 1 - swl) should be zero: assert float_df_checker(gasoil.table, "son", 0.0, "krog", 0) if gasoil.sgcr > swtol: assert float_df_checker(gasoil.table, "sg", 0, "krog", kromax) assert float_df_checker(gasoil.table, "sg", gasoil.sgcr, "krog", kroend) else: if not np.isclose(gasoil.table["krog"].max(), kroend): print(gasoil.table.head()) assert np.isclose(gasoil.table["krog"].max(), kroend) assert float_df_checker(gasoil.table, "sgn", 0.0, "krg", 0) assert float_df_checker(gasoil.table, "sg", gasoil.sgcr, "krg", 0) # If krgendanchor == "sorg" then krgmax is irrelevant. if gasoil.sorg > swtol and gasoil.sorg > gasoil.h and gasoil.krgendanchor == "sorg": assert float_df_checker(gasoil.table, "sgn", 1.0, "krg", krgend) assert np.isclose(gasoil.table["krg"].max(), krgmax) if gasoil.krgendanchor != "sorg": assert np.isclose(gasoil.table["krg"].max(), krgend)
def test_ip_kromax(): """Test behaviour of kroend/kromax under interpolation""" wo_low = WaterOil(swl=0, swcr=0.1, sorw=0.2) wo_low.add_corey_water(nw=2, krwend=0.5, krwmax=0.7) wo_low.add_corey_oil(now=2, kroend=0.6, kromax=0.9) wo_high = WaterOil(swl=0.02, swcr=0.05, sorw=0.1) wo_high.add_corey_water(nw=2, krwend=0.5, krwmax=0.72) wo_high.add_corey_oil(now=2, kroend=0.6, kromax=0.8) # Interpolate to midpoint between the curves above wo_ip = utils.interpolate_wo(wo_low, wo_high, 0.5) # kromax at mean swl: assert float_df_checker(wo_ip.table, "sw", 0.01, "krow", 0.85) # kroend is 0.6 in low and high, but swcr has changed: assert float_df_checker(wo_ip.table, "sw", 0.075, "krow", 0.6) assert float_df_checker(wo_ip.table, "sw", 1, "krw", 0.71) assert float_df_checker(wo_ip.table, "sw", 1 - 0.15, "krw", 0.5)
def check_endpoints(wateroil, krwend, krwmax, kroend, kromax): """Check that the code produces correct endpoints for parametrizations, on discrete cases""" swtol = 1 / SWINTEGERS # Check endpoints for oil curve: # krow at swcr should be kroend: if wateroil.swcr > wateroil.swl + swtol: assert float_df_checker(wateroil.table, "son", 1.0, "krow", kroend) # krow at sorw should be zero: assert float_df_checker(wateroil.table, "son", 0.0, "krow", 0.0) if wateroil.swcr > wateroil.swl + swtol: # krow at swl should be kromax: assert float_df_checker(wateroil.table, "sw", wateroil.swl, "krow", kromax) else: # kromax not used when swcr is close to swl. assert np.isclose(wateroil.table["krow"].max(), kroend) # Check endpoints for water curve: (np.isclose is only reliable around 1) assert float_df_checker(wateroil.table, "swn", 0.0, "krw", 0.0) assert float_df_checker(wateroil.table, "sw", wateroil.swcr, "krw", 0) if wateroil.sorw > swtol: # (hard to get it right when sorw is less than h and close to zero) assert float_df_checker(wateroil.table, "sw", 1 - wateroil.sorw, "krw", krwend) assert np.isclose(wateroil.table["krw"].max(), krwmax) else: assert np.isclose(wateroil.table["krw"].max(), krwend)
def test_gasoil_normalization(swl, sgcr, sorg, h, tag): """Check that normalization (sgn and son) is correct for all possible saturation endpoints""" gasoil = GasOil( swirr=0.0, swl=swl, sgcr=sgcr, sorg=sorg, h=h, krgendanchor="sorg", tag=tag ) assert not gasoil.table.empty assert not gasoil.table.isnull().values.any() # Check that son is 1 at sgcr assert float_df_checker(gasoil.table, "sg", gasoil.sgcr, "son", 1) # Check that son is 0 at sorg with this krgendanchor assert float_df_checker(gasoil.table, "sg", 1 - gasoil.sorg - gasoil.swl, "son", 0) # Check that sgn is 0 at sgcr assert float_df_checker(gasoil.table, "sg", gasoil.sgcr, "sgn", 0) # Check that sgn is 1 at sorg assert float_df_checker(gasoil.table, "sg", 1 - gasoil.sorg - gasoil.swl, "sgn", 1) # Redo with different krgendanchor gasoil = GasOil( swirr=0.0, swl=swl, sgcr=sgcr, sorg=sorg, h=h, krgendanchor="", tag=tag ) assert float_df_checker(gasoil.table, "sg", 1 - gasoil.swl, "sgn", 1) assert float_df_checker(gasoil.table, "sg", gasoil.sgcr, "sgn", 0)
def test_wateroil_normalization(swirr, swl, swcr, sorw, h, tag): """Shoot with more realistic values and test normalized saturations""" wateroil = WaterOil(swirr=swirr, swl=swl, swcr=swcr, sorw=sorw, h=h, tag=tag) assert not wateroil.table.empty assert not wateroil.table.isnull().values.any() # Check that son is 1 at swcr: assert float_df_checker(wateroil.table, "sw", wateroil.swcr, "son", 1) # Check that son is 0 at sorw: if wateroil.sorw > h: assert float_df_checker(wateroil.table, "sw", 1 - wateroil.sorw, "son", 0) # Check that swn is 0 at swcr: assert float_df_checker(wateroil.table, "sw", wateroil.swcr, "swn", 0) # Check that swn is 1 at 1 - sorw if wateroil.sorw > 1 / SWINTEGERS: assert float_df_checker(wateroil.table, "sw", 1 - wateroil.sorw, "swn", 1) # Check that swnpc is 0 at swirr and 1 at 1: if wateroil.swirr >= wateroil.swl + h: assert float_df_checker(wateroil.table, "sw", wateroil.swirr, "swnpc", 0) else: # Let this go, when swirr is too close to swl. We # are not guaranteed to have sw=swirr present pass assert float_df_checker(wateroil.table, "sw", 1.0, "swnpc", 1)
def test_normalized_j(): """Test the normalized J-function correlation for capillary pressure""" wateroil = WaterOil(swirr=0.1, h=0.1) with pytest.raises(ValueError): wateroil.add_normalized_J(a=0.5, b=-0.2, poro=0.2, perm=10, sigma_costau=30) wateroil = WaterOil(swirr=0, swl=0.1, h=0.1) wateroil.add_normalized_J(a=0.5, b=-0.2, poro=0.2, perm=10, sigma_costau=30) check_table(wateroil.table) # Sample numerical tests taken from a prior implementation # NB: Prior implementation created Pc in atm, we create in bar bar_to_atm = 1.0 / 1.01325 wateroil.add_normalized_J(a=0.22, b=-0.5, perm=100, poro=0.2, sigma_costau=30) float_df_checker(wateroil.table, "sw", 0.1, "pc", 2.039969 * bar_to_atm) float_df_checker(wateroil.table, "sw", 0.6, "pc", 0.056666 * bar_to_atm) float_df_checker(wateroil.table, "sw", 1.0, "pc", 0.02040 * bar_to_atm)
def test_gasoil_kromax_fast(): """Test some code in fast mode""" gasoil = GasOil(h=0.1, sgcr=0.1, fast=True) gasoil.add_corey_oil(2, 0.5) # Default for kromax assert float_df_checker(gasoil.table, "sg", 0.0, "krog", 1.0) assert float_df_checker(gasoil.table, "sg", 0.1, "krog", 0.5) gasoil.add_corey_oil(2, 0.5, 0.7) assert float_df_checker(gasoil.table, "sg", 0.0, "krog", 0.7) assert float_df_checker(gasoil.table, "sg", 0.1, "krog", 0.5) gasoil = GasOil(h=0.1, sgcr=0.0) gasoil.add_corey_oil(2, 0.5) assert float_df_checker(gasoil.table, "sg", 0.0, "krog", 0.5) gasoil.add_corey_oil(2, 0.5, 1) # A warning will be given assert float_df_checker(gasoil.table, "sg", 0.0, "krog", 0.5)
def test_gasoil_kromax(): """Manual test of kromax behaviour""" gasoil = GasOil(h=0.1, sgcr=0.1) gasoil.add_corey_oil(2, 0.5) # Default for kromax assert float_df_checker(gasoil.table, "sg", 0.0, "krog", 1.0) assert float_df_checker(gasoil.table, "sg", 0.1, "krog", 0.5) gasoil.add_corey_oil(2, 0.5, 0.7) assert float_df_checker(gasoil.table, "sg", 0.0, "krog", 0.7) assert float_df_checker(gasoil.table, "sg", 0.1, "krog", 0.5) gasoil = GasOil(h=0.1, sgcr=0.0) gasoil.add_corey_oil(2, 0.5) assert float_df_checker(gasoil.table, "sg", 0.0, "krog", 0.5) gasoil.add_corey_oil(2, 0.5, 1) # A warning will be given assert float_df_checker(gasoil.table, "sg", 0.0, "krog", 0.5)
def test_ip_krogmax(): """Test behaviour of kroend/kromax under interpolation""" go_low = GasOil(swl=0, sgcr=0.1, sorg=0.2) go_low.add_corey_gas(ng=2, krgend=0.5, krgmax=0.7) go_low.add_corey_oil(nog=2, kroend=0.6, kromax=0.9) go_high = GasOil(swl=0.02, sgcr=0.05, sorg=0.1) go_high.add_corey_gas(ng=2, krgend=0.5, krgmax=0.72) go_high.add_corey_oil(nog=2, kroend=0.6, kromax=0.8) # Interpolate to midpoint between the curves above go_ip = utils.interpolate_go(go_low, go_high, 0.5) # kromax at sg zero, should be kromax: assert float_df_checker(go_ip.table, "sg", 0.0, "krog", 0.85) # kroend is 0.6 in low and high, but sgcr has changed: assert float_df_checker(go_ip.table, "sg", 0.075, "krog", 0.6) assert np.isclose(go_ip.swl, 0.01) assert np.isclose(go_ip.sorg, 0.15) # krgmax at 1 - swl: assert float_df_checker(go_ip.table, "sg", 1 - go_ip.swl, "krg", 0.71) # krgend at 1 - swl - sorg assert float_df_checker(go_ip.table, "sg", 1 - go_ip.swl - go_ip.sorg, "krg", 0.5) # If krgendanchor is None, then krgmax should be irrelevant go_low = GasOil(swl=0, sgcr=0.1, sorg=0.2, krgendanchor="") go_low.add_corey_gas(ng=5, krgend=0.5, krgmax=0.7) # krgmax will trigger warning message, intended, as the 0.7 # value here will mean nothing. go_low.add_corey_oil(nog=2, kroend=0.6, kromax=0.9) go_high = GasOil(swl=0.02, sgcr=0.05, sorg=0.1, krgendanchor="") go_high.add_corey_gas(ng=4, krgend=0.5, krgmax=0.72) # krgmax will trigger warning message, intended. go_high.add_corey_oil(nog=2, kroend=0.6, kromax=0.8) # Interpolate to midpoint between the curves above go_ip = utils.interpolate_go(go_low, go_high, 0.5, h=0.1) assert float_df_checker(go_ip.table, "sg", 0.0, "krog", 0.85) # kroend is 0.6 in low and high, but swcr has changed: assert float_df_checker(go_ip.table, "sg", 0.075, "krog", 0.6) # Activate these line to see a bug, interpolation_go # does not honor krgendanchorA: # from matplotlib import pyplot as plt # _, mpl_ax = plt.subplots() # go_low.plotkrgkrog(mpl_ax=mpl_ax, color="red") # go_high.plotkrgkrog(mpl_ax=mpl_ax, color="blue") # go_ip.plotkrgkrog(mpl_ax=mpl_ax, color="green") # plt.show() # krgmax is irrelevant, krgend is used here: assert float_df_checker(go_ip.table, "sg", 1 - 0.01, "krg", 0.5) # Do we get into trouble if krgendanchor is different in low and high? go_low = GasOil(swl=0, sgcr=0.1, sorg=0.2, krgendanchor="sorg") go_low.add_corey_gas(ng=2, krgend=0.5, krgmax=0.7) go_low.add_corey_oil(nog=2, kroend=0.6, kromax=0.9) go_high = GasOil(swl=0.02, sgcr=0.05, sorg=0.1, krgendanchor="") go_high.add_corey_gas(ng=2, krgend=0.5) go_high.add_corey_oil(nog=2, kroend=0.6, kromax=0.8) # Interpolate to midpoint between the curves above go_ip = utils.interpolate_go(go_low, go_high, 0.5) assert float_df_checker(go_ip.table, "sg", 0.0, "krog", 0.85) # kroend is 0.6 in low and high, but swcr has changed: assert float_df_checker(go_ip.table, "sg", 0.075, "krog", 0.6) # max(krg) is here avg of krgmax and krgend from the differnt tables: assert float_df_checker(go_ip.table, "sg", 1 - 0.01, "krg", 0.6) # krgend at 1 - swl - sorg, non-trivial expression, so a numerical # value is used here in the test: assert float_df_checker(go_ip.table, "sg", 1 - 0.01 - 0.15, "krg", 0.4491271) # krog-zero at 1 - swl - sorg: assert float_df_checker(go_ip.table, "sg", 1 - 0.01 - 0.15, "krog", 0)