1#include <c10/util/Array.h>
2#include <gtest/gtest.h>
3
4using c10::guts::array;
5using c10::guts::to_array;
6
7namespace {
8namespace test_equals {
9static_assert(array<int, 0>{{}} == array<int, 0>{{}}, "");
10static_assert(array<int, 3>{{2, 3, 4}} == array<int, 3>{{2, 3, 4}}, "");
11static_assert(!(array<int, 3>{{2, 3, 4}} == array<int, 3>{{1, 3, 4}}), "");
12static_assert(!(array<int, 3>{{2, 3, 4}} == array<int, 3>{{2, 1, 4}}), "");
13static_assert(!(array<int, 3>{{2, 3, 4}} == array<int, 3>{{2, 3, 1}}), "");
14} // namespace test_equals
15
16namespace test_notequals {
17static_assert(!(array<int, 0>{{}} != array<int, 0>{{}}), "");
18static_assert(!(array<int, 3>{{2, 3, 4}} != array<int, 3>{{2, 3, 4}}), "");
19static_assert(array<int, 3>{{2, 3, 4}} != array<int, 3>{{1, 3, 4}}, "");
20static_assert(array<int, 3>{{2, 3, 4}} != array<int, 3>{{2, 1, 4}}, "");
21static_assert(array<int, 3>{{2, 3, 4}} != array<int, 3>{{2, 3, 1}}, "");
22} // namespace test_notequals
23
24namespace test_lessthan {
25static_assert(!(array<int, 0>{{}} < array<int, 0>{{}}), "");
26static_assert(!(array<int, 1>{{2}} < array<int, 1>{{1}}), "");
27static_assert(array<int, 1>{{1}} < array<int, 1>{{2}}, "");
28static_assert(!(array<int, 3>{{1, 2, 3}} < array<int, 3>{{1, 2, 3}}), "");
29static_assert(array<int, 3>{{1, 2, 3}} < array<int, 3>{{2, 2, 3}}, "");
30static_assert(!(array<int, 3>{{1, 2, 3}} < array<int, 3>{{0, 2, 3}}), "");
31static_assert(array<int, 3>{{1, 2, 3}} < array<int, 3>{{1, 3, 3}}, "");
32static_assert(!(array<int, 3>{{1, 2, 3}} < array<int, 3>{{1, 1, 3}}), "");
33static_assert(array<int, 3>{{1, 2, 3}} < array<int, 3>{{1, 2, 4}}, "");
34static_assert(!(array<int, 3>{{1, 2, 3}} < array<int, 3>{{1, 2, 2}}), "");
35} // namespace test_lessthan
36
37namespace test_greaterthan {
38static_assert(!(array<int, 0>{{}} > array<int, 0>{{}}), "");
39static_assert(!(array<int, 1>{{1}} > array<int, 1>{{2}}), "");
40static_assert(array<int, 1>{{2}} > array<int, 1>{{1}}, "");
41static_assert(!(array<int, 3>{{1, 2, 3}} > array<int, 3>{{1, 2, 3}}), "");
42static_assert(array<int, 3>{{2, 2, 3}} > array<int, 3>{{1, 2, 3}}, "");
43static_assert(!(array<int, 3>{{0, 2, 3}} > array<int, 3>{{1, 2, 3}}), "");
44static_assert(array<int, 3>{{1, 3, 3}} > array<int, 3>{{1, 2, 3}}, "");
45static_assert(!(array<int, 3>{{1, 1, 3}} > array<int, 3>{{1, 2, 3}}), "");
46static_assert(array<int, 3>{{1, 2, 4}} > array<int, 3>{{1, 2, 3}}, "");
47static_assert(!(array<int, 3>{{1, 2, 2}} > array<int, 3>{{1, 2, 3}}), "");
48} // namespace test_greaterthan
49
50namespace test_lessequals {
51static_assert(array<int, 0>{{}} <= array<int, 0>{{}}, "");
52static_assert(!(array<int, 1>{{2}} <= array<int, 1>{{1}}), "");
53static_assert(array<int, 1>{{1}} <= array<int, 1>{{2}}, "");
54static_assert(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{1, 2, 3}}, "");
55static_assert(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{2, 2, 3}}, "");
56static_assert(!(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{0, 2, 3}}), "");
57static_assert(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{1, 3, 3}}, "");
58static_assert(!(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{1, 1, 3}}), "");
59static_assert(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{1, 2, 4}}, "");
60static_assert(!(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{1, 2, 2}}), "");
61} // namespace test_lessequals
62
63namespace test_greaterequals {
64static_assert(array<int, 0>{{}} >= array<int, 0>{{}}, "");
65static_assert(!(array<int, 1>{{1}} >= array<int, 1>{{2}}), "");
66static_assert(array<int, 1>{{2}} >= array<int, 1>{{1}}, "");
67static_assert(array<int, 3>{{1, 2, 3}} >= array<int, 3>{{1, 2, 3}}, "");
68static_assert(array<int, 3>{{2, 2, 3}} >= array<int, 3>{{1, 2, 3}}, "");
69static_assert(!(array<int, 3>{{0, 2, 3}} >= array<int, 3>{{1, 2, 3}}), "");
70static_assert(array<int, 3>{{1, 3, 3}} >= array<int, 3>{{1, 2, 3}}, "");
71static_assert(!(array<int, 3>{{1, 1, 3}} >= array<int, 3>{{1, 2, 3}}), "");
72static_assert(array<int, 3>{{1, 2, 4}} >= array<int, 3>{{1, 2, 3}}, "");
73static_assert(!(array<int, 3>{{1, 2, 2}} >= array<int, 3>{{1, 2, 3}}), "");
74} // namespace test_greaterequals
75
76namespace test_tail {
77static_assert(array<int, 2>{{3, 4}} == tail(array<int, 3>{{2, 3, 4}}), "");
78static_assert(array<int, 0>{{}} == tail(array<int, 1>{{3}}), "");
79} // namespace test_tail
80
81namespace test_prepend {
82static_assert(
83 array<int, 3>{{2, 3, 4}} == prepend(2, array<int, 2>{{3, 4}}),
84 "");
85static_assert(array<int, 1>{{3}} == prepend(3, array<int, 0>{{}}), "");
86} // namespace test_prepend
87
88namespace test_to_std_array {
89// NOLINTNEXTLINE(cppcoreguidelines-avoid-c-arrays,modernize-avoid-c-arrays)
90constexpr int obj2[3] = {3, 5, 6};
91static_assert(array<int, 3>{{3, 5, 6}} == to_array(obj2), "");
92static_assert(array<int, 3>{{3, 5, 6}} == to_array<int, 3>({3, 5, 6}), "");
93} // namespace test_to_std_array
94
95} // namespace
96