Skip to content

Commit e096a2f

Browse files
committed
[EquivClasses] Address review
1 parent dafbe78 commit e096a2f

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

llvm/unittests/ADT/EquivalenceClassesTest.cpp

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
//===----------------------------------------------------------------------===//
88

99
#include "llvm/ADT/EquivalenceClasses.h"
10+
#include "gmock/gmock.h"
1011
#include "gtest/gtest.h"
1112

1213
using namespace llvm;
@@ -74,12 +75,9 @@ TEST(EquivalenceClassesTest, MembersIterator) {
7475
EC.unionSets(5, 1);
7576
EXPECT_EQ(EC.getNumClasses(), 2u);
7677

77-
std::set<int> ActualMembers;
78-
std::set<int> ExpectedMembers{1, 2, 5};
7978
EquivalenceClasses<int>::iterator I = EC.findValue(EC.getLeaderValue(1));
80-
for (auto M : EC.members(I))
81-
ActualMembers.insert(M);
82-
EXPECT_EQ(ActualMembers, ExpectedMembers);
79+
EXPECT_THAT(EC.members(I), testing::ElementsAre(1, 2, 5));
80+
EXPECT_THAT(EC.members(EC.end()), testing::IsEmpty());
8381
}
8482

8583
// Type-parameterized tests: Run the same test cases with different element

0 commit comments

Comments
 (0)