diff options
Diffstat (limited to 'llvm/unittests/ProfileData/InstrProfTest.cpp')
| -rw-r--r-- | llvm/unittests/ProfileData/InstrProfTest.cpp | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/llvm/unittests/ProfileData/InstrProfTest.cpp b/llvm/unittests/ProfileData/InstrProfTest.cpp index f366b22..273b89c 100644 --- a/llvm/unittests/ProfileData/InstrProfTest.cpp +++ b/llvm/unittests/ProfileData/InstrProfTest.cpp @@ -26,6 +26,7 @@ using namespace llvm; using ::llvm::memprof::LineLocation; +using ::testing::ElementsAre; using ::testing::EndsWith; using ::testing::IsSubsetOf; using ::testing::Pair; @@ -564,19 +565,16 @@ TEST_F(InstrProfTest, test_caller_callee_pairs) { auto It = Pairs.find(0x123); ASSERT_NE(It, Pairs.end()); - ASSERT_THAT(It->second, SizeIs(2)); - EXPECT_THAT(It->second[0], Pair(LineLocation(1, 2), 0x234U)); - EXPECT_THAT(It->second[1], Pair(LineLocation(5, 6), 0x345U)); + EXPECT_THAT(It->second, ElementsAre(Pair(LineLocation(1, 2), 0x234U), + Pair(LineLocation(5, 6), 0x345U))); It = Pairs.find(0x234); ASSERT_NE(It, Pairs.end()); - ASSERT_THAT(It->second, SizeIs(1)); - EXPECT_THAT(It->second[0], Pair(LineLocation(3, 4), 0U)); + EXPECT_THAT(It->second, ElementsAre(Pair(LineLocation(3, 4), 0U))); It = Pairs.find(0x345); ASSERT_NE(It, Pairs.end()); - ASSERT_THAT(It->second, SizeIs(1)); - EXPECT_THAT(It->second[0], Pair(LineLocation(7, 8), 0U)); + EXPECT_THAT(It->second, ElementsAre(Pair(LineLocation(7, 8), 0U))); } TEST_F(InstrProfTest, test_memprof_getrecord_error) { |