-
Notifications
You must be signed in to change notification settings - Fork 10.4k
/
Copy pathImmutablePointerSetTest.cpp
129 lines (112 loc) · 4.84 KB
/
ImmutablePointerSetTest.cpp
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
118
119
120
121
122
123
124
125
126
127
128
129
//===--- ImmutablePointerSetTest.cpp --------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#include "swift/Basic/ImmutablePointerSet.h"
#include "llvm/ADT/ArrayRef.h"
#include "llvm/Support/Allocator.h"
#include "gtest/gtest.h"
using namespace swift;
TEST(ImmutableSortedSet, OneElementSets) {
llvm::BumpPtrAllocator BPA;
ImmutablePointerSetFactory<unsigned *> F(BPA);
unsigned *ptr1 = (unsigned *)3;
auto *OneEltSet1 = F.get(ptr1);
EXPECT_EQ(OneEltSet1, F.get(ptr1));
EXPECT_EQ(OneEltSet1, F.merge(OneEltSet1, OneEltSet1));
unsigned *ptr2 = (unsigned *)2;
auto *OneEltSet2 = F.get(ptr2);
EXPECT_EQ(OneEltSet2, F.get(ptr2));
EXPECT_EQ(OneEltSet2, F.merge(OneEltSet2, OneEltSet2));
EXPECT_NE(OneEltSet2, OneEltSet1);
auto *Merge1 = F.merge(OneEltSet1, OneEltSet2);
auto *Merge2 = F.merge(OneEltSet2, OneEltSet1);
EXPECT_NE(OneEltSet1, Merge1);
EXPECT_NE(OneEltSet2, Merge1);
EXPECT_EQ(Merge1, Merge2);
EXPECT_EQ(Merge1, F.merge(Merge1, Merge1));
EXPECT_EQ(Merge2, F.merge(Merge2, Merge2));
EXPECT_EQ(Merge1, F.merge(Merge1, OneEltSet1));
EXPECT_EQ(Merge1, F.merge(Merge1, OneEltSet2));
EXPECT_EQ(Merge1->size(), 2U);
EXPECT_FALSE(Merge1->empty());
EXPECT_EQ(*Merge1->begin(), (unsigned *)2);
EXPECT_EQ(*std::next(Merge1->begin()), (unsigned *)3);
EXPECT_EQ(F.getEmptySet(), F.getEmptySet());
EXPECT_EQ(OneEltSet1, F.merge(F.getEmptySet(), OneEltSet1));
EXPECT_EQ(OneEltSet1, F.merge(OneEltSet1, F.getEmptySet()));
}
TEST(ImmutablePointerSet, MultipleElementSets) {
llvm::BumpPtrAllocator BPA;
ImmutablePointerSetFactory<unsigned *> F(BPA);
unsigned *Ptr1 = (unsigned *)3;
unsigned *Ptr2 = (unsigned *)4;
unsigned *Ptr3 = (unsigned *)3;
unsigned *Ptr4 = (unsigned *)5;
unsigned *Ptr5 = (unsigned *)6;
llvm::SmallVector<unsigned *, 2> Data1 = {Ptr1, Ptr2};
llvm::MutableArrayRef<unsigned *> Array = Data1;
auto *TwoEltSet = F.get(Array);
EXPECT_FALSE(TwoEltSet->empty());
EXPECT_EQ(TwoEltSet->size(), 2u);
EXPECT_TRUE(TwoEltSet->count(Ptr1));
EXPECT_TRUE(TwoEltSet->count(Ptr2));
EXPECT_TRUE(TwoEltSet->count(Ptr3));
EXPECT_FALSE(TwoEltSet->count(Ptr4));
EXPECT_FALSE(TwoEltSet->count(Ptr5));
auto *ThreeEltSet = F.merge(F.get(Ptr4), TwoEltSet);
EXPECT_FALSE(ThreeEltSet->empty());
EXPECT_EQ(ThreeEltSet->size(), 3u);
EXPECT_NE(*ThreeEltSet, *TwoEltSet);
EXPECT_TRUE(ThreeEltSet->count(Ptr1));
EXPECT_TRUE(ThreeEltSet->count(Ptr2));
EXPECT_TRUE(ThreeEltSet->count(Ptr3));
EXPECT_TRUE(ThreeEltSet->count(Ptr4));
EXPECT_FALSE(ThreeEltSet->count(Ptr5));
EXPECT_EQ(ThreeEltSet, F.merge(TwoEltSet, ThreeEltSet));
llvm::SmallVector<unsigned *, 3> Data2 = {Ptr3, Ptr4, Ptr5};
auto *PartialOverlapSet = F.get(Data2);
EXPECT_FALSE(PartialOverlapSet->empty());
EXPECT_EQ(PartialOverlapSet->size(), 3u);
EXPECT_TRUE(PartialOverlapSet->count(Ptr1));
EXPECT_FALSE(PartialOverlapSet->count(Ptr2));
EXPECT_TRUE(PartialOverlapSet->count(Ptr3));
EXPECT_TRUE(PartialOverlapSet->count(Ptr4));
EXPECT_TRUE(PartialOverlapSet->count(Ptr5));
EXPECT_NE(*PartialOverlapSet, *ThreeEltSet);
EXPECT_NE(*PartialOverlapSet, *TwoEltSet);
auto *MixOfThreeAndPartialOverlap = ThreeEltSet->merge(PartialOverlapSet);
EXPECT_FALSE(MixOfThreeAndPartialOverlap->empty());
EXPECT_EQ(MixOfThreeAndPartialOverlap->size(), 4u);
EXPECT_TRUE(MixOfThreeAndPartialOverlap->count(Ptr1));
EXPECT_TRUE(MixOfThreeAndPartialOverlap->count(Ptr2));
EXPECT_TRUE(MixOfThreeAndPartialOverlap->count(Ptr3));
EXPECT_TRUE(MixOfThreeAndPartialOverlap->count(Ptr4));
EXPECT_TRUE(MixOfThreeAndPartialOverlap->count(Ptr5));
EXPECT_NE(*MixOfThreeAndPartialOverlap, *PartialOverlapSet);
EXPECT_NE(*MixOfThreeAndPartialOverlap, *ThreeEltSet);
EXPECT_NE(*MixOfThreeAndPartialOverlap, *TwoEltSet);
}
TEST(ImmutablePointerSet, EmptyIntersectionTests) {
llvm::BumpPtrAllocator BPA;
ImmutablePointerSetFactory<unsigned *> F(BPA);
unsigned *Ptr1 = (unsigned *)3;
unsigned *Ptr2 = (unsigned *)4;
unsigned *Ptr3 = (unsigned *)3;
unsigned *Ptr4 = (unsigned *)5;
unsigned *Ptr5 = (unsigned *)6;
llvm::SmallVector<unsigned *, 2> Data1 = {Ptr1, Ptr2};
llvm::SmallVector<unsigned *, 2> Data2 = {Ptr3, Ptr2};
llvm::SmallVector<unsigned *, 2> Data3 = {Ptr4, Ptr5};
llvm::SmallVector<unsigned *, 2> Data4 = {Ptr2, Ptr4};
EXPECT_FALSE(F.get(Data1)->hasEmptyIntersection(F.get(Data2)));
EXPECT_TRUE(F.get(Data1)->hasEmptyIntersection(F.get(Data3)));
EXPECT_FALSE(F.get(Data1)->hasEmptyIntersection(F.get(Data4)));
}