from kerrgeodesic_gw import surface_density_gaussian
param = param = [6.5, 0., 0.3]
s_plot = lambda r, phi: surface_density_gaussian(r, phi, param)
r, phi, z = var('r phi z')
g = plot3d(s_plot, (r, 6, 8), (phi, -0.4, 0.4), \
transformation=(r*cos(phi), r*sin(phi), z))
sphinx_plot(g)