1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*
* Copyright (C) 2018 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "bit_table.h"
#include "gtest/gtest.h"
namespace art {
TEST(BitTableTest, TestVarint) {
for (size_t start_bit_offset = 0; start_bit_offset <= 32; start_bit_offset++) {
uint32_t values[] = { 0, 1, 11, 12, 15, 16, 255, 256, ~1u, ~0u };
for (uint32_t value : values) {
std::vector<uint8_t> buffer;
size_t encode_bit_offset = start_bit_offset;
EncodeVarintBits(&buffer, &encode_bit_offset, value);
size_t decode_bit_offset = start_bit_offset;
BitMemoryRegion region(MemoryRegion(buffer.data(), buffer.size()));
uint32_t result = DecodeVarintBits(region, &decode_bit_offset);
EXPECT_EQ(encode_bit_offset, decode_bit_offset);
EXPECT_EQ(value, result);
}
}
}
TEST(BitTableTest, TestEmptyTable) {
std::vector<uint8_t> buffer;
size_t encode_bit_offset = 0;
BitTableBuilder<1> builder;
builder.Encode(&buffer, &encode_bit_offset);
size_t decode_bit_offset = 0;
BitTable<1> table(buffer.data(), buffer.size(), &decode_bit_offset);
EXPECT_EQ(encode_bit_offset, decode_bit_offset);
EXPECT_EQ(0u, table.NumRows());
}
TEST(BitTableTest, TestSingleColumnTable) {
constexpr uint32_t kNoValue = -1;
std::vector<uint8_t> buffer;
size_t encode_bit_offset = 0;
BitTableBuilder<1> builder;
builder.AddRow(42u);
builder.AddRow(kNoValue);
builder.AddRow(1000u);
builder.AddRow(kNoValue);
builder.Encode(&buffer, &encode_bit_offset);
size_t decode_bit_offset = 0;
BitTable<1> table(buffer.data(), buffer.size(), &decode_bit_offset);
EXPECT_EQ(encode_bit_offset, decode_bit_offset);
EXPECT_EQ(4u, table.NumRows());
EXPECT_EQ(42u, table.Get(0));
EXPECT_EQ(kNoValue, table.Get(1));
EXPECT_EQ(1000u, table.Get(2));
EXPECT_EQ(kNoValue, table.Get(3));
EXPECT_EQ(10u, table.NumColumnBits(0));
}
TEST(BitTableTest, TestUnalignedTable) {
for (size_t start_bit_offset = 0; start_bit_offset <= 32; start_bit_offset++) {
std::vector<uint8_t> buffer;
size_t encode_bit_offset = start_bit_offset;
BitTableBuilder<1> builder;
builder.AddRow(42u);
builder.Encode(&buffer, &encode_bit_offset);
size_t decode_bit_offset = start_bit_offset;
BitTable<1> table(buffer.data(), buffer.size(), &decode_bit_offset);
EXPECT_EQ(encode_bit_offset, decode_bit_offset) << " start_bit_offset=" << start_bit_offset;
EXPECT_EQ(1u, table.NumRows());
EXPECT_EQ(42u, table.Get(0));
}
}
TEST(BitTableTest, TestBigTable) {
constexpr uint32_t kNoValue = -1;
std::vector<uint8_t> buffer;
size_t encode_bit_offset = 0;
BitTableBuilder<4> builder;
builder.AddRow(42u, kNoValue, 0u, static_cast<uint32_t>(-2));
builder.AddRow(62u, kNoValue, 63u, static_cast<uint32_t>(-3));
builder.Encode(&buffer, &encode_bit_offset);
size_t decode_bit_offset = 0;
BitTable<4> table(buffer.data(), buffer.size(), &decode_bit_offset);
EXPECT_EQ(encode_bit_offset, decode_bit_offset);
EXPECT_EQ(2u, table.NumRows());
EXPECT_EQ(42u, table.Get(0, 0));
EXPECT_EQ(kNoValue, table.Get(0, 1));
EXPECT_EQ(0u, table.Get(0, 2));
EXPECT_EQ(static_cast<uint32_t>(-2), table.Get(0, 3));
EXPECT_EQ(62u, table.Get(1, 0));
EXPECT_EQ(kNoValue, table.Get(1, 1));
EXPECT_EQ(63u, table.Get(1, 2));
EXPECT_EQ(static_cast<uint32_t>(-3), table.Get(1, 3));
EXPECT_EQ(6u, table.NumColumnBits(0));
EXPECT_EQ(0u, table.NumColumnBits(1));
EXPECT_EQ(7u, table.NumColumnBits(2));
EXPECT_EQ(32u, table.NumColumnBits(3));
}
} // namespace art
|