sunwen/Aurora
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add ifftshift

Browse Source
This commit is contained in:
kradchen
2023-06-08 15:24:49 +08:00
parent eca0e77fcc
commit a0db3777d2
3 changed files with 38 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
test/Function2D_Test.cpp
View File

@@ -369,6 +369,17 @@ TEST_F(Function2D_Test, fftAndComplexAndIfft){
EXPECT_DOUBLE_EQ(-3.4641, fourDecimalRound(result3[4].imag()));
EXPECT_DOUBLE_EQ(-1.0, fourDecimalRound(result3[13].real()));
EXPECT_DOUBLE_EQ(-3.4641, fourDecimalRound(result3[16].imag()));
{
auto fm = Aurora::Matrix::fromRawData(new double[7]{1,2,3, 4, 5, 6, 7},7,1);
Aurora::fftshift(fm);
EXPECT_DOUBLE_EQ(5, fm[0]);
EXPECT_DOUBLE_EQ(1, fm[3]);
EXPECT_DOUBLE_EQ(4, fm[6]);
Aurora::ifftshift(fm);
EXPECT_DOUBLE_EQ(1, fm[0]);
EXPECT_DOUBLE_EQ(4, fm[3]);
EXPECT_DOUBLE_EQ(7, fm[6]);
}
}
TEST_F(Function2D_Test, hilbert) {