return find(Key) == end() ? 0 : 1;
}
- template <typename InputTy>
- size_type count(const StringMapEntry<InputTy> &MapEntry) const {
- return count(MapEntry.getKey());
- }
-
/// insert - Insert the specified key/value pair into the map. If the key
/// already exists in the map, return false and ignore the request, otherwise
/// insert it and return true.
for (auto It = Begin; It != End; ++It)
base::insert(std::make_pair(*It, '\0'));
}
-
- template <typename ValueTy>
- std::pair<typename base::iterator, bool>
- insert(const StringMapEntry<ValueTy> &MapEntry) {
- return insert(MapEntry.getKey());
- }
};
} // end namespace llvm
StringExtrasTest.cpp
StringMapTest.cpp
StringRefTest.cpp
- StringSetTest.cpp
StringSwitchTest.cpp
TinyPtrVectorTest.cpp
TripleTest.cpp
//===----------------------------------------------------------------------===//
#include "llvm/ADT/StringMap.h"
+#include "llvm/ADT/StringSet.h"
#include "llvm/ADT/Twine.h"
#include "llvm/Support/DataTypes.h"
#include "gtest/gtest.h"
EXPECT_EQ(Expected, Keys);
}
+TEST_F(StringMapTest, IterSetKeys) {
+ StringSet<> Set;
+ Set.insert("A");
+ Set.insert("B");
+ Set.insert("C");
+ Set.insert("D");
+
+ auto Keys = to_vector<4>(Set.keys());
+ llvm::sort(Keys);
+
+ SmallVector<StringRef, 4> Expected = {"A", "B", "C", "D"};
+ EXPECT_EQ(Expected, Keys);
+}
+
// Create a non-default constructable value
struct StringMapTestStruct {
StringMapTestStruct(int i) : i(i) {}
+++ /dev/null
-//===- llvm/unittest/ADT/StringSetTest.cpp - StringSet unit tests ----------===//
-//
-// 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 "llvm/ADT/StringSet.h"
-#include "gtest/gtest.h"
-using namespace llvm;
-
-namespace {
-
-// Test fixture
-class StringSetTest : public testing::Test {};
-
-TEST_F(StringSetTest, IterSetKeys) {
- StringSet<> Set;
- Set.insert("A");
- Set.insert("B");
- Set.insert("C");
- Set.insert("D");
-
- auto Keys = to_vector<4>(Set.keys());
- llvm::sort(Keys);
-
- SmallVector<StringRef, 4> Expected = {"A", "B", "C", "D"};
- EXPECT_EQ(Expected, Keys);
-}
-
-TEST_F(StringSetTest, InsertAndCountStringMapEntry) {
- // Test insert(StringMapEntry) and count(StringMapEntry)
- // which are required for set_difference(StringSet, StringSet).
- StringSet<> Set;
- StringMapEntry<StringRef> Element(1, "A");
- Set.insert(Element);
- size_t Count = Set.count(Element);
- size_t Expected = 1;
- EXPECT_EQ(Expected, Count);
-}
-
-} // end anonymous namespace