Files

53 lines
1.5 KiB
C++

#include <vector>
#include "gtest/gtest.h"
#include "parallel_hashmap/phmap.h"
namespace phmap {
namespace priv {
namespace {
TEST(EraseIf, FlatHashSet_uint32) {
phmap::flat_hash_set<uint32_t> st1 = { 3, 6, 7, 9 };
auto num_erased = erase_if(st1, [](const uint32_t& v) { return v >= 7; });
EXPECT_TRUE(num_erased == 2);
phmap::flat_hash_set<uint32_t> st2 = { 0, 2, 3, 6 };
num_erased = erase_if(st2, [](const uint32_t& v) { return v <= 2; });
EXPECT_TRUE(num_erased == 2);
EXPECT_TRUE(st1 == st2);
}
TEST(EraseIf, FlatHashMap_uint64_uint32) {
using map = phmap::flat_hash_map<uint32_t, uint32_t>;
map st1 = { {3, 0}, {6, 0}, {7, 0}, {9, 0} };
auto num_erased = erase_if(st1, [](const map::value_type& v) { return v.first >= 7; });
EXPECT_TRUE(num_erased == 2);
map st2 = { {0, 0}, {2, 0}, {3, 0}, {6, 0} };
num_erased = erase_if(st2, [](const map::value_type& v) { return v.first <= 2; });
EXPECT_TRUE(num_erased == 2);
EXPECT_TRUE(st1 == st2);
}
TEST(EraseIf, ParallelFlatHashMap_uint64_uint32) {
using map = phmap::parallel_flat_hash_map<uint32_t, uint32_t>;
map st1 = { {3, 0}, {6, 0}, {7, 0}, {9, 0} };
auto num_erased = erase_if(st1, [](const map::value_type& v) { return v.first >= 7; });
EXPECT_TRUE(num_erased == 2);
map st2 = { {0, 0}, {2, 0}, {3, 0}, {6, 0} };
num_erased = erase_if(st2, [](const map::value_type& v) { return v.first <= 2; });
EXPECT_TRUE(num_erased == 2);
EXPECT_TRUE(st1 == st2);
}
}
}
}