test_plot.py 文件源码

def test_plot_rv_curves():

    pars = dict(P=[30.]*u.day, K=[10.]*u.km/u.s, ecc=[0.11239],
                omega=[0.]*u.radian, phi0=[np.pi/2]*u.radian)

    samples = JokerSamples(VelocityTrend2)
    for k in pars:
        samples[k] = pars[k]

    samples['v0'] = [150] * u.km/u.s
    samples['v1'] = [0.01] * u.km/u.s/u.day

    t_grid = np.random.uniform(56000, 56500, 1024)
    t_grid.sort()

    fig, ax = plt.subplots(1, 1, figsize=(12,5))
    plot_rv_curves(samples, t_grid, ax=ax, trend_t0=56000.)