@@ -29,18 +29,15 @@ class IstateInfoTest : public ::testing::Test
2929{
3030 protected:
3131 K_Vectors* kv = nullptr ;
32- Parallel_Kpoints* Pkpoints = nullptr ;
3332 ModuleBase::matrix ekb;
3433 ModuleBase::matrix wg;
3534 void SetUp ()
3635 {
3736 kv = new K_Vectors;
38- Pkpoints = new Parallel_Kpoints;
3937 }
4038 void TearDown ()
4139 {
4240 delete kv;
43- delete Pkpoints;
4441 }
4542};
4643
@@ -64,12 +61,11 @@ TEST_F(IstateInfoTest, OutIstateInfoS1)
6461 GlobalV::MY_POOL);
6562 kv->set_nkstot (100 );
6663 int nkstot = kv->get_nkstot ();
67- Pkpoints
68- ->kinfo (nkstot, GlobalV::KPAR, GlobalV::MY_POOL, GlobalV::RANK_IN_POOL, GlobalV::NPROC_IN_POOL, PARAM.input .nspin );
64+ kv->para_k .kinfo (nkstot, GlobalV::KPAR, GlobalV::MY_POOL, GlobalV::RANK_IN_POOL, GlobalV::NPROC_IN_POOL, PARAM.input .nspin );
6965 // std::cout<<"my_rank "<<GlobalV::MY_RANK<<" pool rank/size: "
7066 // <<GlobalV::RANK_IN_POOL<<"/"<<GlobalV::NPROC_IN_POOL<<std::endl;
7167 // std::cout<<"MY_POOL "<<GlobalV::MY_POOL<<std::endl;
72- kv->set_nks (Pkpoints-> nks_pool [GlobalV::MY_POOL]);
68+ kv->set_nks (kv-> para_k . nks_pool [GlobalV::MY_POOL]);
7369 // std::cout<<"nks "<<kv->get_nks()<<std::endl;
7470 ekb.create (kv->get_nks (), PARAM.input .nbands );
7571 wg.create (kv->get_nks (), PARAM.input .nbands );
@@ -82,7 +78,7 @@ TEST_F(IstateInfoTest, OutIstateInfoS1)
8278 kd.set (0.01 * i, 0.01 * i, 0.01 * i);
8379 ++i;
8480 }
85- ModuleIO::write_istate_info (ekb, wg, *kv, Pkpoints );
81+ ModuleIO::write_istate_info (ekb, wg, *kv);
8682 std::ifstream ifs;
8783 ifs.open (" istate.info" );
8884 std::string str ((std::istreambuf_iterator<char >(ifs)), std::istreambuf_iterator<char >());
@@ -115,12 +111,11 @@ TEST_F(IstateInfoTest, OutIstateInfoS2)
115111 GlobalV::MY_POOL);
116112 kv->set_nkstot (100 );
117113 int nkstot = kv->get_nkstot ();
118- Pkpoints
119- ->kinfo (nkstot, GlobalV::KPAR, GlobalV::MY_POOL, GlobalV::RANK_IN_POOL, GlobalV::NPROC_IN_POOL, PARAM.input .nspin );
114+ kv->para_k .kinfo (nkstot, GlobalV::KPAR, GlobalV::MY_POOL, GlobalV::RANK_IN_POOL, GlobalV::NPROC_IN_POOL, PARAM.input .nspin );
120115 // std::cout<<"my_rank "<<GlobalV::MY_RANK<<" pool rank/size: "
121116 // <<GlobalV::RANK_IN_POOL<<"/"<<GlobalV::NPROC_IN_POOL<<std::endl;
122117 // std::cout<<"MY_POOL "<<GlobalV::MY_POOL<<std::endl;
123- kv->set_nks (Pkpoints-> nks_pool [GlobalV::MY_POOL]);
118+ kv->set_nks (kv-> para_k . nks_pool [GlobalV::MY_POOL]);
124119 // std::cout<<"nks "<<kv->get_nks()<<std::endl;
125120 ekb.create (kv->get_nks (), PARAM.input .nbands );
126121 wg.create (kv->get_nks (), PARAM.input .nbands );
@@ -133,7 +128,7 @@ TEST_F(IstateInfoTest, OutIstateInfoS2)
133128 kd.set (0.01 * i, 0.01 * i, 0.01 * i);
134129 ++i;
135130 }
136- ModuleIO::write_istate_info (ekb, wg, *kv, Pkpoints );
131+ ModuleIO::write_istate_info (ekb, wg, *kv);
137132 std::ifstream ifs;
138133 ifs.open (" istate.info" );
139134 std::string str ((std::istreambuf_iterator<char >(ifs)), std::istreambuf_iterator<char >());
0 commit comments