@@ -1698,18 +1698,24 @@ TEST_F(SparseTensorAttributeTest, size_on_disk_factor_is_calculated_and_used) {
16981698 }
16991699 EXPECT_LT (10 , attr.getCommittedDocIdLimit ());
17001700 EXPECT_THAT (attr.getEstimatedSaveByteSize (), AllOf (Ge (40_Ki), Le (50_Ki)));
1701- EXPECT_EQ (attr.getCommittedDocIdLimit () * sizeof (EntryRef), attr.transient_memory_for_flush ());
1701+ EXPECT_EQ (attr.getCommittedDocIdLimit () * sizeof (EntryRef), attr.transient_memory_for_flush (false ));
1702+ EXPECT_EQ (attr.getCommittedDocIdLimit () * sizeof (EntryRef) + attr.getEstimatedSaveByteSize (),
1703+ attr.transient_memory_for_flush (true ));
17021704 attr.save ((_test_dir / " tensor" ).string ());
17031705 auto size_on_disk = attr.size_on_disk ();
17041706 EXPECT_LT (60_Ki, size_on_disk);
17051707 EXPECT_THAT (attr.getEstimatedSaveByteSize (), AllOf (Ge (size_on_disk - 4_Ki), Le (size_on_disk + 4_Ki)));
1706- EXPECT_EQ (attr.getCommittedDocIdLimit () * sizeof (EntryRef), attr.transient_memory_for_flush ());
1708+ EXPECT_EQ (attr.getCommittedDocIdLimit () * sizeof (EntryRef), attr.transient_memory_for_flush (false ));
1709+ EXPECT_EQ (attr.getCommittedDocIdLimit () * sizeof (EntryRef) + attr.getEstimatedSaveByteSize (),
1710+ attr.transient_memory_for_flush (true ));
17071711 auto real_attr2 = std::make_shared<SerializedFastValueAttribute>((_test_dir / " tensor" ).string (), cfg);
17081712 AttributeVector& attr2 = *real_attr2;
17091713 ASSERT_TRUE (attr2.load ());
17101714 EXPECT_EQ (size_on_disk, attr2.size_on_disk ());
17111715 EXPECT_THAT (attr2.getEstimatedSaveByteSize (), AllOf (Ge (size_on_disk - 4_Ki), Le (size_on_disk + 4_Ki)));
1712- EXPECT_EQ (attr2.getCommittedDocIdLimit () * sizeof (EntryRef), attr2.transient_memory_for_flush ());
1716+ EXPECT_EQ (attr2.getCommittedDocIdLimit () * sizeof (EntryRef), attr2.transient_memory_for_flush (false ));
1717+ EXPECT_EQ (attr2.getCommittedDocIdLimit () * sizeof (EntryRef) + attr2.getEstimatedSaveByteSize (),
1718+ attr2.transient_memory_for_flush (true ));
17131719 EXPECT_EQ (dynamic_memory_usage, attr2.getStatus ().get_used_minus_dead_and_onhold () - initial_memory_usage);
17141720 EXPECT_EQ (attr.getEstimatedSaveByteSize (), attr2.getEstimatedSaveByteSize ());
17151721}
0 commit comments