test(mpi): UT of 2 functions in parallel_global.cpp

Author: hongriTianqi

source/src_parallel/test/CMakeLists.txt

@@ -4,10 +4,22 @@ AddTest(
   SOURCES parallel_common_test.cpp ../../module_base/global_variable.cpp ../parallel_common.cpp
 )
 
+AddTest(
+  TARGET ParaGlobal
+  LIBS MPI::MPI_CXX 
+  SOURCES parallel_global_test.cpp ../../module_base/global_variable.cpp ../parallel_global.cpp
+)
+
 install(FILES parallel_common_test.sh DESTINATION ${CMAKE_CURRENT_BINARY_DIR})
+install(FILES parallel_global_test.sh DESTINATION ${CMAKE_CURRENT_BINARY_DIR})
 
 find_program(BASH bash)
+
 add_test(NAME parallel_common_test
       COMMAND ${BASH} parallel_common_test.sh
       WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
 )
+add_test(NAME parallel_global_test
+      COMMAND ${BASH} parallel_global_test.sh
+      WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
+)

source/src_parallel/test/parallel_global_test.cpp

@@ -0,0 +1,100 @@
+#ifdef __MPI
+#include "mpi.h"
+#include "gtest/gtest.h"
+#include "module_base/global_variable.h"
+#include "src_parallel/parallel_global.h"
+#include <complex>
+#include <string>
+#include <cstring>
+
+/************************************************
+ *  unit test of functions in parallel_global.cpp
+ ***********************************************/
+
+/**
+ * The tested functions are:
+ *   i. Parallel_Global::split_diag_world(), which is
+ *   used in David diagonalization in pw basis 
+ *   calculation.
+ *   ii. Parallel_Global::split_grid_world()
+ */
+
+TEST(ParaGlobal,SplitGrid)
+{
+	// NPROC is set to 6 in parallel_global_test.sh
+	if(GlobalV::NPROC==6)
+	{
+		Parallel_Global::split_grid_world(2);
+		EXPECT_EQ(GlobalV::GSIZE,3);
+		if(GlobalV::MY_RANK==0) EXPECT_EQ(GlobalV::GRANK,0);
+		if(GlobalV::MY_RANK==1) EXPECT_EQ(GlobalV::GRANK,1);
+		if(GlobalV::MY_RANK==2) EXPECT_EQ(GlobalV::GRANK,2);
+		if(GlobalV::MY_RANK==3) EXPECT_EQ(GlobalV::GRANK,0);
+		if(GlobalV::MY_RANK==4) EXPECT_EQ(GlobalV::GRANK,1);
+		if(GlobalV::MY_RANK==5) EXPECT_EQ(GlobalV::GRANK,2);
+		Parallel_Global::split_grid_world(6);
+		EXPECT_EQ(GlobalV::GSIZE,1);
+		if(GlobalV::MY_RANK==0) EXPECT_EQ(GlobalV::GRANK,0);
+		if(GlobalV::MY_RANK==1) EXPECT_EQ(GlobalV::GRANK,0);
+		if(GlobalV::MY_RANK==2) EXPECT_EQ(GlobalV::GRANK,0);
+		if(GlobalV::MY_RANK==3) EXPECT_EQ(GlobalV::GRANK,0);
+		if(GlobalV::MY_RANK==4) EXPECT_EQ(GlobalV::GRANK,0);
+		if(GlobalV::MY_RANK==5) EXPECT_EQ(GlobalV::GRANK,0);
+	}
+	else
+	{
+		Parallel_Global::split_grid_world(GlobalV::NPROC);
+		EXPECT_EQ(GlobalV::GSIZE,1);
+		EXPECT_EQ(GlobalV::GRANK,0);
+	}
+	//std::cout<<GlobalV::MY_RANK<<" "<<GlobalV::NPROC<<" ";
+	//std::cout<<GlobalV::GRANK<<" "<<GlobalV::GSIZE<<std::endl;
+}
+
+TEST(ParaGlobal,SplitDiag)
+{
+	// NPROC is set to 6 in parallel_global_test.sh
+	if(GlobalV::NPROC==6)
+	{
+		Parallel_Global::split_diag_world(2);
+		EXPECT_EQ(GlobalV::DSIZE,2);
+		if(GlobalV::MY_RANK==0) EXPECT_EQ(GlobalV::DRANK,0);
+		if(GlobalV::MY_RANK==1) EXPECT_EQ(GlobalV::DRANK,0);
+		if(GlobalV::MY_RANK==2) EXPECT_EQ(GlobalV::DRANK,0);
+		if(GlobalV::MY_RANK==3) EXPECT_EQ(GlobalV::DRANK,1);
+		if(GlobalV::MY_RANK==4) EXPECT_EQ(GlobalV::DRANK,1);
+		if(GlobalV::MY_RANK==5) EXPECT_EQ(GlobalV::DRANK,1);
+		Parallel_Global::split_diag_world(6);
+		EXPECT_EQ(GlobalV::DSIZE,6);
+		if(GlobalV::MY_RANK==0) EXPECT_EQ(GlobalV::DRANK,0);
+		if(GlobalV::MY_RANK==1) EXPECT_EQ(GlobalV::DRANK,1);
+		if(GlobalV::MY_RANK==2) EXPECT_EQ(GlobalV::DRANK,2);
+		if(GlobalV::MY_RANK==3) EXPECT_EQ(GlobalV::DRANK,3);
+		if(GlobalV::MY_RANK==4) EXPECT_EQ(GlobalV::DRANK,4);
+		if(GlobalV::MY_RANK==5) EXPECT_EQ(GlobalV::DRANK,5);
+	}
+	else
+	{
+		Parallel_Global::split_diag_world(GlobalV::NPROC);
+		EXPECT_EQ(GlobalV::DSIZE,GlobalV::NPROC);
+	}
+	//std::cout<<GlobalV::MY_RANK<<" "<<GlobalV::NPROC<<" ";
+	//std::cout<<GlobalV::DRANK<<" "<<GlobalV::DSIZE<<std::endl;
+}
+
+int main(int argc, char **argv)
+{
+
+    MPI_Init(&argc, &argv);
+    testing::InitGoogleTest(&argc, argv);
+
+    MPI_Comm_size(MPI_COMM_WORLD,&GlobalV::NPROC);
+    MPI_Comm_rank(MPI_COMM_WORLD,&GlobalV::MY_RANK);
+
+    int result = RUN_ALL_TESTS();
+
+    MPI_Finalize();
+
+    return result;
+}
+#endif

source/src_parallel/test/parallel_global_test.sh

@@ -0,0 +1,13 @@
+#!/bin/bash
+
+np=`cat /proc/cpuinfo | grep "cpu cores" | uniq| awk '{print $NF}'`
+echo "nprocs in this machine is $np"
+
+for i in 6;do
+    if [[ $i -gt $np ]];then
+        continue
+    fi
+    echo "TEST in parallel, nprocs=$i"
+    mpirun -np $i ./ParaGlobal
+    break    
+done