1 // Copyright 2024 The Pigweed Authors
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License"); you may not
4 // use this file except in compliance with the License. You may obtain a copy of
5 // the License at
6 //
7 // https://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
11 // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12 // License for the specific language governing permissions and limitations under
13 // the License.
14
15 #include "pw_allocator/fragmentation.h"
16
17 #include "pw_unit_test/framework.h"
18
19 namespace {
20
21 using ::pw::allocator::Fragmentation;
22
TEST(FragmentationTest,ValuesAreCorrect)23 TEST(FragmentationTest, ValuesAreCorrect) {
24 Fragmentation fragmentation;
25 fragmentation.AddFragment(867);
26 fragmentation.AddFragment(5309);
27 EXPECT_EQ(fragmentation.sum_of_squares.hi, 0U);
28 EXPECT_EQ(fragmentation.sum_of_squares.lo, 867U * 867U + 5309U * 5309U);
29 EXPECT_EQ(fragmentation.sum, 867U + 5309U);
30 }
31
TEST(FragmentationTest,HandlesOverflow)32 TEST(FragmentationTest, HandlesOverflow) {
33 constexpr size_t kHalfWord = size_t(1) << sizeof(size_t) * 4;
34 Fragmentation fragmentation;
35 fragmentation.AddFragment(kHalfWord);
36 fragmentation.AddFragment(kHalfWord);
37 fragmentation.AddFragment(kHalfWord);
38 fragmentation.AddFragment(kHalfWord);
39 EXPECT_EQ(fragmentation.sum_of_squares.hi, 4U);
40 EXPECT_EQ(fragmentation.sum_of_squares.lo, 0U);
41 EXPECT_EQ(fragmentation.sum, 4 * kHalfWord);
42 }
43
44 } // namespace
45