def test_recon():
a, diff_v = make_synthetic_data()
total_n = 10
sup_radius = 20
# inital phase and support
init_phase = generate_random_phase_field(diff_v)
sup = generate_box_support(sup_radius, diff_v.shape)
# run reconstruction
outv1, error_dict = cdi_recon(diff_v, init_phase, sup, sw_flag=False,
n_iterations=total_n, sw_step=2)
outv1 = np.abs(outv1)
outv2, error_dict = cdi_recon(diff_v, init_phase, sup, pi_modulus_flag='Real',
sw_flag=True, n_iterations=total_n, sw_step=2)
outv2 = np.abs(outv2)
# compare the area of supports
assert_array_equal(outv1.shape, outv2.shape)
def test_cdi_plotter():
a, diff_v = make_synthetic_data()
total_n = 10
sup_radius = 20
# inital phase and support
init_phase = generate_random_phase_field(diff_v)
sup = generate_box_support(sup_radius, diff_v.shape)
# assign wrong plot_function
outv, error_d = cdi_recon(diff_v, init_phase, sup, sw_flag=True,
n_iterations=total_n, sw_step=2, cb_function=10)