mirror of
https://github.com/RPCS3/llvm.git
synced 2024-12-28 07:05:03 +00:00
d9ebc5991b
Actually update the changed indexes in the map portion of `MapVector` when erasing from the middle. Add a unit test that checks for this. Note that `MapVector::erase()` is a linear time operation (it was and still is). I'll commit a new method in a moment called `MapVector::remove_if()` that deletes multiple entries in linear time, which should be slightly less painful. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@213084 91177308-0d34-0410-b5e6-96231b3b80d8
71 lines
1.7 KiB
C++
71 lines
1.7 KiB
C++
//===- unittest/ADT/MapVectorTest.cpp - MapVector unit tests ----*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "llvm/ADT/MapVector.h"
|
|
#include <utility>
|
|
|
|
using namespace llvm;
|
|
|
|
TEST(MapVectorTest, insert_pop) {
|
|
MapVector<int, int> MV;
|
|
std::pair<MapVector<int, int>::iterator, bool> R;
|
|
|
|
R = MV.insert(std::make_pair(1, 2));
|
|
ASSERT_EQ(R.first, MV.begin());
|
|
EXPECT_EQ(R.first->first, 1);
|
|
EXPECT_EQ(R.first->second, 2);
|
|
EXPECT_TRUE(R.second);
|
|
|
|
R = MV.insert(std::make_pair(1, 3));
|
|
ASSERT_EQ(R.first, MV.begin());
|
|
EXPECT_EQ(R.first->first, 1);
|
|
EXPECT_EQ(R.first->second, 2);
|
|
EXPECT_FALSE(R.second);
|
|
|
|
R = MV.insert(std::make_pair(4, 5));
|
|
ASSERT_NE(R.first, MV.end());
|
|
EXPECT_EQ(R.first->first, 4);
|
|
EXPECT_EQ(R.first->second, 5);
|
|
EXPECT_TRUE(R.second);
|
|
|
|
EXPECT_EQ(MV.size(), 2u);
|
|
EXPECT_EQ(MV[1], 2);
|
|
EXPECT_EQ(MV[4], 5);
|
|
|
|
MV.pop_back();
|
|
EXPECT_EQ(MV.size(), 1u);
|
|
EXPECT_EQ(MV[1], 2);
|
|
|
|
R = MV.insert(std::make_pair(4, 7));
|
|
ASSERT_NE(R.first, MV.end());
|
|
EXPECT_EQ(R.first->first, 4);
|
|
EXPECT_EQ(R.first->second, 7);
|
|
EXPECT_TRUE(R.second);
|
|
|
|
EXPECT_EQ(MV.size(), 2u);
|
|
EXPECT_EQ(MV[1], 2);
|
|
EXPECT_EQ(MV[4], 7);
|
|
}
|
|
|
|
TEST(MapVectorTest, erase) {
|
|
MapVector<int, int> MV;
|
|
|
|
MV.insert(std::make_pair(1, 2));
|
|
MV.insert(std::make_pair(3, 4));
|
|
MV.insert(std::make_pair(5, 6));
|
|
ASSERT_EQ(MV.size(), 3u);
|
|
|
|
MV.erase(MV.find(1));
|
|
ASSERT_EQ(MV.size(), 2u);
|
|
ASSERT_EQ(MV.find(1), MV.end());
|
|
ASSERT_EQ(MV[3], 4);
|
|
ASSERT_EQ(MV[5], 6);
|
|
}
|