Merge pull request #905 from hongriTianqi/develop

test(mpi): UT of functions in parallel_common.cpp

Author: hongriTianqi

source/src_parallel/CMakeLists.txt

@@ -13,4 +13,8 @@ add_library(
     parallel
     OBJECT
     ${objects}
-)
+)
+
+IF (BUILD_TESTING)
+  add_subdirectory(test)
+endif()

source/src_parallel/test/CMakeLists.txt

@@ -0,0 +1,13 @@
+AddTest(
+  TARGET ParaCommon
+  LIBS MPI::MPI_CXX 
+  SOURCES parallel_common_test.cpp ../../module_base/global_variable.cpp ../parallel_common.cpp
+)
+
+install(FILES parallel_common_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}
+)

source/src_parallel/test/parallel_common_test.cpp

@@ -0,0 +1,105 @@
+#ifdef __MPI
+#include "mpi.h"
+#include "gtest/gtest.h"
+#include "module_base/global_variable.h"
+#include "src_parallel/parallel_common.h"
+#include <complex>
+#include <string>
+#include <cstring>
+
+/************************************************
+ *  unit test of functions in parallel_common.cpp
+ ***********************************************/
+
+/**
+ * The tested functions are wrappers of MPI_Bcast
+ * in ABACUS, as defined in src_parallel/parallel_common.h.
+ * The source is process 0 in all MPI_Bcast 
+ * wrappers.
+ */
+
+class ParaCommon : public testing::Test
+{
+protected:
+	bool boo=true;
+	int is=0;
+	double fs = 0.0;
+	std::complex<double> imgs{0.0,0.0};
+	std::string chs ="abacus";
+	char cha[7]="abacus";
+	int iv[10]={0};
+	double fv[10] = {0};
+	std::complex<double> imgv[10]={0.0,0.0};
+	std::string chv[10] = {""};
+};
+
+TEST_F(ParaCommon,Bcast)
+{
+    // reset data in the first process
+    if(GlobalV::MY_RANK==0) 
+    {
+	    boo=false;
+	    is=1;
+	    fs=1.0;
+	    imgs=std::complex<double>(1.0,-1.0);
+	    chs ="ABACUS";
+	    strcpy(cha,chs.c_str());
+	    for (int i=0;i<10;i++)
+	    {
+		double ii=static_cast<double>(i);
+	        iv[i] = i;
+		fv[i] = ii;
+		imgv[i]=std::complex<double>(ii,-ii);
+		std::stringstream ss;
+		ss<<chs<<i;
+		chv[i] = ss.str();
+	    }
+    }
+    // call bcast wrappers
+    Parallel_Common::bcast_bool(boo);
+    Parallel_Common::bcast_int(is);
+    Parallel_Common::bcast_double(fs);
+    Parallel_Common::bcast_complex_double(imgs);
+    Parallel_Common::bcast_string(chs);
+    Parallel_Common::bcast_char(cha,7);
+    Parallel_Common::bcast_int(iv,10);
+    Parallel_Common::bcast_double(fv,10);
+    Parallel_Common::bcast_complex_double(imgv,10);
+    Parallel_Common::bcast_string(chv,10);
+    // make comparisons
+    EXPECT_FALSE(boo);
+    EXPECT_EQ(is,1);
+    EXPECT_EQ(fs,1.0);
+    EXPECT_NEAR(imgs.real(),1.0,1E-15);
+    EXPECT_NEAR(imgs.imag(),-1.0,1E-15);
+    EXPECT_EQ(chs,"ABACUS");
+    EXPECT_STREQ(cha,"ABACUS");
+    for (int i=0;i<10;i++)
+    {
+	double ii=static_cast<double>(i);
+    	EXPECT_EQ(iv[i],i);
+    	EXPECT_NEAR(fv[i],ii,1E-15);
+    	EXPECT_NEAR(imgv[i].real(),ii,1E-15);
+    	EXPECT_NEAR(imgv[i].imag(),-ii,1E-15);
+	std::stringstream ss;
+	ss<<chs<<i;
+	EXPECT_EQ(chv[i],ss.str());
+    }
+}
+
+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_common_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 3;do
+    if [[ $i -gt $np ]];then
+        continue
+    fi
+    echo "TEST in parallel, nprocs=$i"
+    mpirun -np $i ./ParaCommon
+    break    
+done