1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 // UNSUPPORTED: c++03, c++11, c++14, c++17
10 // UNSUPPORTED: libcpp-has-no-incomplete-ranges
11
12 // iota_view::<iterator>::operator{<,>,<=,>=,==,!=,<=>}
13
14 #include <ranges>
15 #include <compare>
16
17 #include "test_macros.h"
18 #include "test_iterators.h"
19 #include "../types.h"
20
test()21 constexpr bool test() {
22 {
23 // Test `int`, which has operator<=>; the iota iterator should also have operator<=>.
24 using R = std::ranges::iota_view<int>;
25 static_assert(std::three_way_comparable<std::ranges::iterator_t<R>>);
26
27 std::same_as<R> auto r = std::views::iota(42);
28 auto iter1 = r.begin();
29 auto iter2 = iter1 + 1;
30
31 assert(!(iter1 < iter1)); assert(iter1 < iter2); assert(!(iter2 < iter1));
32 assert(iter1 <= iter1); assert(iter1 <= iter2); assert(!(iter2 <= iter1));
33 assert(!(iter1 > iter1)); assert(!(iter1 > iter2)); assert(iter2 > iter1);
34 assert(iter1 >= iter1); assert(!(iter1 >= iter2)); assert(iter2 >= iter1);
35 assert(iter1 == iter1); assert(!(iter1 == iter2)); assert(iter2 == iter2);
36 assert(!(iter1 != iter1)); assert(iter1 != iter2); assert(!(iter2 != iter2));
37
38 assert((iter1 <=> iter2) == std::strong_ordering::less);
39 assert((iter1 <=> iter1) == std::strong_ordering::equal);
40 assert((iter2 <=> iter1) == std::strong_ordering::greater);
41 }
42
43 {
44 // Test a new-school iterator with operator<=>; the iota iterator should also have operator<=>.
45 using It = three_way_contiguous_iterator<int*>;
46 static_assert(std::three_way_comparable<It>);
47 using R = std::ranges::iota_view<It>;
48 static_assert(std::three_way_comparable<std::ranges::iterator_t<R>>);
49
50 int a[] = {1,2,3};
51 std::same_as<R> auto r = std::views::iota(It(a));
52 auto iter1 = r.begin();
53 auto iter2 = iter1 + 1;
54
55 assert(!(iter1 < iter1)); assert(iter1 < iter2); assert(!(iter2 < iter1));
56 assert(iter1 <= iter1); assert(iter1 <= iter2); assert(!(iter2 <= iter1));
57 assert(!(iter1 > iter1)); assert(!(iter1 > iter2)); assert(iter2 > iter1);
58 assert(iter1 >= iter1); assert(!(iter1 >= iter2)); assert(iter2 >= iter1);
59 assert(iter1 == iter1); assert(!(iter1 == iter2)); assert(iter2 == iter2);
60 assert(!(iter1 != iter1)); assert(iter1 != iter2); assert(!(iter2 != iter2));
61
62 assert((iter1 <=> iter2) == std::strong_ordering::less);
63 assert((iter1 <=> iter1) == std::strong_ordering::equal);
64 assert((iter2 <=> iter1) == std::strong_ordering::greater);
65 }
66
67 {
68 // Test an old-school iterator with no operator<=>; the iota iterator shouldn't have operator<=> either.
69 using It = random_access_iterator<int*>;
70 static_assert(!std::three_way_comparable<It>);
71 using R = std::ranges::iota_view<It>;
72 static_assert(!std::three_way_comparable<std::ranges::iterator_t<R>>);
73
74 int a[] = {1,2,3};
75 std::same_as<R> auto r = std::views::iota(It(a));
76 auto iter1 = r.begin();
77 auto iter2 = iter1 + 1;
78
79 assert(!(iter1 < iter1)); assert(iter1 < iter2); assert(!(iter2 < iter1));
80 assert(iter1 <= iter1); assert(iter1 <= iter2); assert(!(iter2 <= iter1));
81 assert(!(iter1 > iter1)); assert(!(iter1 > iter2)); assert(iter2 > iter1);
82 assert(iter1 >= iter1); assert(!(iter1 >= iter2)); assert(iter2 >= iter1);
83 assert(iter1 == iter1); assert(!(iter1 == iter2)); assert(iter2 == iter2);
84 assert(!(iter1 != iter1)); assert(iter1 != iter2); assert(!(iter2 != iter2));
85 }
86
87 return true;
88 }
89
main(int,char **)90 int main(int, char**) {
91 test();
92 static_assert(test());
93
94 return 0;
95 }
96