//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
// UNSUPPORTED: c++03, c++11, c++14, c++17
// iota_view::<iterator>::operator{<,>,<=,>=,==,!=,<=>}
#include <ranges>
#include <compare>
#include "test_macros.h"
#include "test_iterators.h"
#include "../types.h"
constexpr bool test() {
{
// Test `int`, which has operator<=>; the iota iterator should also have operator<=>.
using R = std::ranges::iota_view<int>;
static_assert(std::three_way_comparable<std::ranges::iterator_t<R>>);
std::same_as<R> auto r = std::views::iota(42);
auto iter1 = r.begin();
auto iter2 = iter1 + 1;
assert(!(iter1 < iter1)); assert(iter1 < iter2); assert(!(iter2 < iter1));
assert(iter1 <= iter1); assert(iter1 <= iter2); assert(!(iter2 <= iter1));
assert(!(iter1 > iter1)); assert(!(iter1 > iter2)); assert(iter2 > iter1);
assert(iter1 >= iter1); assert(!(iter1 >= iter2)); assert(iter2 >= iter1);
assert(iter1 == iter1); assert(!(iter1 == iter2)); assert(iter2 == iter2);
assert(!(iter1 != iter1)); assert(iter1 != iter2); assert(!(iter2 != iter2));
assert((iter1 <=> iter2) == std::strong_ordering::less);
assert((iter1 <=> iter1) == std::strong_ordering::equal);
assert((iter2 <=> iter1) == std::strong_ordering::greater);
}
{
// Test a new-school iterator with operator<=>; the iota iterator should also have operator<=>.
using It = three_way_contiguous_iterator<int*>;
static_assert(std::three_way_comparable<It>);
using R = std::ranges::iota_view<It>;
static_assert(std::three_way_comparable<std::ranges::iterator_t<R>>);
int a[] = {1,2,3};
std::same_as<R> auto r = std::views::iota(It(a));
auto iter1 = r.begin();
auto iter2 = iter1 + 1;
assert(!(iter1 < iter1)); assert(iter1 < iter2); assert(!(iter2 < iter1));
assert(iter1 <= iter1); assert(iter1 <= iter2); assert(!(iter2 <= iter1));
assert(!(iter1 > iter1)); assert(!(iter1 > iter2)); assert(iter2 > iter1);
assert(iter1 >= iter1); assert(!(iter1 >= iter2)); assert(iter2 >= iter1);
assert(iter1 == iter1); assert(!(iter1 == iter2)); assert(iter2 == iter2);
assert(!(iter1 != iter1)); assert(iter1 != iter2); assert(!(iter2 != iter2));
assert((iter1 <=> iter2) == std::strong_ordering::less);
assert((iter1 <=> iter1) == std::strong_ordering::equal);
assert((iter2 <=> iter1) == std::strong_ordering::greater);
}
{
// Test an old-school iterator with no operator<=>; the iota iterator shouldn't have operator<=> either.
using It = random_access_iterator<int*>;
static_assert(!std::three_way_comparable<It>);
using R = std::ranges::iota_view<It>;
static_assert(!std::three_way_comparable<std::ranges::iterator_t<R>>);
int a[] = {1,2,3};
std::same_as<R> auto r = std::views::iota(It(a));
auto iter1 = r.begin();
auto iter2 = iter1 + 1;
assert(!(iter1 < iter1)); assert(iter1 < iter2); assert(!(iter2 < iter1));
assert(iter1 <= iter1); assert(iter1 <= iter2); assert(!(iter2 <= iter1));
assert(!(iter1 > iter1)); assert(!(iter1 > iter2)); assert(iter2 > iter1);
assert(iter1 >= iter1); assert(!(iter1 >= iter2)); assert(iter2 >= iter1);
assert(iter1 == iter1); assert(!(iter1 == iter2)); assert(iter2 == iter2);
assert(!(iter1 != iter1)); assert(iter1 != iter2); assert(!(iter2 != iter2));
}
return true;
}
int main(int, char**) {
test();
static_assert(test());
return 0;
}