mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
129 lines
3.1 KiB
C++
129 lines
3.1 KiB
C++
#include "swift/Basic/TreeScopedHashTable.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace swift;
|
|
|
|
TEST(TreeScopedHashTableTest, T1) {
|
|
using HashtableTy = TreeScopedHashTable<unsigned, unsigned>;
|
|
using ScopeTy = HashtableTy::ScopeTy;
|
|
HashtableTy HT;
|
|
|
|
auto S1 = new ScopeTy(HT, 0);
|
|
HT.insertIntoScope(*S1, 1, 1001);
|
|
HT.insertIntoScope(*S1, 2, 1002);
|
|
|
|
auto S2 = new ScopeTy(HT, S1);
|
|
HT.insertIntoScope(*S2, 3, 2003);
|
|
HT.insertIntoScope(*S2, 4, 2004);
|
|
|
|
EXPECT_EQ(1001U, HT.lookup(*S1, 1));
|
|
EXPECT_EQ(1002U, HT.lookup(*S1, 2));
|
|
EXPECT_EQ(0U, HT.lookup(*S1, 3));
|
|
EXPECT_EQ(0U, HT.lookup(*S1, 4));
|
|
|
|
EXPECT_EQ(1001U, HT.lookup(*S2, 1));
|
|
EXPECT_EQ(1002U, HT.lookup(*S2, 2));
|
|
EXPECT_EQ(2003U, HT.lookup(*S2, 3));
|
|
EXPECT_EQ(2004U, HT.lookup(*S2, 4));
|
|
|
|
delete S2;
|
|
delete S1;
|
|
}
|
|
|
|
TEST(TreeScopedHashTableTest, T2) {
|
|
using HashtableTy = TreeScopedHashTable<unsigned, unsigned>;
|
|
using ScopeTy = HashtableTy::ScopeTy;
|
|
HashtableTy HT;
|
|
|
|
auto S1 = new ScopeTy(HT, 0);
|
|
HT.insertIntoScope(*S1, 1, 1001);
|
|
HT.insertIntoScope(*S1, 2, 1002);
|
|
|
|
auto S2 = new ScopeTy(HT, S1);
|
|
HT.insertIntoScope(*S2, 3, 2003);
|
|
HT.insertIntoScope(*S2, 4, 2004);
|
|
|
|
auto S3 = new ScopeTy(HT, S1);
|
|
HT.insertIntoScope(*S3, 1, 3001);
|
|
HT.insertIntoScope(*S3, 4, 3004);
|
|
|
|
EXPECT_EQ(1001U, HT.lookup(*S1, 1));
|
|
EXPECT_EQ(1002U, HT.lookup(*S1, 2));
|
|
EXPECT_EQ(0U, HT.lookup(*S1, 3));
|
|
EXPECT_EQ(0U, HT.lookup(*S1, 4));
|
|
|
|
EXPECT_EQ(1001U, HT.lookup(*S2, 1));
|
|
EXPECT_EQ(1002U, HT.lookup(*S2, 2));
|
|
EXPECT_EQ(2003U, HT.lookup(*S2, 3));
|
|
EXPECT_EQ(2004U, HT.lookup(*S2, 4));
|
|
|
|
EXPECT_EQ(3001U, HT.lookup(*S3, 1));
|
|
EXPECT_EQ(1002U, HT.lookup(*S3, 2));
|
|
EXPECT_EQ(0U, HT.lookup(*S3, 3));
|
|
EXPECT_EQ(3004U, HT.lookup(*S3, 4));
|
|
|
|
EXPECT_EQ(true, HT.count(*S3, 1));
|
|
EXPECT_EQ(true, HT.count(*S3, 2));
|
|
EXPECT_EQ(false, HT.count(*S3, 3));
|
|
EXPECT_EQ(true, HT.count(*S3, 4));
|
|
|
|
delete S2;
|
|
|
|
EXPECT_EQ(1001U, HT.lookup(*S1, 1));
|
|
EXPECT_EQ(1002U, HT.lookup(*S1, 2));
|
|
EXPECT_EQ(0U, HT.lookup(*S1, 3));
|
|
EXPECT_EQ(0U, HT.lookup(*S1, 4));
|
|
|
|
EXPECT_EQ(3001U, HT.lookup(*S3, 1));
|
|
EXPECT_EQ(1002U, HT.lookup(*S3, 2));
|
|
EXPECT_EQ(0U, HT.lookup(*S3, 3));
|
|
EXPECT_EQ(3004U, HT.lookup(*S3, 4));
|
|
|
|
delete S3;
|
|
|
|
EXPECT_EQ(1001U, HT.lookup(*S1, 1));
|
|
EXPECT_EQ(1002U, HT.lookup(*S1, 2));
|
|
EXPECT_EQ(0U, HT.lookup(*S1, 3));
|
|
EXPECT_EQ(0U, HT.lookup(*S1, 4));
|
|
|
|
delete S1;
|
|
}
|
|
|
|
TEST(TreeScopedHashTableTest, IteratorTest) {
|
|
using HashtableTy = TreeScopedHashTable<unsigned, int>;
|
|
using ScopeTy = HashtableTy::ScopeTy;
|
|
HashtableTy HT;
|
|
|
|
auto S1 = new ScopeTy(HT, 0);
|
|
HT.insertIntoScope(*S1, 1, 1001);
|
|
HT.insertIntoScope(*S1, 2, 1002);
|
|
|
|
auto S2 = new ScopeTy(HT, S1);
|
|
HT.insertIntoScope(*S2, 1, 2001);
|
|
HT.insertIntoScope(*S2, 4, 2004);
|
|
|
|
auto S3 = new ScopeTy(HT, S1);
|
|
HT.insertIntoScope(*S3, 1, 3001);
|
|
HT.insertIntoScope(*S3, 4, 3004);
|
|
|
|
{
|
|
auto I = HT.begin(*S1, 1);
|
|
EXPECT_EQ(1001, *I);
|
|
I++;
|
|
EXPECT_EQ(HT.end(), I);
|
|
}
|
|
{
|
|
auto I = HT.begin(*S3, 1);
|
|
EXPECT_EQ(3001, *I);
|
|
I++;
|
|
EXPECT_EQ(1001, *I);
|
|
I++;
|
|
EXPECT_EQ(HT.end(), I);
|
|
}
|
|
|
|
delete S3;
|
|
delete S2;
|
|
delete S1;
|
|
}
|
|
|