diff --git a/hw3-heat-fft/fft_generate_test_values.py b/hw3-heat-fft/fft_generate_test_values.py new file mode 100644 index 00000000..68c9db99 --- /dev/null +++ b/hw3-heat-fft/fft_generate_test_values.py @@ -0,0 +1,26 @@ +import numpy as np + +N = 20 +vec = np.fft.fftfreq(N) * N +print(vec) +print("\n") + +mat = np.zeros((N,N)) +for i in range(0, N): + for j in range(0, N): + mat[i,j] = vec[i]**2 + vec[j]**2 +print(mat) +print("\n") +np.savetxt("build/fftfreq_test_values_even.txt", mat) + +N = 19 +vec = np.fft.fftfreq(N) * N +print(vec) +print("\n") +mat = np.zeros((N,N)) +for i in range(0, N): + for j in range(0, N): + mat[i,j] = vec[i]**2 + vec[j]**2 +print(mat) +print("\n") +np.savetxt("build/fftfreq_test_values_odd.txt", mat)