def test_fence_from_vertical_polygon():
"""Test fence from polygon which only has vertical samples, e.g. a vertical well"""
pol = Polygons()
mypoly = {
pol.xname: [0.0, 0.0, 0.0],
pol.yname: [100.0, 100.0, 100.0],
pol.zname: [0, 50, 60],
pol.pname: [1, 1, 1],
}
pol.dataframe = pd.DataFrame(mypoly)
fence = pol.get_fence(distance=10,
nextend=1,
name="SOMENAME",
asnumpy=False,
atleast=3)
logger.info(fence.dataframe.to_string())
assert len(fence.dataframe) == 161
assert fence.dataframe.H_DELTALEN.mean() == 0.125
assert fence.dataframe.H_DELTALEN.std() <= 0.001
assert fence.dataframe.H_CUMLEN.max() == 10
assert fence.dataframe.H_CUMLEN.min() == -10.0
def test_fence_from_almost_vertical_polygon():
"""Test fence from polygon which only has close to vertical samples"""
pol = Polygons()
mypoly = {
pol.xname: [0.1, 0.2, 0.3],
pol.yname: [100.1, 100.2, 100.3],
pol.zname: [0, 50, 60],
pol.pname: [1, 1, 1],
}
pol.dataframe = pd.DataFrame(mypoly)
fence = pol.get_fence(distance=10,
nextend=1,
name="SOMENAME",
asnumpy=False,
atleast=3)
logger.info(fence.dataframe.to_string())
assert len(fence.dataframe) == 145
assert fence.dataframe.H_DELTALEN.mean() == pytest.approx(0.1414, abs=0.01)
assert fence.dataframe.H_DELTALEN.std() <= 0.001
assert fence.dataframe.H_CUMLEN.max() == pytest.approx(10.0, abs=0.5)
assert fence.dataframe.H_CUMLEN.min() == pytest.approx(-10.0, abs=0.5)
def test_fence_from_polygon():
"""Test polygons get_fence method"""
pol = Polygons(POLSET2)
df = pol.dataframe[0:3]
df.at[0, "X_UTME"] = 0.0
df.at[1, "X_UTME"] = 100.0
df.at[2, "X_UTME"] = 100.0
df.at[0, "Y_UTMN"] = 20.0
df.at[1, "Y_UTMN"] = 20.0
df.at[2, "Y_UTMN"] = 100.0
df.at[0, "Z_TVDSS"] = 0.0
df.at[1, "Z_TVDSS"] = 1000.0
df.at[2, "Z_TVDSS"] = 2000.0
pol.dataframe = df
fence = pol.get_fence(
distance=100, nextend=4, name="SOMENAME", asnumpy=False, atleast=10
)
logger.info(fence.dataframe)
tsetup.assert_almostequal(fence.dataframe.at[13, "X_UTME"], -202.3, 0.01)
def test_fence_from_polygon():
"""Test polygons get_fence method"""
pol = Polygons(POLSET2)
df = pol.dataframe[0:3]
df.at[0, "X_UTME"] = 0.0
df.at[1, "X_UTME"] = 100.0
df.at[2, "X_UTME"] = 100.0
df.at[0, "Y_UTMN"] = 20.0
df.at[1, "Y_UTMN"] = 20.0
df.at[2, "Y_UTMN"] = 100.0
df.at[0, "Z_TVDSS"] = 0.0
df.at[1, "Z_TVDSS"] = 1000.0
df.at[2, "Z_TVDSS"] = 2000.0
pol.dataframe = df
fence = pol.get_fence(distance=100,
nextend=4,
name="SOMENAME",
asnumpy=False,
atleast=10)
logger.info(fence.dataframe)
assert fence.dataframe.H_DELTALEN.mean() == pytest.approx(19.98, abs=0.01)
assert fence.dataframe.H_DELTALEN.std() <= 0.05
def test_raise_non_existing_name(name_attribute):
pol = Polygons()
data = pd.DataFrame({"ANYTHING": [1, 2]})
pol.dataframe = data
with pytest.raises(ValueError) as execinfo:
setattr(pol, name_attribute, "NON_EXISTING")
assert "does not exist" in str(execinfo.value)
def test_raise_incorrect_name_type(test_name, name_attribute):
pol = Polygons()
data = pd.DataFrame({"ANYTHING": [1, 2]})
pol.dataframe = data
with pytest.raises(ValueError) as execinfo:
setattr(pol, name_attribute, test_name)
assert "Wrong type of input" in str(execinfo.value)
def test_fence_from_more_slanted_polygon():
"""Test fence from poly which is slanted; and total HLEN is > than distance"""
pol = Polygons()
mypoly = {
pol.xname: [0.0, 7.0, 15.0],
pol.yname: [100.0, 110.0, 120.0],
pol.zname: [0, 50, 60],
pol.pname: [1, 1, 1],
}
pol.dataframe = pd.DataFrame(mypoly)
fence = pol.get_fence(distance=10,
nextend=1,
name="SOMENAME",
asnumpy=False,
atleast=3)
assert len(fence.dataframe) == 5
assert fence.dataframe.H_DELTALEN.mean() == pytest.approx(12.49, abs=0.02)
assert fence.dataframe.H_DELTALEN.std() <= 0.001
def test_fence_from_slanted_polygon():
"""Test fence from polygon which is slanted; but total HLEN is less than distance"""
pol = Polygons()
mypoly = {
pol.xname: [0.0, 3.0, 6.0],
pol.yname: [100.0, 102.0, 104.0],
pol.zname: [0, 50, 60],
pol.pname: [1, 1, 1],
}
pol.dataframe = pd.DataFrame(mypoly)
fence = pol.get_fence(distance=10,
nextend=1,
name="SOMENAME",
asnumpy=False,
atleast=3)
assert len(fence.dataframe) == 9
assert fence.dataframe.H_DELTALEN.mean() == pytest.approx(3.6, abs=0.02)
assert fence.dataframe.H_DELTALEN.std() <= 0.001
def test_check_column_names():
pol = Polygons()
data = pd.DataFrame({"T_DELTALEN": [1, 2]})
pol.dataframe = data
assert pol.dtname == "T_DELTALEN"
assert pol.dhname is None