| @@ -19,21 +19,18 @@ | |||||
| #define private public | #define private public | ||||
| #define protected public | #define protected public | ||||
| #include "session/inner_session.h" | #include "session/inner_session.h" | ||||
| #undef private | |||||
| #undef protected | |||||
| using namespace std; | using namespace std; | ||||
| namespace ge { | namespace ge { | ||||
| class Utest_Inner_session : public testing::Test { | |||||
| class UtestInnerSession : public testing::Test { | |||||
| protected: | protected: | ||||
| void SetUp() override {} | void SetUp() override {} | ||||
| void TearDown() override {} | void TearDown() override {} | ||||
| }; | }; | ||||
| TEST_F(Utest_Inner_session, build_graph_success) { | |||||
| TEST_F(UtestInnerSession, build_graph_success) { | |||||
| std::map <string, string> options; | std::map <string, string> options; | ||||
| uint64_t session_id = 1; | uint64_t session_id = 1; | ||||
| InnerSession inner_seesion(session_id, options); | InnerSession inner_seesion(session_id, options); | ||||
| @@ -44,17 +41,15 @@ TEST_F(Utest_Inner_session, build_graph_success) { | |||||
| EXPECT_NE(ret, ge::SUCCESS); | EXPECT_NE(ret, ge::SUCCESS); | ||||
| } | } | ||||
| TEST_F(Utest_Inner_session, initialize) { | |||||
| std::map<std::string, std::string> options = { | |||||
| {ge::MODIFY_MIXLIST, "/modify.json"} | |||||
| }; | |||||
| TEST_F(UtestInnerSession, initialize) { | |||||
| std::map<std::string, std::string> options = {}; | |||||
| uint64_t session_id = 1; | uint64_t session_id = 1; | ||||
| InnerSession inner_session(session_id, options); | InnerSession inner_session(session_id, options); | ||||
| auto ret = inner_session.Initialize(); | |||||
| EXPECT_NE(ret, ge::SUCCESS); | |||||
| EXPECT_EQ(inner_session.Initialize(), SUCCESS); | |||||
| EXPECT_EQ(inner_session.Finalize(), SUCCESS); | |||||
| } | } | ||||
| TEST_F(Utest_Inner_session, check_op_precision_mode) { | |||||
| TEST_F(UtestInnerSession, check_op_precision_mode) { | |||||
| std::map<std::string, std::string> options = { | std::map<std::string, std::string> options = { | ||||
| {ge::OP_PRECISION_MODE, "./op_precision_mode.ini"} | {ge::OP_PRECISION_MODE, "./op_precision_mode.ini"} | ||||
| }; | }; | ||||