Skip to content

Commit 3074060

Browse files
[memprof] Use SizeIs (NFC) (#88984)
1 parent d19bd05 commit 3074060

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

llvm/unittests/ProfileData/MemProfTest.cpp

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -183,13 +183,13 @@ TEST(MemProf, FillsValue) {
183183
// We expect 4 records. We attach alloc site data to foo and bar, i.e.
184184
// all frames bottom up until we find a non-inline frame. We attach call site
185185
// data to bar, xyz and abc.
186-
ASSERT_EQ(Records.size(), 4U);
186+
ASSERT_THAT(Records, SizeIs(4));
187187

188188
// Check the memprof record for foo.
189189
const llvm::GlobalValue::GUID FooId = IndexedMemProfRecord::getGUID("foo");
190190
ASSERT_EQ(Records.count(FooId), 1U);
191191
const MemProfRecord &Foo = Records[FooId];
192-
ASSERT_EQ(Foo.AllocSites.size(), 1U);
192+
ASSERT_THAT(Foo.AllocSites, SizeIs(1));
193193
EXPECT_EQ(Foo.AllocSites[0].Info.getAllocCount(), 1U);
194194
EXPECT_THAT(Foo.AllocSites[0].CallStack[0],
195195
FrameContains("foo", 5U, 30U, true));
@@ -205,7 +205,7 @@ TEST(MemProf, FillsValue) {
205205
const llvm::GlobalValue::GUID BarId = IndexedMemProfRecord::getGUID("bar");
206206
ASSERT_EQ(Records.count(BarId), 1U);
207207
const MemProfRecord &Bar = Records[BarId];
208-
ASSERT_EQ(Bar.AllocSites.size(), 1U);
208+
ASSERT_THAT(Bar.AllocSites, SizeIs(1));
209209
EXPECT_EQ(Bar.AllocSites[0].Info.getAllocCount(), 1U);
210210
EXPECT_THAT(Bar.AllocSites[0].CallStack[0],
211211
FrameContains("foo", 5U, 30U, true));
@@ -216,17 +216,17 @@ TEST(MemProf, FillsValue) {
216216
EXPECT_THAT(Bar.AllocSites[0].CallStack[3],
217217
FrameContains("abc", 5U, 30U, false));
218218

219-
ASSERT_EQ(Bar.CallSites.size(), 1U);
220-
ASSERT_EQ(Bar.CallSites[0].size(), 2U);
219+
ASSERT_THAT(Bar.CallSites, SizeIs(1));
220+
ASSERT_THAT(Bar.CallSites[0], SizeIs(2));
221221
EXPECT_THAT(Bar.CallSites[0][0], FrameContains("foo", 5U, 30U, true));
222222
EXPECT_THAT(Bar.CallSites[0][1], FrameContains("bar", 51U, 20U, false));
223223

224224
// Check the memprof record for xyz.
225225
const llvm::GlobalValue::GUID XyzId = IndexedMemProfRecord::getGUID("xyz");
226226
ASSERT_EQ(Records.count(XyzId), 1U);
227227
const MemProfRecord &Xyz = Records[XyzId];
228-
ASSERT_EQ(Xyz.CallSites.size(), 1U);
229-
ASSERT_EQ(Xyz.CallSites[0].size(), 2U);
228+
ASSERT_THAT(Xyz.CallSites, SizeIs(1));
229+
ASSERT_THAT(Xyz.CallSites[0], SizeIs(2));
230230
// Expect the entire frame even though in practice we only need the first
231231
// entry here.
232232
EXPECT_THAT(Xyz.CallSites[0][0], FrameContains("xyz", 5U, 30U, true));
@@ -237,8 +237,8 @@ TEST(MemProf, FillsValue) {
237237
ASSERT_EQ(Records.count(AbcId), 1U);
238238
const MemProfRecord &Abc = Records[AbcId];
239239
EXPECT_TRUE(Abc.AllocSites.empty());
240-
ASSERT_EQ(Abc.CallSites.size(), 1U);
241-
ASSERT_EQ(Abc.CallSites[0].size(), 2U);
240+
ASSERT_THAT(Abc.CallSites, SizeIs(1));
241+
ASSERT_THAT(Abc.CallSites[0], SizeIs(2));
242242
EXPECT_THAT(Abc.CallSites[0][0], FrameContains("xyz", 5U, 30U, true));
243243
EXPECT_THAT(Abc.CallSites[0][1], FrameContains("abc", 5U, 30U, false));
244244
}
@@ -393,9 +393,9 @@ TEST(MemProf, SymbolizationFilter) {
393393
Records.push_back(KeyRecordPair.second);
394394
}
395395

396-
ASSERT_EQ(Records.size(), 1U);
397-
ASSERT_EQ(Records[0].AllocSites.size(), 1U);
398-
ASSERT_EQ(Records[0].AllocSites[0].CallStack.size(), 1U);
396+
ASSERT_THAT(Records, SizeIs(1));
397+
ASSERT_THAT(Records[0].AllocSites, SizeIs(1));
398+
ASSERT_THAT(Records[0].AllocSites[0].CallStack, SizeIs(1));
399399
EXPECT_THAT(Records[0].AllocSites[0].CallStack[0],
400400
FrameContains("foo", 5U, 30U, false));
401401
}
@@ -427,9 +427,9 @@ TEST(MemProf, BaseMemProfReader) {
427427
Records.push_back(KeyRecordPair.second);
428428
}
429429

430-
ASSERT_EQ(Records.size(), 1U);
431-
ASSERT_EQ(Records[0].AllocSites.size(), 1U);
432-
ASSERT_EQ(Records[0].AllocSites[0].CallStack.size(), 2U);
430+
ASSERT_THAT(Records, SizeIs(1));
431+
ASSERT_THAT(Records[0].AllocSites, SizeIs(1));
432+
ASSERT_THAT(Records[0].AllocSites[0].CallStack, SizeIs(2));
433433
EXPECT_THAT(Records[0].AllocSites[0].CallStack[0],
434434
FrameContains("foo", 20U, 5U, true));
435435
EXPECT_THAT(Records[0].AllocSites[0].CallStack[1],

0 commit comments

Comments
 (0)