1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38 package ffx.numerics.fft;
39
40 import static org.junit.Assert.assertEquals;
41
42 import java.util.Arrays;
43 import java.util.Collection;
44 import java.util.Random;
45
46 import ffx.utilities.FFXTest;
47 import org.junit.Before;
48 import org.junit.Test;
49 import org.junit.runner.RunWith;
50 import org.junit.runners.Parameterized;
51 import org.junit.runners.Parameterized.Parameters;
52
53
54 @RunWith(Parameterized.class)
55 public class Complex3DTest extends FFXTest {
56
57 private final String info;
58 private final int nx;
59 private final int ny;
60 private final int nz;
61 private final int tot;
62 private final double[] data;
63 private final double[] expected;
64 private final double[] recip;
65 private final double tolerance = 1.0e-14;
66
67 public Complex3DTest(String info, int nx, int ny, int nz) {
68 this.info = info;
69 this.nx = nx;
70 this.ny = ny;
71 this.nz = nz;
72 tot = nx * ny * nz;
73 data = new double[tot * 2];
74 expected = new double[tot];
75 recip = new double[tot];
76 }
77
78 @Parameters
79 public static Collection<Object[]> data() {
80 return Arrays.asList(
81 new Object[][] {
82 {"Test nx=32, ny=32, nz=32}", 32, 32, 32}, {"Test nx=32, ny=45, nz=21}", 32, 45, 21}
83 });
84 }
85
86 @Before
87 public void setUp() {
88 Random random = new Random();
89 for (int i = 0; i < tot; i++) {
90 int index = i * 2;
91 double r = random.nextDouble();
92 data[index] = r;
93 expected[i] = r;
94 recip[i] = 1.0e0;
95 }
96 }
97
98
99 @Test
100 public void testConvolution() {
101 Complex3D complex3D = new Complex3D(nx, ny, nz);
102 complex3D.setRecip(recip);
103 complex3D.convolution(data);
104 for (int i = 0; i < tot; i++) {
105 int index = i * 2;
106 double actual = data[index] / tot;
107 double orig = expected[i];
108 assertEquals(info, orig, actual, tolerance);
109 }
110 }
111
112
113 @Test
114 public void testFft() {
115 Complex3D complex3D = new Complex3D(nx, ny, nz);
116 complex3D.fft(data);
117 complex3D.ifft(data);
118 for (int i = 0; i < tot; i++) {
119 int index = i * 2;
120 double actual = data[index] / tot;
121 double orig = expected[i];
122 assertEquals(info, orig, actual, tolerance);
123 }
124 }
125 }