nanshe/wavelet_transform.py: Convert the input image to float32 and only use float32 for output.
nanshe/wavelet_transform.py: Added a doctest for non-float32 data.