Update fft_test2.py