def test_AlphaModelholo_likelihood(): holo = get_example_data('image0001') s = Sphere( prior.Gaussian(.5, .1), prior.Gaussian(1.6, .1), (prior.Gaussian(5, 1), prior.Gaussian(5, 1), prior.Gaussian(5, 1))) model = AlphaModel(s, alpha=prior.Gaussian(.7, .1), noise_sd=.01) assert_pickle_roundtrip(model)
def test_emcee(): holo = get_example_data('image0001.yaml') s = Sphere( prior.Gaussian(.5, .1), prior.Gaussian(1.6, .1), (prior.Gaussian(5, 1), prior.Gaussian(5, 1), prior.Gaussian(5, 1))) model = AlphaModel(s, Mie, alpha=prior.Gaussian(.7, .1), noise_sd=.01) emcee = Emcee(model, holo) assert_pickle_roundtrip(emcee)
def test_PerfectLensModelholo_likelihood(): holo = get_example_data('image0001') sphere_center = (prior.Gaussian(5, 1), prior.Gaussian(5, 1), prior.Gaussian(5, 1)) s = Sphere(n=prior.Gaussian(1.6, .1), r=prior.Gaussian(.5, .1), center=sphere_center) model = PerfectLensModel(s, noise_sd=0.01, lens_angle=0.8) assert_pickle_roundtrip(model)
def test_ExactModelholo_likelihood(): holo = get_example_data('image0001') sphere_center = (prior.Gaussian(5, 1), prior.Gaussian(5, 1), prior.Gaussian(5, 1)) s = Sphere(n=prior.Gaussian(1.6, .1), r=prior.Gaussian(.5, .1), center=sphere_center) model = ExactModel(s, noise_sd=.01) assert_pickle_roundtrip(model)
def test_TimeSeriesAlphaModel(): n = TimeIndependent(prior.Gaussian(5, .5)) assert_pickle_roundtrip(n) st = Sphere(n=n, r=TimeIndependent(prior.BoundedGaussian(1.6, .1, 0, np.inf)), center=(prior.Gaussian(10, 1), prior.Gaussian(10, 1), prior.BoundedGaussian(1.6, .1, 0, np.inf))) assert_pickle_roundtrip(st) noise_sd = .1 mt = TimeSeriesAlphaModel(st, Mie, noise_sd, alpha=prior.Uniform(0, 1), n_frames=2) assert_pickle_roundtrip(mt)
def test_image(): holo = get_example_data('image0001') assert_pickle_roundtrip(holo)
def test_AlphaModelholo_likelihood(): holo = get_example_data('image0001') s = Sphere(prior.Gaussian(.5, .1), prior.Gaussian(1.6, .1), (prior.Gaussian(5, 1), prior.Gaussian(5, 1), prior.Gaussian(5, 1))) model = AlphaModel(s, alpha = prior.Gaussian(.7, .1), noise_sd=.01) assert_pickle_roundtrip(model)
def test_prior(): g = prior.Gaussian(1, 1) assert_pickle_roundtrip(g) assert_pickle_roundtrip(g.lnprob)
def test_pickle_mie_object(): m = Mie() assert_pickle_roundtrip(m) assert_pickle_roundtrip(m.calc_holo, True)
def test_pickle_calc_holo(): assert_pickle_roundtrip(Mie.calc_holo)
def test_prior(): g = prior.Gaussian(1, 1) assert_pickle_roundtrip(g) assert_pickle_roundtrip(g.lnprob)
def test_pickle_mie_object(): m = Mie() assert_pickle_roundtrip(m)
def test_pickle_calc_holo(): assert_pickle_roundtrip(Mie.calc_holo)
def test_pickle_mie_object(): m = Mie() assert_pickle_roundtrip(m) assert_pickle_roundtrip(m.calc_holo, True)
def test_pickle_mie_object(): m = Mie() assert_pickle_roundtrip(m)
def test_image(): holo = get_example_data('image0001.yaml') assert_pickle_roundtrip(holo)