1 files changed, 43 insertions, 0 deletions
diff --git a/source/blender/blenlib/tests/BLI_index_mask_test.cc b/source/blender/blenlib/tests/BLI_index_mask_test.cc
new file mode 100644
index 00000000000..4d6060e51c9
--- /dev/null
+++ b/source/blender/blenlib/tests/BLI_index_mask_test.cc
@@ -0,0 +1,43 @@
+/* Apache License, Version 2.0 */
+
+#include "BLI_index_mask.hh"
+#include "testing/testing.h"
+
+namespace blender::tests {
+
+TEST(index_mask, DefaultConstructor)
+{
+  IndexMask mask;
+  EXPECT_EQ(mask.min_array_size(), 0);
+  EXPECT_EQ(mask.size(), 0);
+}
+
+TEST(index_mask, ArrayConstructor)
+{
+  [](IndexMask mask) {
+    EXPECT_EQ(mask.size(), 4);
+    EXPECT_EQ(mask.min_array_size(), 8);
+    EXPECT_FALSE(mask.is_range());
+    EXPECT_EQ(mask[0], 3);
+    EXPECT_EQ(mask[1], 5);
+    EXPECT_EQ(mask[2], 6);
+    EXPECT_EQ(mask[3], 7);
+  }({3, 5, 6, 7});
+}
+
+TEST(index_mask, RangeConstructor)
+{
+  IndexMask mask = IndexRange(3, 5);
+  EXPECT_EQ(mask.size(), 5);
+  EXPECT_EQ(mask.min_array_size(), 8);
+  EXPECT_EQ(mask.last(), 7);
+  EXPECT_TRUE(mask.is_range());
+  EXPECT_EQ(mask.as_range().first(), 3);
+  EXPECT_EQ(mask.as_range().last(), 7);
+  Span<int64_t> indices = mask.indices();
+  EXPECT_EQ(indices[0], 3);
+  EXPECT_EQ(indices[1], 4);
+  EXPECT_EQ(indices[2], 5);
+}
+
+}  // namespace blender::tests