1 files changed, 17 insertions, 2 deletions
diff --git a/tests/gtests/blenlib/BLI_map_test.cc b/tests/gtests/blenlib/BLI_map_test.cc
index 7be23157e94..5a19216fa7c 100644
--- a/tests/gtests/blenlib/BLI_map_test.cc
+++ b/tests/gtests/blenlib/BLI_map_test.cc
@@ -1,5 +1,5 @@
-#include "BLI_map.h"
-#include "BLI_set.h"
+#include "BLI_map.hh"
+#include "BLI_set.hh"
 #include "testing/testing.h"
 
 using BLI::Map;
@@ -9,16 +9,20 @@ TEST(map, DefaultConstructor)
 {
   IntFloatMap map;
   EXPECT_EQ(map.size(), 0);
+  EXPECT_TRUE(map.is_empty());
 }
 
 TEST(map, AddIncreasesSize)
 {
   IntFloatMap map;
   EXPECT_EQ(map.size(), 0);
+  EXPECT_TRUE(map.is_empty());
   map.add(2, 5.0f);
   EXPECT_EQ(map.size(), 1);
+  EXPECT_FALSE(map.is_empty());
   map.add(6, 2.0f);
   EXPECT_EQ(map.size(), 2);
+  EXPECT_FALSE(map.is_empty());
 }
 
 TEST(map, Contains)
@@ -209,6 +213,17 @@ TEST(map, AddOverride)
   EXPECT_EQ(map.lookup(3), 7.0f);
 }
 
+TEST(map, LookupOrAddDefault)
+{
+  IntFloatMap map;
+  map.lookup_or_add_default(3) = 6;
+  EXPECT_EQ(map.lookup(3), 6);
+  map.lookup_or_add_default(5) = 2;
+  EXPECT_EQ(map.lookup(5), 2);
+  map.lookup_or_add_default(3) += 4;
+  EXPECT_EQ(map.lookup(3), 10);
+}
+
 TEST(map, MoveConstructorSmall)
 {
   IntFloatMap map1;