1 //===- llvm/unittest/ADT/EnumeratedArrayTest.cpp ----------------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // EnumeratedArray unit tests.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "llvm/ADT/EnumeratedArray.h"
14 #include "gtest/gtest.h"
15 
16 namespace llvm {
17 
18 //===--------------------------------------------------------------------===//
19 // Test initialization and use of operator[] for both read and write.
20 //===--------------------------------------------------------------------===//
21 
22 TEST(EnumeratedArray, InitAndIndex) {
23 
24   enum class Colors { Red, Blue, Green, Last = Green };
25 
26   EnumeratedArray<float, Colors, Colors::Last, size_t> Array1;
27 
28   Array1[Colors::Red] = 1.0;
29   Array1[Colors::Blue] = 2.0;
30   Array1[Colors::Green] = 3.0;
31 
32   EXPECT_EQ(Array1[Colors::Red], 1.0);
33   EXPECT_EQ(Array1[Colors::Blue], 2.0);
34   EXPECT_EQ(Array1[Colors::Green], 3.0);
35 
36   EnumeratedArray<bool, Colors> Array2(true);
37 
38   EXPECT_TRUE(Array2[Colors::Red]);
39   EXPECT_TRUE(Array2[Colors::Blue]);
40   EXPECT_TRUE(Array2[Colors::Green]);
41 
42   Array2[Colors::Red] = true;
43   Array2[Colors::Blue] = false;
44   Array2[Colors::Green] = true;
45 
46   EXPECT_TRUE(Array2[Colors::Red]);
47   EXPECT_FALSE(Array2[Colors::Blue]);
48   EXPECT_TRUE(Array2[Colors::Green]);
49 }
50 
51 } // namespace llvm
52