def wavefunction(n, x):
 (n, x) = (S(n), S(x))
 return (exp(((n * I) * x)) / sqrt((2 * pi)))

def wavefunction ( n , x ) : ( n , x ) = ( S ( n ) , S ( x ) ) return ( exp ( ( ( n * I ) * x ) ) / sqrt ( ( 2 * pi ) ) )
returns the wavefunction for particle on ring .