37 lines
883 B
C++
37 lines
883 B
C++
#define RABIT_CXXTESTDEFS_H
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <string>
|
|
#include <iostream>
|
|
#include "../../src/allreduce_mock.h"
|
|
|
|
TEST(allreduce_mock, mock_allreduce)
|
|
{
|
|
rabit::engine::AllreduceMock m;
|
|
|
|
std::string mock_str = "mock=0,0,0,0";
|
|
char cmd[mock_str.size()+1];
|
|
std::copy(mock_str.begin(), mock_str.end(), cmd);
|
|
cmd[mock_str.size()] = '\0';
|
|
|
|
char* argv[] = {cmd};
|
|
m.Init(1, argv);
|
|
m.rank = 0;
|
|
EXPECT_THROW(m.Allreduce(nullptr,0,0,nullptr,nullptr,nullptr), dmlc::Error);
|
|
}
|
|
|
|
TEST(allreduce_mock, mock_broadcast)
|
|
{
|
|
rabit::engine::AllreduceMock m;
|
|
std::string mock_str = "mock=0,1,2,0";
|
|
char cmd[mock_str.size()+1];
|
|
std::copy(mock_str.begin(), mock_str.end(), cmd);
|
|
cmd[mock_str.size()] = '\0';
|
|
char* argv[] = {cmd};
|
|
m.Init(1, argv);
|
|
m.rank = 0;
|
|
m.version_number=1;
|
|
m.seq_counter=2;
|
|
EXPECT_THROW(m.Broadcast(nullptr,0,0), dmlc::Error);
|
|
}
|