Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove stop process. (#143)

Browse Source
This commit is contained in:
Jiaming Yuan
2020-08-06 01:12:00 +08:00
committed by GitHub
parent e6cd74ead3
commit 4acdd7c6f6
9 changed files with 25 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/cpp/CMakeLists.txt
View File

@@ -3,6 +3,7 @@ find_package(GTest REQUIRED)
add_executable(
unit_tests
test_io.cc
test_utils.cc
allreduce_robust_test.cc
allreduce_base_test.cc
allreduce_mock_test.cc

4
test/cpp/allreduce_mock_test.cc
View File

@@ -17,7 +17,7 @@ TEST(allreduce_mock, mock_allreduce)
char* argv[] = {cmd};
m.Init(1, argv);
m.rank = 0;
EXPECT_EXIT(m.Allreduce(nullptr,0,0,nullptr,nullptr,nullptr), ::testing::ExitedWithCode(255), "");
EXPECT_THROW(m.Allreduce(nullptr,0,0,nullptr,nullptr,nullptr), dmlc::Error);
}
TEST(allreduce_mock, mock_broadcast)
@@ -32,5 +32,5 @@ TEST(allreduce_mock, mock_broadcast)
m.rank = 0;
m.version_number=1;
m.seq_counter=2;
EXPECT_EXIT(m.Broadcast(nullptr,0,0), ::testing::ExitedWithCode(255), "");
EXPECT_THROW(m.Broadcast(nullptr,0,0), dmlc::Error);
}

7
test/cpp/allreduce_mock_test.cpp
View File

@@ -3,6 +3,7 @@
#include <string>
#include <iostream>
#include <dmlc/logging.h>
#include "../../src/allreduce_mock.h"
TEST(allreduce_mock, mock_allreduce)
@@ -17,7 +18,7 @@ TEST(allreduce_mock, mock_allreduce)
char* argv[] = {cmd};
m.Init(1, argv);
m.rank = 0;
EXPECT_EXIT(m.Allreduce(nullptr,0,0,nullptr,nullptr,nullptr), ::testing::ExitedWithCode(255), "");
EXPECT_THROW({m.Allreduce(nullptr,0,0,nullptr,nullptr,nullptr);}, dmlc::Error);
}
TEST(allreduce_mock, mock_broadcast)
@@ -32,7 +33,7 @@ TEST(allreduce_mock, mock_broadcast)
m.rank = 0;
m.version_number=1;
m.seq_counter=2;
EXPECT_EXIT(m.Broadcast(nullptr,0,0), ::testing::ExitedWithCode(255), "");
EXPECT_THROW({m.Broadcast(nullptr,0,0);}, dmlc::Error);
}
TEST(allreduce_mock, mock_gather)
@@ -47,5 +48,5 @@ TEST(allreduce_mock, mock_gather)
m.rank = 3;
m.version_number=13;
m.seq_counter=22;
EXPECT_EXIT(m.Allgather(nullptr,0,0,0,0), ::testing::ExitedWithCode(255), "");
EXPECT_THROW({m.Allgather(nullptr,0,0,0,0);}, dmlc::Error);
}

6
test/cpp/test_utils.cc Normal file
View File

@@ -0,0 +1,6 @@
#include <gtest/gtest.h>
#include <rabit/internal/utils.h>
TEST(Utils, Assert) {
EXPECT_THROW({rabit::utils::Assert(false, "foo");}, dmlc::Error);
}