def test_angle():
a = afnumpy.random.random((2,3))+afnumpy.random.random((2,3))*1.0j
b = numpy.array(a)
fassert(afnumpy.angle(a), numpy.angle(b))
def test_angle():
a = afnumpy.random.random((2, 3)) + afnumpy.random.random((2, 3)) * 1.0j
b = numpy.array(a)
fassert(afnumpy.angle(a), numpy.angle(b))
ims[0].set_clim([np.abs(image).min(), np.abs(image).max()])
ims[1].set_data(np.abs(image)*support)
ims[1].set_clim([np.abs(image).min(), np.abs(image).max()])
ims[2].set_data(phase[::2,::2])
ims[2].set_clim([-np.pi, np.pi])
ims[3].set_data(intensities)
ims[3].set_clim([intensities.min(), intensities.max()])
line[0].set_xdata(range(iteration+1))
line[0].set_ydata(error)
axes[4].set_xlim([0,iteration+1])
axes[4].set_ylim([0.001, 0.1])
text[0].set_text('Iteration = %d' %iteration)
plt.draw()
print np.angle(data_fourier).min(), np.angle(data_fourier).max()
fourier = fft.fftn(image)
update_plot(0, image[200:-200,200:-200], np.angle(fourier), get_error(fourier, intensities), support[200:-200,200:-200], intensities)
print "Sleep for 20 seconds"
time.sleep(int(sys.argv[2]))
print "Starting now"
# Time the reconstruction
t0 = time.time()
# Store error
error = []
# Start the reconstruction (using ER)
