def test_from_dict():
"""
Test gen from dictionary.
"""
striplog = Striplog.from_dict(top_dict)
assert len(striplog) == 7
assert striplog.thinnest().summary() == '0.00 m of Cody'
def add_striplog(self, data: pd.DataFrame):
unique_borehole = np.unique(data.index)
self.add_wells(unique_borehole)
for b in unique_borehole:
w = self.p.get_well(b)
data_dict = data.loc[b].to_dict('list')
s = Striplog.from_dict(data_dict, points=True)
start, stop, step_size = self._calculate_basis_parameters(w, w.location.md.shape[0])
s_log, basis, table = s.to_log(step_size, start, stop, return_meta=True)
w.data['lith'] = s
w.data['lith_log'] = Curve(s_log, basis)
return self.p
def test_striplog_2():
data = pd.read_excel(data_path.joinpath('borehole_lith.xlsx'))
well_name_column = 'SITE_ID'
well_names = data[well_name_column].unique()
foo = data.groupby(well_name_column).get_group(well_names[0])
data_dict = foo.to_dict('list')
s = Striplog.from_dict(data_dict,
remap={
'DEPTH_FROM': 'top',
'DEPTH_TO': 'base',
'LITHOLOGY': 'component lith',
'SITE_ID': 'description'
})
s.plot()
plt.show()
print(s)
