def test_sphere():
pb = sphere(256, 60, 1.0, 20.0)
pyplot.subplot(1, 2, 1)
pyplot.imshow(pb, cmap=pyplot.cm.gray)
pyplot.subplot(1, 2, 2)
pyplot.imshow(util.wrap_phase(pb), cmap=pyplot.cm.gray, vmin=-np.pi, vmax=np.pi)
pyplot.show()
def test_plane_box_small():
pb = plane_box(32, 10, 1.0, 3.0)
pyplot.subplot(1, 2, 1)
pyplot.imshow(pb, cmap=pyplot.cm.gray)
pyplot.subplot(1, 2, 2)
pyplot.imshow(util.wrap_phase(pb), cmap=pyplot.cm.gray, vmin=-np.pi, vmax=np.pi)
pyplot.show()