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 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 90 int main(int, char**) { 91 test(); 92 static_assert(test()); 93 94 return 0; 95 } 96