1 // llvm/unittest/ADT/ImmutableSetTest.cpp - ImmutableSet unit tests -*- C++ -*-//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "gtest/gtest.h"
11 #include "llvm/ADT/ImmutableSet.h"
16 class ImmutableSetTest : public testing::Test {
20 TEST_F(ImmutableSetTest, EmptyIntSetTest) {
21 ImmutableSet<int>::Factory f;
23 EXPECT_TRUE(f.GetEmptySet() == f.GetEmptySet());
24 EXPECT_FALSE(f.GetEmptySet() != f.GetEmptySet());
25 EXPECT_TRUE(f.GetEmptySet().isEmpty());
27 ImmutableSet<int> S = f.GetEmptySet();
28 EXPECT_EQ(0u, S.getHeight());
29 EXPECT_TRUE(S.begin() == S.end());
30 EXPECT_FALSE(S.begin() != S.end());
34 TEST_F(ImmutableSetTest, OneElemIntSetTest) {
35 ImmutableSet<int>::Factory f;
36 ImmutableSet<int> S = f.GetEmptySet();
38 ImmutableSet<int> S2 = f.Add(S, 3);
39 EXPECT_TRUE(S.isEmpty());
40 EXPECT_FALSE(S2.isEmpty());
41 EXPECT_FALSE(S == S2);
43 EXPECT_FALSE(S.contains(3));
44 EXPECT_TRUE(S2.contains(3));
45 EXPECT_FALSE(S2.begin() == S2.end());
46 EXPECT_TRUE(S2.begin() != S2.end());
48 ImmutableSet<int> S3 = f.Add(S, 2);
49 EXPECT_TRUE(S.isEmpty());
50 EXPECT_FALSE(S3.isEmpty());
51 EXPECT_FALSE(S == S3);
53 EXPECT_FALSE(S.contains(2));
54 EXPECT_TRUE(S3.contains(2));
56 EXPECT_FALSE(S2 == S3);
57 EXPECT_TRUE(S2 != S3);
58 EXPECT_FALSE(S2.contains(2));
59 EXPECT_FALSE(S3.contains(3));
62 TEST_F(ImmutableSetTest, MultiElemIntSetTest) {
63 ImmutableSet<int>::Factory f;
64 ImmutableSet<int> S = f.GetEmptySet();
66 ImmutableSet<int> S2 = f.Add(f.Add(f.Add(S, 3), 4), 5);
67 ImmutableSet<int> S3 = f.Add(f.Add(f.Add(S2, 9), 20), 43);
68 ImmutableSet<int> S4 = f.Add(S2, 9);
70 EXPECT_TRUE(S.isEmpty());
71 EXPECT_FALSE(S2.isEmpty());
72 EXPECT_FALSE(S3.isEmpty());
73 EXPECT_FALSE(S4.isEmpty());
75 EXPECT_FALSE(S.contains(3));
76 EXPECT_FALSE(S.contains(9));
78 EXPECT_TRUE(S2.contains(3));
79 EXPECT_TRUE(S2.contains(4));
80 EXPECT_TRUE(S2.contains(5));
81 EXPECT_FALSE(S2.contains(9));
82 EXPECT_FALSE(S2.contains(0));
84 EXPECT_TRUE(S3.contains(43));
85 EXPECT_TRUE(S3.contains(20));
86 EXPECT_TRUE(S3.contains(9));
87 EXPECT_TRUE(S3.contains(3));
88 EXPECT_TRUE(S3.contains(4));
89 EXPECT_TRUE(S3.contains(5));
90 EXPECT_FALSE(S3.contains(0));
92 EXPECT_TRUE(S4.contains(9));
93 EXPECT_TRUE(S4.contains(3));
94 EXPECT_TRUE(S4.contains(4));
95 EXPECT_TRUE(S4.contains(5));
96 EXPECT_FALSE(S4.contains(20));
97 EXPECT_FALSE(S4.contains(43));
100 TEST_F(ImmutableSetTest, RemoveIntSetTest) {
101 ImmutableSet<int>::Factory f;
102 ImmutableSet<int> S = f.GetEmptySet();
104 ImmutableSet<int> S2 = f.Add(f.Add(S, 4), 5);
105 ImmutableSet<int> S3 = f.Add(S2, 3);
106 ImmutableSet<int> S4 = f.Remove(S3, 3);
108 EXPECT_TRUE(S3.contains(3));
109 EXPECT_FALSE(S2.contains(3));
110 EXPECT_FALSE(S4.contains(3));
112 EXPECT_TRUE(S2 == S4);
113 EXPECT_TRUE(S3 != S2);
114 EXPECT_TRUE(S3 != S4);
116 EXPECT_TRUE(S3.contains(4));
117 EXPECT_TRUE(S3.contains(5));
119 EXPECT_TRUE(S4.contains(4));
120 EXPECT_TRUE(S4.contains(5));
124 static char *ptr; // tmp var
127 MyIter() : counter(0) {}
128 void operator()(char c) {
134 TEST_F(ImmutableSetTest, CallbackCharSetTest) {
135 ImmutableSet<char>::Factory f;
136 ImmutableSet<char> S = f.GetEmptySet();
138 ImmutableSet<char> S2 = f.Add(f.Add(f.Add(S, 'a'), 'e'), 'i');
139 ImmutableSet<char> S3 = f.Add(f.Add(S2, 'o'), 'u');
141 char buffer[6] = {0};
143 S3.foreach<MyIter>();
145 ASSERT_EQ(buffer[0], 'a');
146 ASSERT_EQ(buffer[1], 'e');
147 ASSERT_EQ(buffer[2], 'i');
148 ASSERT_EQ(buffer[3], 'o');
149 ASSERT_EQ(buffer[4], 'u');
150 ASSERT_EQ(buffer[5], 0);
153 TEST_F(ImmutableSetTest, Callback2CharSetTest) {
154 ImmutableSet<char>::Factory f;
155 ImmutableSet<char> S = f.GetEmptySet();
157 ImmutableSet<char> S2 = f.Add(f.Add(f.Add(S, 'b'), 'c'), 'd');
158 ImmutableSet<char> S3 = f.Add(f.Add(f.Add(S2, 'f'), 'g'), 'h');
160 char buffer[7] = {0};
163 S3.foreach<MyIter>(obj);
165 ASSERT_EQ(buffer[0], 'b');
166 ASSERT_EQ(buffer[1], 'c');
167 ASSERT_EQ(buffer[2], 'd');
168 ASSERT_EQ(buffer[3], 'f');
169 ASSERT_EQ(buffer[4], 'g');
170 ASSERT_EQ(buffer[5], 'h');
171 ASSERT_EQ(buffer[6], 0);
173 ASSERT_EQ(obj.counter, 6);
175 S2.foreach<MyIter>(obj);
176 ASSERT_EQ(obj.counter, 6+3);
178 S.foreach<MyIter>(obj);
179 ASSERT_EQ(obj.counter, 6+3+0);
182 TEST_F(ImmutableSetTest, IterLongSetTest) {
183 ImmutableSet<long>::Factory f;
184 ImmutableSet<long> S = f.GetEmptySet();
186 ImmutableSet<long> S2 = f.Add(f.Add(f.Add(S, 0), 1), 2);
187 ImmutableSet<long> S3 = f.Add(f.Add(f.Add(S2, 3), 4), 5);
190 for (ImmutableSet<long>::iterator I = S.begin(), E = S.end(); I != E; ++I) {
196 for (ImmutableSet<long>::iterator I = S2.begin(), E = S2.end(); I != E; ++I) {
202 for (ImmutableSet<long>::iterator I = S3.begin(), E = S3.end(); I != E; I++) {