Skip to content

Commit

Permalink
#702: test_termination_chaining_collective: clean up test code
Browse files Browse the repository at this point in the history
  • Loading branch information
nmm0 committed Aug 25, 2020
1 parent a6e53bb commit 727bd06
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions tests/unit/termination/test_term_chaining.cc
Original file line number Diff line number Diff line change
Expand Up @@ -236,25 +236,21 @@ TEST_F(TestTermChaining, test_termination_chaining_1) {
TEST_F(TestTermChaining, test_termination_chaining_collective_1) {
auto const& num_nodes = theContext()->getNumNodes();

if (num_nodes == 2) {
chain = vt::messaging::DependentSendChain{};
handler_count = 0;

chain = vt::messaging::DependentSendChain{};
if (num_nodes == 2) {
epoch = theTerm()->makeEpochCollective();

handler_count = 0;

theMsg()->pushEpoch(epoch);
chain_reduce();
theTerm()->finishedEpoch(epoch);
theMsg()->popEpoch(epoch);
vt::runSchedulerThrough(epoch);
EXPECT_EQ(handler_count, 3);
} else if (num_nodes == 1) {
chain = vt::messaging::DependentSendChain{};
epoch = theTerm()->makeEpochCollective();

handler_count = 0;

theMsg()->pushEpoch(epoch);
chain_reduce_single();
theTerm()->finishedEpoch(epoch);
Expand Down

0 comments on commit 727bd06

Please sign in to comment.