import numpy as np
import matplotlib.pyplot as plt

from beast.physicsmodel.priormodel import PriorDistanceModel

fig, ax = plt.subplots()

# met grid with linear spacing
dists = np.linspace(8e6, 9e6)

met_prior_models = [{"name": "flat"}]

for mp_mod in met_prior_models:
    pmod = PriorDistanceModel(mp_mod)
    ax.plot(dists, pmod(dists), label=mp_mod["name"])

ax.set_ylabel("probability")
ax.set_xlabel("distance")
ax.legend(loc="best")
plt.tight_layout()
plt.show()