forked from catchorg/Catch2
111 lines
3.4 KiB
C++
111 lines
3.4 KiB
C++
|
|
// Copyright Catch2 Authors
|
|
// Distributed under the Boost Software License, Version 1.0.
|
|
// (See accompanying file LICENSE.txt or copy at
|
|
// https://www.boost.org/LICENSE_1_0.txt)
|
|
|
|
// SPDX-License-Identifier: BSL-1.0
|
|
#ifndef CATCH_GENERATORS_RANGE_HPP_INCLUDED
|
|
#define CATCH_GENERATORS_RANGE_HPP_INCLUDED
|
|
|
|
#include <catch2/generators/catch_generators.hpp>
|
|
|
|
#include <iterator>
|
|
#include <type_traits>
|
|
|
|
namespace Catch {
|
|
namespace Generators {
|
|
|
|
|
|
template <typename T>
|
|
class RangeGenerator final : public IGenerator<T> {
|
|
T m_current;
|
|
T m_end;
|
|
T m_step;
|
|
bool m_positive;
|
|
|
|
public:
|
|
RangeGenerator(T const& start, T const& end, T const& step):
|
|
m_current(start),
|
|
m_end(end),
|
|
m_step(step),
|
|
m_positive(m_step > T(0))
|
|
{
|
|
assert(m_current != m_end && "Range start and end cannot be equal");
|
|
assert(m_step != T(0) && "Step size cannot be zero");
|
|
assert(((m_positive && m_current <= m_end) || (!m_positive && m_current >= m_end)) && "Step moves away from end");
|
|
}
|
|
|
|
RangeGenerator(T const& start, T const& end):
|
|
RangeGenerator(start, end, (start < end) ? T(1) : T(-1))
|
|
{}
|
|
|
|
T const& get() const override {
|
|
return m_current;
|
|
}
|
|
|
|
bool next() override {
|
|
m_current += m_step;
|
|
return (m_positive) ? (m_current < m_end) : (m_current > m_end);
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
GeneratorWrapper<T> range(T const& start, T const& end, T const& step) {
|
|
static_assert(std::is_arithmetic<T>::value && !std::is_same<T, bool>::value, "Type must be numeric");
|
|
return GeneratorWrapper<T>(Catch::Detail::make_unique<RangeGenerator<T>>(start, end, step));
|
|
}
|
|
|
|
template <typename T>
|
|
GeneratorWrapper<T> range(T const& start, T const& end) {
|
|
static_assert(std::is_integral<T>::value && !std::is_same<T, bool>::value, "Type must be an integer");
|
|
return GeneratorWrapper<T>(Catch::Detail::make_unique<RangeGenerator<T>>(start, end));
|
|
}
|
|
|
|
|
|
template <typename T>
|
|
class IteratorGenerator final : public IGenerator<T> {
|
|
static_assert(!std::is_same<T, bool>::value,
|
|
"IteratorGenerator currently does not support bools"
|
|
"because of std::vector<bool> specialization");
|
|
|
|
std::vector<T> m_elems;
|
|
size_t m_current = 0;
|
|
public:
|
|
template <typename InputIterator, typename InputSentinel>
|
|
IteratorGenerator(InputIterator first, InputSentinel last):m_elems(first, last) {
|
|
if (m_elems.empty()) {
|
|
Detail::throw_generator_exception("IteratorGenerator received no valid values");
|
|
}
|
|
}
|
|
|
|
T const& get() const override {
|
|
return m_elems[m_current];
|
|
}
|
|
|
|
bool next() override {
|
|
++m_current;
|
|
return m_current != m_elems.size();
|
|
}
|
|
};
|
|
|
|
template <typename InputIterator,
|
|
typename InputSentinel,
|
|
typename ResultType = typename std::iterator_traits<InputIterator>::value_type>
|
|
GeneratorWrapper<ResultType> from_range(InputIterator from, InputSentinel to) {
|
|
return GeneratorWrapper<ResultType>(Catch::Detail::make_unique<IteratorGenerator<ResultType>>(from, to));
|
|
}
|
|
|
|
template <typename Container,
|
|
typename ResultType = typename Container::value_type>
|
|
GeneratorWrapper<ResultType> from_range(Container const& cnt) {
|
|
return GeneratorWrapper<ResultType>(Catch::Detail::make_unique<IteratorGenerator<ResultType>>(cnt.begin(), cnt.end()));
|
|
}
|
|
|
|
|
|
} // namespace Generators
|
|
} // namespace Catch
|
|
|
|
|
|
#endif // CATCH_GENERATORS_RANGE_HPP_INCLUDED
|