@@ -116,9 +116,13 @@ TEST_F(PWTEST, pw_basis_k_C2C_double)
116116 h_rhog[ig] = 1.0 / (pwtest.getgk2 (ik, ig) + 1 );
117117 ModuleBase::Vector3<double > f = pwtest.getgdirect (ik, ig);
118118 if (f.y > 0 )
119+ {
119120 h_rhog[ig] += ModuleBase::IMAG_UNIT / (std::abs (f.x + 1 ) + 1 );
121+ }
120122 else if (f.y < 0 )
123+ {
121124 h_rhog[ig] -= ModuleBase::IMAG_UNIT / (std::abs (-f.x + 1 ) + 1 );
125+ }
122126 }
123127 complex <double >* d_rhog = nullptr ;
124128 complex <double >* d_rhor = nullptr ;
@@ -252,8 +256,6 @@ TEST_F(PWTEST, pw_basis_k_C2C_float)
252256 const int iy = ixy % ny;
253257 for (int iz = 0 ; iz < nz; ++iz)
254258 {
255- int ix = ixy / ny;
256- int iy = ixy % ny;
257259 ModuleBase::Vector3<float > real_r (ix, iy, iz);
258260 float phase_im = -delta_g * real_r;
259261 complex <float > phase (0 , ModuleBase::TWO_PI * phase_im);
@@ -269,9 +271,13 @@ TEST_F(PWTEST, pw_basis_k_C2C_float)
269271 h_rhog[ig] = float (1.0 / (pwtest.getgk2 (ik, ig) + 1 ));
270272 ModuleBase::Vector3<double > f = pwtest.getgdirect (ik, ig);
271273 if (f.y > 0 )
274+ {
272275 h_rhog[ig] += std::complex <float >(0 ,1.0 ) / (std::abs (float (f.x ) + 1 ) + 1 );
276+ }
273277 else if (f.y < 0 )
278+ {
274279 h_rhog[ig] -= std::complex <float >(0 ,1.0 ) / (std::abs (float (-f.x ) + 1 ) + 1 );
280+ }
275281 }
276282 complex <float >* d_rhog = nullptr ;
277283 complex <float >* d_rhor = nullptr ;
0 commit comments