from stratmc.data import load_object, load_trace
from stratmc.plotting import accumulation_rate_stratigraphy

full_trace = load_trace('examples/example_docs_trace')
example_sample_path = 'examples/example_sample_df'
example_ages_path = 'examples/example_ages_df'
sample_df = load_object(example_sample_path)
ages_df = load_object(example_ages_path)

accumulation_rate_stratigraphy(full_trace, sample_df, ages_df, sections = ['1'])

plt.show()