1 //===-- flang/unittests/Runtime/Inquiry.cpp -------------------------------===//
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 #include "flang/Runtime/inquiry.h"
10 #include "gtest/gtest.h"
11 #include "tools.h"
12 #include "flang/Runtime/type-code.h"
13
14 using namespace Fortran::runtime;
15 using Fortran::common::TypeCategory;
16
TEST(Inquiry,LboundDim)17 TEST(Inquiry, LboundDim) {
18 // ARRAY 1 3 5
19 // 2 4 6
20 auto array{MakeArray<TypeCategory::Integer, 4>(
21 std::vector<int>{2, 3}, std::vector<std::int32_t>{1, 2, 3, 4, 5, 6})};
22 array->GetDimension(0).SetLowerBound(0);
23 array->GetDimension(1).SetLowerBound(-1);
24
25 EXPECT_EQ(RTNAME(LboundDim)(*array, 1, __FILE__, __LINE__), std::int64_t{0});
26 EXPECT_EQ(RTNAME(LboundDim)(*array, 2, __FILE__, __LINE__), std::int64_t{-1});
27 }
28
TEST(Inquiry,Lbound)29 TEST(Inquiry, Lbound) {
30 // ARRAY 1 3 5
31 // 2 4 6
32 auto array{MakeArray<TypeCategory::Integer, 4>(
33 std::vector<int>{2, 3}, std::vector<std::int32_t>{1, 2, 3, 4, 5, 6})};
34 array->GetDimension(0).SetLowerBound(0);
35 array->GetDimension(1).SetLowerBound(-1);
36
37 // LBOUND(ARRAY, KIND=1)
38 auto int8Result{
39 MakeArray<TypeCategory::Integer, 1>(std::vector<int>{array->rank()},
40 std::vector<std::int8_t>(array->rank(), 0))};
41 RTNAME(Lbound)
42 (int8Result->raw().base_addr, *array, /*KIND=*/1, __FILE__, __LINE__);
43 EXPECT_EQ(*int8Result->ZeroBasedIndexedElement<std::int8_t>(0), 0);
44 EXPECT_EQ(*int8Result->ZeroBasedIndexedElement<std::int8_t>(1), -1);
45
46 // LBOUND(ARRAY, KIND=4)
47 auto int32Result{
48 MakeArray<TypeCategory::Integer, 4>(std::vector<int>{array->rank()},
49 std::vector<std::int32_t>(array->rank(), 0))};
50 RTNAME(Lbound)
51 (int32Result->raw().base_addr, *array, /*KIND=*/4, __FILE__, __LINE__);
52 EXPECT_EQ(*int32Result->ZeroBasedIndexedElement<std::int32_t>(0), 0);
53 EXPECT_EQ(*int32Result->ZeroBasedIndexedElement<std::int32_t>(1), -1);
54
55 // LBOUND(ARRAY, KIND=8)
56 auto int64Result{
57 MakeArray<TypeCategory::Integer, 8>(std::vector<int>{array->rank()},
58 std::vector<std::int64_t>(array->rank(), 0))};
59 RTNAME(Lbound)
60 (int64Result->raw().base_addr, *array, /*KIND=*/8, __FILE__, __LINE__);
61 EXPECT_EQ(*int64Result->ZeroBasedIndexedElement<std::int64_t>(0), 0);
62 EXPECT_EQ(*int64Result->ZeroBasedIndexedElement<std::int64_t>(1), -1);
63 }
64
TEST(Inquiry,Ubound)65 TEST(Inquiry, Ubound) {
66 // ARRAY 1 3 5
67 // 2 4 6
68 auto array{MakeArray<TypeCategory::Integer, 4>(
69 std::vector<int>{2, 3}, std::vector<std::int32_t>{1, 2, 3, 4, 5, 6})};
70 array->GetDimension(0).SetLowerBound(1000);
71 array->GetDimension(1).SetLowerBound(1);
72
73 // UBOUND(ARRAY, KIND=1)
74 auto int8Result{
75 MakeArray<TypeCategory::Integer, 1>(std::vector<int>{array->rank()},
76 std::vector<std::int8_t>(array->rank(), 0))};
77 RTNAME(Ubound)
78 (int8Result->raw().base_addr, *array, /*KIND=*/1, __FILE__, __LINE__);
79 EXPECT_EQ(*int8Result->ZeroBasedIndexedElement<std::int8_t>(0), -23);
80 EXPECT_EQ(*int8Result->ZeroBasedIndexedElement<std::int8_t>(1), 3);
81
82 // UBOUND(ARRAY, KIND=4)
83 auto int32Result{
84 MakeArray<TypeCategory::Integer, 4>(std::vector<int>{array->rank()},
85 std::vector<std::int32_t>(array->rank(), 0))};
86 RTNAME(Ubound)
87 (int32Result->raw().base_addr, *array, /*KIND=*/4, __FILE__, __LINE__);
88 EXPECT_EQ(*int32Result->ZeroBasedIndexedElement<std::int32_t>(0), 1001);
89 EXPECT_EQ(*int32Result->ZeroBasedIndexedElement<std::int32_t>(1), 3);
90
91 // UBOUND(ARRAY, KIND=8)
92 auto int64Result{
93 MakeArray<TypeCategory::Integer, 8>(std::vector<int>{array->rank()},
94 std::vector<std::int64_t>(array->rank(), 0))};
95 RTNAME(Ubound)
96 (int64Result->raw().base_addr, *array, /*KIND=*/8, __FILE__, __LINE__);
97 EXPECT_EQ(*int64Result->ZeroBasedIndexedElement<std::int64_t>(0), 1001);
98 EXPECT_EQ(*int64Result->ZeroBasedIndexedElement<std::int64_t>(1), 3);
99 }
100
TEST(Inquiry,Size)101 TEST(Inquiry, Size) {
102 // ARRAY 1 3 5
103 // 2 4 6
104 auto array{MakeArray<TypeCategory::Integer, 4>(
105 std::vector<int>{2, 3}, std::vector<std::int32_t>{1, 2, 3, 4, 5, 6})};
106 array->GetDimension(0).SetLowerBound(0); // shouldn't matter
107 array->GetDimension(1).SetLowerBound(-1);
108
109 EXPECT_EQ(RTNAME(SizeDim)(*array, 1, __FILE__, __LINE__), std::int64_t{2});
110 EXPECT_EQ(RTNAME(SizeDim)(*array, 2, __FILE__, __LINE__), std::int64_t{3});
111 EXPECT_EQ(RTNAME(Size)(*array, __FILE__, __LINE__), std::int64_t{6});
112 }
113
TEST(Inquiry,Shape)114 TEST(Inquiry, Shape) {
115 // ARRAY 1 3 5
116 // 2 4 6
117 auto array{MakeArray<TypeCategory::Integer, 4>(
118 std::vector<int>{2, 3}, std::vector<std::int32_t>{1, 2, 3, 4, 5, 6})};
119
120 // SHAPE(ARRAY, KIND=1)
121 auto int8Result{
122 MakeArray<TypeCategory::Integer, 1>(std::vector<int>{array->rank()},
123 std::vector<std::int8_t>(array->rank(), 0))};
124 RTNAME(Shape)
125 (int8Result->raw().base_addr, *array, /*KIND=*/1, __FILE__, __LINE__);
126 EXPECT_EQ(*int8Result->ZeroBasedIndexedElement<std::int8_t>(0), 2);
127 EXPECT_EQ(*int8Result->ZeroBasedIndexedElement<std::int8_t>(1), 3);
128
129 // SHAPE(ARRAY, KIND=4)
130 auto int32Result{
131 MakeArray<TypeCategory::Integer, 4>(std::vector<int>{array->rank()},
132 std::vector<std::int32_t>(array->rank(), 0))};
133 RTNAME(Shape)
134 (int32Result->raw().base_addr, *array, /*KIND=*/4, __FILE__, __LINE__);
135 EXPECT_EQ(*int32Result->ZeroBasedIndexedElement<std::int32_t>(0), 2);
136 EXPECT_EQ(*int32Result->ZeroBasedIndexedElement<std::int32_t>(1), 3);
137
138 // SHAPE(ARRAY, KIND=8)
139 auto int64Result{
140 MakeArray<TypeCategory::Integer, 8>(std::vector<int>{array->rank()},
141 std::vector<std::int64_t>(array->rank(), 0))};
142 RTNAME(Shape)
143 (int64Result->raw().base_addr, *array, /*KIND=*/8, __FILE__, __LINE__);
144 EXPECT_EQ(*int64Result->ZeroBasedIndexedElement<std::int64_t>(0), 2);
145 EXPECT_EQ(*int64Result->ZeroBasedIndexedElement<std::int64_t>(1), 3);
146 }
147