2020-05-02 16:47:33 -05:00
|
|
|
#ifndef THIS_HASH_MAP
|
|
|
|
|
#define THIS_HASH_MAP parallel_flat_hash_map
|
|
|
|
|
#define THIS_TEST_NAME ParallelFlatHashMap
|
|
|
|
|
#endif
|
|
|
|
|
|
2020-05-02 20:45:46 -05:00
|
|
|
#include "flat_hash_map_test.cc"
|
|
|
|
|
|
2020-05-02 16:47:33 -05:00
|
|
|
namespace phmap {
|
2020-07-27 09:14:43 -04:00
|
|
|
namespace priv {
|
2020-05-02 16:47:33 -05:00
|
|
|
namespace {
|
|
|
|
|
|
|
|
|
|
TEST(THIS_TEST_NAME, ThreadSafeContains) {
|
|
|
|
|
// We can't test mutable keys, or non-copyable keys with ThisMap.
|
|
|
|
|
// Test that the nodes have the proper API.
|
2020-10-18 10:53:29 -04:00
|
|
|
using Map = ThisMap<int, int>;
|
|
|
|
|
|
|
|
|
|
Map m = { {1, 7}, {2, 9} };
|
|
|
|
|
const Map& const_m(m);
|
2020-08-07 12:58:43 -04:00
|
|
|
|
|
|
|
|
auto val = 0;
|
|
|
|
|
auto get_value = [&val](const int& v) { val = v; };
|
|
|
|
|
EXPECT_TRUE(const_m.if_contains(2, get_value));
|
2020-05-02 16:47:33 -05:00
|
|
|
EXPECT_EQ(val, 9);
|
|
|
|
|
|
2020-08-07 12:58:43 -04:00
|
|
|
EXPECT_FALSE(m.if_contains(3, get_value));
|
|
|
|
|
|
2020-08-16 18:27:20 -04:00
|
|
|
auto set_value = [](int& v) { v = 11; };
|
|
|
|
|
EXPECT_TRUE(m.modify_if(2, set_value));
|
2020-08-07 12:58:43 -04:00
|
|
|
EXPECT_EQ(m[2], 11);
|
2020-08-07 13:10:31 -04:00
|
|
|
|
2020-10-18 11:23:57 -04:00
|
|
|
EXPECT_FALSE(m.modify_if(3, set_value)); // because m[3] does not exist
|
2020-08-16 18:27:20 -04:00
|
|
|
|
|
|
|
|
// overwrite an existing value
|
|
|
|
|
m.try_emplace_l(2, [](int& v) { v = 5; });
|
|
|
|
|
EXPECT_EQ(m[2], 5);
|
|
|
|
|
|
2020-10-16 20:05:51 -05:00
|
|
|
// insert a value that is not already present. Will be default initialised to 0 and lambda not called
|
2020-10-18 11:23:57 -04:00
|
|
|
m.try_emplace_l(3,
|
|
|
|
|
[](int& v) { v = 6; }, // called only when key was already present
|
|
|
|
|
1); // argument to construct new value is key not present
|
|
|
|
|
EXPECT_EQ(m[3], 1);
|
2020-08-16 18:27:20 -04:00
|
|
|
|
2020-10-16 20:05:51 -05:00
|
|
|
// insert a value that is not already present, provide argument to value-construct it
|
2020-10-18 11:23:57 -04:00
|
|
|
m.try_emplace_l(4,
|
|
|
|
|
[](int& ) {}, // called only when key was already present
|
|
|
|
|
999); // argument to construct new value is key not present
|
2020-08-16 18:44:12 -04:00
|
|
|
EXPECT_EQ(m[4], 999);
|
2020-10-16 20:05:51 -05:00
|
|
|
|
|
|
|
|
// insert a value that is not already present.
|
2020-10-18 11:23:57 -04:00
|
|
|
// right now m[5] does not exist
|
2020-10-16 23:06:39 -04:00
|
|
|
m.lazy_emplace_l(5,
|
2020-10-18 11:23:57 -04:00
|
|
|
[](int& v) { v = 6; }, // called only when key was already present
|
|
|
|
|
[](const Map::constructor& ctor) { ctor(5, 13); }); // construct value_type in place when key not present
|
2020-10-16 23:06:39 -04:00
|
|
|
|
2020-10-16 20:05:51 -05:00
|
|
|
EXPECT_EQ(m[5], 13);
|
|
|
|
|
|
2020-10-18 11:23:57 -04:00
|
|
|
// change a value that is present. Currently m[5] == 13
|
2020-10-16 23:06:39 -04:00
|
|
|
m.lazy_emplace_l(5,
|
2020-10-18 11:23:57 -04:00
|
|
|
[](int& v) { v = 6; }, // called only when key was already present
|
|
|
|
|
[](const Map::constructor& ctor) { ctor(5, 13); }); // construct value_type in place when key not present
|
2020-10-16 20:05:51 -05:00
|
|
|
EXPECT_EQ(m[5], 6);
|
2021-03-15 11:20:49 -04:00
|
|
|
|
|
|
|
|
// test erase_if
|
|
|
|
|
// -------------
|
2021-03-15 12:24:22 -04:00
|
|
|
EXPECT_EQ(m.erase_if(9, [](int& v) { assert(0); return v==12; }), false); // m[9] not present - lambda not called
|
2021-03-15 11:20:49 -04:00
|
|
|
EXPECT_EQ(m.erase_if(5, [](int& v) { return v==12; }), false); // m[5] == 6, so erase not performed
|
|
|
|
|
EXPECT_EQ(m[5], 6);
|
|
|
|
|
EXPECT_EQ(m.erase_if(5, [](int& v) { return v==6; }), true); // lambda returns true, so m[5] erased
|
|
|
|
|
EXPECT_EQ(m[5], 0);
|
|
|
|
|
|
|
|
|
|
|
2020-05-02 16:47:33 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
} // namespace
|
2020-07-27 09:14:43 -04:00
|
|
|
} // namespace priv
|
2020-05-02 16:47:33 -05:00
|
|
|
} // namespace phmap
|