Merge pull request #289 from ericLemanissier/patch-1

gsl::at overload for initializer_list
This commit is contained in:
Neil MacIntosh 2016-06-24 05:35:31 -07:00 committed by GitHub
commit 0535138459
2 changed files with 15 additions and 0 deletions

View File

@ -128,6 +128,10 @@ template <class Cont>
constexpr typename Cont::value_type& at(Cont& cont, size_t index) constexpr typename Cont::value_type& at(Cont& cont, size_t index)
{ Expects(index < cont.size()); return cont[index]; } { Expects(index < cont.size()); return cont[index]; }
template <class T>
constexpr const T& at(std::initializer_list<T> cont, size_t index)
{ Expects(index < cont.size()); return *(cont.begin() + index); }
} // namespace gsl } // namespace gsl

View File

@ -17,6 +17,7 @@
#include <UnitTest++/UnitTest++.h> #include <UnitTest++/UnitTest++.h>
#include <gsl.h> #include <gsl.h>
#include <vector> #include <vector>
#include <initializer_list>
using namespace std; using namespace std;
using namespace gsl; using namespace gsl;
@ -52,6 +53,16 @@ SUITE(at_tests)
CHECK_THROW(at(a, 4), fail_fast); CHECK_THROW(at(a, 4), fail_fast);
} }
TEST(InitializerList)
{
std::initializer_list<int> a = { 1, 2, 3, 4 };
for (int i = 0; i < 4; ++i)
CHECK(at(a, i) == i+1);
CHECK_THROW(at(a, 4), fail_fast);
}
} }
int main(int, const char *[]) int main(int, const char *[])