Add erase_if tests

include/boost/unordered/unordered_flat_map.hpp

@@ -48,10 +48,8 @@ namespace boost {
         static moved_type move(value_type& x)
         {
           // TODO: we probably need to launder here
-          return {
-            std::move(const_cast<raw_key_type&>(x.first)),
-            std::move(const_cast<raw_mapped_type&>(x.second))
-          };
+          return {std::move(const_cast<raw_key_type&>(x.first)),
+            std::move(const_cast<raw_mapped_type&>(x.second))};
         }
       };
 
@@ -61,6 +59,10 @@ namespace boost {
 
       table_type table_;
 
+      template <class K, class V, class H, class KE, class A, class Pred>
+      typename unordered_flat_map<K, V, H, KE, A>::size_type friend erase_if(
+        unordered_flat_map<K, V, H, KE, A>& set, Pred pred);
+
     public:
       using key_type = Key;
       using mapped_type = T;
@@ -572,6 +574,15 @@ namespace boost {
     {
       lhs.swap(rhs);
     }
+
+    template <class Key, class T, class Hash, class KeyEqual, class Allocator,
+      class Pred>
+    typename unordered_flat_map<Key, T, Hash, KeyEqual, Allocator>::size_type
+    erase_if(
+      unordered_flat_map<Key, T, Hash, KeyEqual, Allocator>& map, Pred pred)
+    {
+      return erase_if(map.table_, pred);
+    }
   } // namespace unordered
 } // namespace boost
 

include/boost/unordered/unordered_flat_set.hpp

@@ -42,6 +42,10 @@ namespace boost {
 
       table_type table_;
 
+      template <class K, class H, class KE, class A, class Pred>
+      typename unordered_flat_set<K, H, KE, A>::size_type friend erase_if(
+        unordered_flat_set<K, H, KE, A>& set, Pred pred);
+
     public:
       using key_type = Key;
       using value_type = typename set_types::value_type;
@@ -453,6 +457,14 @@ namespace boost {
     {
       lhs.swap(rhs);
     }
+
+    template <class Key, class Hash, class KeyEqual, class Allocator,
+      class Pred>
+    typename unordered_flat_set<Key, Hash, KeyEqual, Allocator>::size_type
+    erase_if(unordered_flat_set<Key, Hash, KeyEqual, Allocator>& set, Pred pred)
+    {
+      return erase_if(set.table_, pred);
+    }
   } // namespace unordered
 } // namespace boost