test(parallel): all UT of parallel_reduce.cpp

Author: hongriTianqi

source/src_parallel/parallel_reduce.cpp

@@ -162,7 +162,7 @@ void Parallel_Reduce::reduce_complex_double_all(std::complex <double> *object, c
 #ifdef __MPI
 	std::complex<double> *swap = new std::complex<double>[n];
 	for(int i=0;i<n;i++) swap[i] = object[i];
-	MPI_Allreduce(swap, object, n, mpicomplex, myOp, POOL_WORLD);
+	MPI_Allreduce(swap, object, n, mpicomplex, myOp, MPI_COMM_WORLD);
 	delete[] swap;
 #endif
 	return;
@@ -285,27 +285,3 @@ void Parallel_Reduce::gather_min_double_all(double &v)
 	delete[] value;
 #endif
 }
-
-bool Parallel_Reduce::check_if_equal(double &v)
-{
-#ifdef __MPI
-	double *all=new double[GlobalV::NPROC];
-	MPI_Allgather(&v, 1, MPI_DOUBLE, all, 1, MPI_DOUBLE, MPI_COMM_WORLD);
-	for(int i=0; i<GlobalV::NPROC; i++)
-	{
-		if( abs(all[i] - all[0]) > 1.0e-9 )
-		{
-			for(int j=0; j<GlobalV::NPROC; j++)
-			{
-				std::cout << "\n processor = " << j << " value = " << all[j];
-			}
-			delete[] all;
-			return false;
-		}
-	}
-	delete[] all;
-	return true;
-#endif
-	return true;
-}
-

source/src_parallel/test/CMakeLists.txt

@@ -16,9 +16,16 @@ AddTest(
   SOURCES parallel_kpoints_test.cpp ../../module_base/global_variable.cpp ../parallel_global.cpp ../parallel_common.cpp ../parallel_kpoints.cpp
 )
 
+AddTest(
+  TARGET ParaReduce
+  LIBS MPI::MPI_CXX
+  SOURCES parallel_reduce_test.cpp ../../module_base/global_variable.cpp ../parallel_global.cpp ../parallel_common.cpp ../parallel_kpoints.cpp ../parallel_reduce.cpp
+)
+
 install(FILES parallel_common_test.sh DESTINATION ${CMAKE_CURRENT_BINARY_DIR})
 install(FILES parallel_global_test.sh DESTINATION ${CMAKE_CURRENT_BINARY_DIR})
 install(FILES parallel_kpoints_test.sh DESTINATION ${CMAKE_CURRENT_BINARY_DIR})
+install(FILES parallel_reduce_test.sh DESTINATION ${CMAKE_CURRENT_BINARY_DIR})
 
 find_program(BASH bash)
 
@@ -34,3 +41,7 @@ add_test(NAME parallel_kpoints_test
       COMMAND ${BASH} parallel_kpoints_test.sh
       WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
 )
+add_test(NAME parallel_reduce_test
+      COMMAND ${BASH} parallel_reduce_test.sh
+      WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
+)