| 23 | namespace tensorflow { |
| 24 | namespace { |
| 25 | TEST(OrderedSetTest, Insert) { |
| 26 | OrderedSet<int> ordered_set; |
| 27 | EXPECT_TRUE(ordered_set.Insert(90)); |
| 28 | EXPECT_TRUE(ordered_set.Insert(100)); |
| 29 | EXPECT_TRUE(ordered_set.Insert(80)); |
| 30 | |
| 31 | EXPECT_FALSE(ordered_set.Insert(100)); |
| 32 | |
| 33 | EXPECT_EQ(ordered_set.Size(), 3); |
| 34 | |
| 35 | EXPECT_TRUE(ordered_set.Contains(90)); |
| 36 | EXPECT_TRUE(ordered_set.Contains(100)); |
| 37 | EXPECT_TRUE(ordered_set.Contains(80)); |
| 38 | |
| 39 | EXPECT_FALSE(ordered_set.Contains(40)); |
| 40 | |
| 41 | std::array<int, 3> expected_sequence = {90, 100, 80}; |
| 42 | EXPECT_EQ(ordered_set.GetSequence(), expected_sequence); |
| 43 | } |
| 44 | |
| 45 | TEST(OrderedSetTest, Erase) { |
| 46 | OrderedSet<int> ordered_set; |