protected void verifyElements(IComplexNDArray d, IComplexNDArray d2) {
for (int i = 0; i < d.rows(); i++) {
for (int j = 0; j < d.columns(); j++) {
IComplexNumber test1 = d.getComplex(i, j);
IComplexNumber test2 = d2.getComplex(i, j);
assertEquals(
test1.realComponent().doubleValue(), test2.realComponent().doubleValue(), 1e-6);
assertEquals(
test1.imaginaryComponent().doubleValue(),
test2.imaginaryComponent().doubleValue(),
1e-6);
}
}
}
@Test
public void testTranspose() {
IComplexNDArray ndArray =
Nd4j.createComplex(
new double[] {
1.0,
0.0,
2.0,
0.0,
3.0,
0.0,
4.0,
0.0,
5.0,
0.0,
6.0,
0.0,
6.999999999999999,
0.0,
8.0,
0.0,
9.0,
0.0,
10.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0,
0.0
},
new int[] {16, 1});
IComplexNDArray transposed2 = ndArray.transpose();
assertEquals(getFailureMessage(), 16, transposed2.columns());
}