def test_plot_topography(self):
m = GMS(testmodel, verbosity=-1)
t = m.plot_topography()
self.assertIsInstance(t, tri.TriContourSet)
levels = t.levels.tolist()
levels_target = [-2.4999999999999994e-14, 0.0, 2.4999999999999994e-14]
self.assertListEqual(levels, levels_target)
dsigma = np.abs(raw_data[:, 2])*0.001 # Diff. stress / GPa
# Plot
gs_kwd = dict(width_ratios=(5, 1), wspace=0.01)
fig, axes = plt.subplots(1, 2, gridspec_kw=gs_kwd, figsize=(18, 5))
# Profile
ax = axes[0]
triangulation = tri.Triangulation(d, z)
mappable = ax.tricontourf(triangulation, dsigma, 100)
model.plot_layer_bounds(x0, y0, x1, y1, lw=2, only='unique',
unit='km', ax=ax)
plt.colorbar(mappable=mappable, ax=ax, label='Yield strength / GPa')
ax.set_xlabel('Distance / km')
ax.set_ylabel('Elevation / km')
ax.set_title('y = ' + str(y0) + 'm')
ax = axes[1]
model.plot_topography(ax=ax)
ax.plot([x0, x1], [y0, y1], color='red', linewidth=2)
ax.yaxis.tick_right()
format_km = FuncFormatter(lambda x, pos: '{0:g}'.format(x/1000))
ax.xaxis.set_major_formatter(format_km)
ax.yaxis.set_major_formatter(format_km)
savetitle = os.path.splitext(f)[0] + '.png'
fig.savefig(savetitle, dpi=100, bbox_inches='tight', facecolor='w')
print('>', savetitle, 'saved!')
print('Done!')