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
|
//===-- Unittests for BlockStore ------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "src/__support/blockstore.h"
#include "test/UnitTest/Test.h"
struct Element {
int a;
long b;
unsigned c;
};
class LlvmLibcBlockStoreTest : public __llvm_libc::testing::Test {
public:
template <size_t BLOCK_SIZE, size_t ELEMENT_COUNT, bool REVERSE>
void populate_and_iterate() {
__llvm_libc::cpp::BlockStore<Element, BLOCK_SIZE, REVERSE> block_store;
for (int i = 0; i < int(ELEMENT_COUNT); ++i)
ASSERT_TRUE(block_store.push_back({i, 2 * i, 3 * unsigned(i)}));
auto end = block_store.end();
int i = 0;
for (auto iter = block_store.begin(); iter != end; ++iter, ++i) {
Element &e = *iter;
if (REVERSE) {
int j = ELEMENT_COUNT - 1 - i;
ASSERT_EQ(e.a, j);
ASSERT_EQ(e.b, long(j * 2));
ASSERT_EQ(e.c, unsigned(j * 3));
} else {
ASSERT_EQ(e.a, i);
ASSERT_EQ(e.b, long(i * 2));
ASSERT_EQ(e.c, unsigned(i * 3));
}
}
ASSERT_EQ(i, int(ELEMENT_COUNT));
__llvm_libc::cpp::BlockStore<Element, BLOCK_SIZE, REVERSE>::destroy(
&block_store);
}
template <bool REVERSE> void back_test() {
using __llvm_libc::cpp::BlockStore;
BlockStore<int, 4, REVERSE> block_store;
for (int i = 0; i < 20; i++)
ASSERT_TRUE(block_store.push_back(i));
for (int i = 19; i >= 0; i--, block_store.pop_back())
ASSERT_EQ(block_store.back(), i);
block_store.destroy(&block_store);
}
template <bool REVERSE> void empty_test() {
using __llvm_libc::cpp::BlockStore;
BlockStore<int, 2, REVERSE> block_store;
ASSERT_TRUE(block_store.empty());
ASSERT_TRUE(block_store.push_back(1));
for (int i = 0; i < 10; i++) {
ASSERT_FALSE(block_store.empty());
ASSERT_TRUE(block_store.push_back(1));
}
block_store.destroy(&block_store);
}
};
TEST_F(LlvmLibcBlockStoreTest, PopulateAndIterate4) {
populate_and_iterate<4, 4, false>();
}
TEST_F(LlvmLibcBlockStoreTest, PopulateAndIterate8) {
populate_and_iterate<4, 8, false>();
}
TEST_F(LlvmLibcBlockStoreTest, PopulateAndIterate10) {
populate_and_iterate<4, 10, false>();
}
TEST_F(LlvmLibcBlockStoreTest, PopulateAndIterateReverse4) {
populate_and_iterate<4, 4, true>();
}
TEST_F(LlvmLibcBlockStoreTest, PopulateAndIterateReverse8) {
populate_and_iterate<4, 8, true>();
}
TEST_F(LlvmLibcBlockStoreTest, PopulateAndIterateReverse10) {
populate_and_iterate<4, 10, true>();
}
TEST_F(LlvmLibcBlockStoreTest, Back) { back_test<false>(); }
// FIXME: Combing this test with the above test makes the AMDGPU backend
// generate code which hangs. This should be fixed in the clang compiler.
TEST_F(LlvmLibcBlockStoreTest, BackReverse) { back_test<true>(); }
TEST_F(LlvmLibcBlockStoreTest, Empty) {
empty_test<false>();
empty_test<true>();
}
|
