pot = PowerLaw(0.5, 3 * 0.2, 3)
exp = Exponential(0.5, 0.2)
fig, (axpot, axf, axcurv) = plt.subplots(3,1)
r = np.linspace(-0.001, 2)
v, dv, ddv = pot.evaluate(r, True, True, True)
axpot.plot(r, v, label="PowerLaw")
axf.plot(r, dv, label="PowerLaw")
axcurv.plot(r, ddv, label="PowerLaw")
v, dv, ddv = exp.evaluate(r, True, True, True)
axpot.plot(r, v, label="exponential")
axf.plot(r, dv, label="exponential")
axcurv.plot(r, ddv, label="exponential")
axpot.set_ylabel("Potential")
axf.set_ylabel("interaction stress")
axcurv.set_ylabel("curvature")
for a in (axpot, axf, axcurv):
a.grid()
axcurv.set_xlabel("gap")
axpot.legend()
fig.savefig("PowerLawPotential.png")