UT of MemAvalibale,TEST_LEVEL and Tamplate of V*M (#2004)

Author: yaohaoze

source/module_base/global_function.cpp

@@ -112,9 +112,8 @@ void DONE(std::ofstream &ofs, const std::string &description, const bool only_ra
 // NAME : TEST_LEVEL
 // control the test_level
 //==========================================================
-void TEST_LEVEL(const std::string &name)
+void TEST_LEVEL(const std::string &name, bool disable=true)
 {
-    bool disable = true;
     if (disable) return;
 
     if (name == "none")

source/module_base/global_function.h

@@ -141,7 +141,7 @@ inline void ZEROS(T *u,const TI n)		// Peize Lin change int to TI at 2020.03.03
 // NAME : TEST_LEVEL
 // control the test_level
 //==========================================================
-void TEST_LEVEL(const std::string &name);
+void TEST_LEVEL(const std::string &name, bool disable);
 
 
 //==========================================================

source/module_base/test/global_function_test.cpp

@@ -60,6 +60,10 @@
  *   - get a copy of the ptr of a vector
  * - VECTOR_TO_PTR_v3double
  *    - get a copy of the ptr of a vector whose elements' type belongs to Vector3<double>
+ * - MemAvailable
+ *    - get the current memory valus
+ * - TETS_LEVEL
+ *    - set the test level
  */
 
 inline void EXPECT_COMPLEX_FLOAT_EQ(const std::complex<float>& a, const std::complex<float>& b)
@@ -659,6 +663,56 @@ TEST_F(GlobalFunctionTest,Vector2Ptr)
     }
 }
 
+TEST_F(GlobalFunctionTest,MemAvailable)
+{
+    for(int i=0;i<5;i++)
+    {
+        std::ifstream ifs("/proc/meminfo");
+        while (ifs.good())
+        {
+            std::string label, size, kB;
+            ifs >> label >> size >> kB;
+            if (label == "MemAvailable:")
+            {
+                EXPECT_LE(std::stol(size)-1000,ModuleBase::GlobalFunc::MemAvailable());
+                EXPECT_GE(std::stol(size)+1000,ModuleBase::GlobalFunc::MemAvailable());
+            }
+        }
+    }
+}
+
+TEST_F(GlobalFunctionTest,TEST_LEVEL)
+{
+    std::string name;
+    bool test_bool=false;
+    name="none";
+    ModuleBase::GlobalFunc::TEST_LEVEL(name,test_bool);
+    EXPECT_EQ(GlobalV::test_wf,0);
+    EXPECT_EQ(GlobalV::test_potential,0);
+    EXPECT_EQ(GlobalV::test_charge,0);
+    name="init_potential";
+    ModuleBase::GlobalFunc::TEST_LEVEL(name,test_bool);
+    EXPECT_EQ(GlobalV::test_wf,1);
+    EXPECT_EQ(GlobalV::test_potential,1);
+    EXPECT_EQ(GlobalV::test_charge,1);
+    name="init_read";
+    ModuleBase::GlobalFunc::TEST_LEVEL(name,test_bool);
+    EXPECT_EQ(GlobalV::test_input,1);
+    EXPECT_EQ(GlobalV::test_winput,1);
+    EXPECT_EQ(GlobalV::test_kpoint,1);
+    EXPECT_EQ(GlobalV::test_atom,1);
+    EXPECT_EQ(GlobalV::test_unitcell,1);
+#ifndef __EPM
+        EXPECT_EQ(GlobalV::test_pseudo_cell,1);
+#else
+        EXPECT_EQ(test_epm_unitcell,1);
+#endif
+    name="pw_init";
+    ModuleBase::GlobalFunc::TEST_LEVEL(name,test_bool);
+    EXPECT_EQ(GlobalV::test_pw,1);
+}
+
+
 /*
 TEST_F(GlobalFunctionTest, Note)
 {

source/module_base/test/matrix3_test.cpp

@@ -58,6 +58,8 @@
  *   - ToMatrix
  *     - change the form of a 3x3 matrix from that of class Matrix3
  *     - to that of class matrix
+ *	 -Template V Multiply M
+ *	   -tamplate classes for (Vector3)*(3x3 matrix)
  */
 
 class Matrix3Test : public testing::Test
@@ -340,3 +342,25 @@ TEST_F(Matrix3Test, ToMatrix)
 	EXPECT_EQ(ma.e21,mb(1,0)); EXPECT_EQ(ma.e22,mb(1,1)); EXPECT_EQ(ma.e23,mb(1,2));
 	EXPECT_EQ(ma.e31,mb(2,0)); EXPECT_EQ(ma.e32,mb(2,1)); EXPECT_EQ(ma.e33,mb(2,2));
 }
+
+TEST_F(Matrix3Test,TemplateVectorMultiplyMatrix)
+{
+	ModuleBase::Vector3<int> ui(1,2,3);
+	ModuleBase::Vector3<double> ud(1.0,2.0,3.0);
+	ModuleBase::Vector3<float> uf(1.0,2.0,3.0);
+	ModuleBase::Vector3<double> vi,vd,vf;
+	ModuleBase::Matrix3 ma;
+	ma = get_random_matrix3();
+	vi=ui*ma;
+	vd=ud*ma;
+	vf=uf*ma;
+	EXPECT_DOUBLE_EQ(vi.x,ui.x * ma.e11 + ui.y * ma.e21 + ui.z * ma.e31);
+	EXPECT_DOUBLE_EQ(vi.y,ui.x * ma.e12 + ui.y * ma.e22 + ui.z * ma.e32);
+	EXPECT_DOUBLE_EQ(vi.z,ui.x * ma.e13 + ui.y * ma.e23 + ui.z * ma.e33);
+	EXPECT_DOUBLE_EQ(vd.x,ud.x * ma.e11 + ud.y * ma.e21 + ud.z * ma.e31);
+	EXPECT_DOUBLE_EQ(vd.y,ud.x * ma.e12 + ud.y * ma.e22 + ud.z * ma.e32);
+	EXPECT_DOUBLE_EQ(vd.z,ud.x * ma.e13 + ud.y * ma.e23 + ud.z * ma.e33);
+	EXPECT_DOUBLE_EQ(vf.x,uf.x * ma.e11 + uf.y * ma.e21 + uf.z * ma.e31);
+	EXPECT_DOUBLE_EQ(vf.y,uf.x * ma.e12 + uf.y * ma.e22 + uf.z * ma.e32);
+	EXPECT_DOUBLE_EQ(vf.z,uf.x * ma.e13 + uf.y * ma.e23 + uf.z * ma.e33);
+}